fix checksum

bccebcfd · Guba Sándor · 785fede2 · bccebcfd
Commit bccebcfd authored Oct 15, 2014 by Guba Sándor
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 2 deletions

disk.py
+6 -2

No files found.
--- a/disk.py
+++ b/disk.py
@@ -53,8 +53,12 @@ class Disk(object):
        self.base_name = base_name
    @property
-    def checksum(self):
+    def checksum(self, blocksize=65536):
-        return md5(self.get_path()).hexdigest()
+        hash = md5()
+        with open(self.get_path(), "r+b") as f:
+            for block in iter(lambda: f.read(blocksize), ""):
+                hash.update(block)
+        return hash.hexdigest()
    @classmethod
    def deserialize(cls, desc):