blake3 #6

Pipfile

@@ -0,0 +1,12 @@
+[[source]]
+url = "https://pypi.org/simple"
+verify_ssl = true
+name = "pypi"
+
+[packages]
+blake3 = ">=0.3.4"
+
+[dev-packages]
+
+[requires]
+python_version = "3.11"

README.md

@@ -39,7 +39,7 @@ Run `chkbit -u PATH` to create/update the chkbit index.
 chkbit will
 
 - create a `.chkbit` index in every subdirectory of the path it was given.
-- update the index with md5/sha512 hashes for every file.
+- update the index with md5/sha512/blake3 hashes for every file.
 - report damage for files that failed the integrity check since the last run (check the exit status).
 
 Run `chkbit PATH` to verify only.
@@ -55,7 +55,7 @@ positional arguments:
 options:
   -h, --help           show this help message and exit
   -u, --update         update indices (without this chkbit will only verify files)
-  --algo ALGO          hash algorithm: md5, sha512
+  --algo ALGO          hash algorithm: md5, sha512, blake3
   -f, --force          force update of damaged items
   -i, --verify-index   verify files in the index only (will not report new files)
   -s, --skip-symlinks  do not follow symlinks
@@ -123,7 +123,7 @@ When you run it again it first checks the modification time,
 
 ### I wish to use a stronger hash algorithm
 
-chkbit now supports sha512. You can specify it with `--algo sha512`.
+chkbit now supports sha512 and blake3. You can specify it with `--algo sha512` or `--algo blake3`.
 
 Note that existing index files will use the hash that they were created with. If you wish to update all hashes you need to delete your existing indexes first.
 

chkbit/context.py

@@ -6,5 +6,5 @@ def __init__(self, verify_index, update, force, hash_algo, skip_symlinks):
         self.hash_algo = hash_algo
         self.skip_symlinks = skip_symlinks
 
-        if hash_algo not in ["md5", "sha512"]:
+        if hash_algo not in ["md5", "sha512", "blake3"]:
             raise Exception(f"{hash_algo} is unknown.")

chkbit/hashfile.py

@@ -9,6 +9,10 @@ def hashfile(path, hash_algo=None):
         h = hashlib.md5()
     elif hash_algo == "sha512":
         h = hashlib.sha512()
+    elif hash_algo == "blake3":
+        from blake3 import blake3
+
+        h = blake3()
     else:
         raise Exception(f"{hash_algo} is unknown.")
 

chkbit/main.py

@@ -46,6 +46,7 @@ def _log(self, idx, stat, path):
 
     def _parse_args(self):
         parser = argparse.ArgumentParser(
+            prog="chkbit",
             description="Checks the data integrity of your files. See https://github.com/laktak/chkbit-py",
             epilog=STATUS_CODES,
             formatter_class=argparse.RawDescriptionHelpFormatter,
@@ -66,7 +67,7 @@ def _parse_args(self):
             "--algo",
             type=str,
             default="md5",
-            help="hash algorithm: md5, sha512",
+            help="hash algorithm: md5, sha512, blake3",
         )
 
         parser.add_argument(

pyproject.toml

@@ -1,11 +1,13 @@
 [project]
 name = "chkbit"
-version = "2.3.0"
+version = "2.4.0"
 description = "chkbit checks the data integrity of your files"
 authors = [
     {name = "Christian Zangl", email = "laktak@cdak.net"},
 ]
-dependencies = []
+dependencies = [
+  "blake3>=0.3.4",
+]
 requires-python = ">=3.6.0"
 readme = "README.md"
 license = {file = "LICENSE"}