more hashlib usage

truenas · Dec 6, 2024 · d82410a · d82410a
1 parent ddd0861
commit d82410a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/scale_build/image/iso.py b/scale_build/image/iso.py
@@ -1,4 +1,5 @@
 import glob
+import hashlib
 import itertools
 import os
 import shutil
@@ -201,9 +202,8 @@ def make_iso_file():
 
     image_version = get_image_version(vendor=TRUENAS_VENDOR)
     with open(os.path.join(RELEASE_DIR, f'TrueNAS-SCALE-{image_version}.iso.sha256'), 'w') as f:
-        f.write(run(
-            ['sha256sum', os.path.join(RELEASE_DIR, f'TrueNAS-SCALE-{image_version}.iso')], log=False
-        ).stdout.replace(f'{RELEASE_DIR}/', '').strip())
+        with open(os.path.join(RELEASE_DIR, f'TrueNAS-SCALE-{image_version}.iso'), 'rb') as sf:
+            f.write(hashlib.file_digest(sf, 'sha256').hexdigest())
 
 
 def pruning_cd_basedir_contents():