Checksum fix

040df5cc · Szeberényi Imre · fc13dc82 · 040df5cc
Commit 040df5cc authored Jan 23, 2022 by Szeberényi Imre
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

disk.py
+3 -0

No files found.
--- a/disk.py
+++ b/disk.py
@@ -59,9 +59,12 @@ class Disk(object):

    @property
    def checksum(self, blocksize=65536):
+        logging.debug("Generating checksum for %s", self.get_path())
        hash = md5()
        with open(self.get_path(), "rb") as f:
            for block in iter(lambda: f.read(blocksize), ""):
+                if len(block) == 0:
+                    break
                hash.update(block)
        return hash.hexdigest()