Make default hash lib configurable without code changes via CLI argument

comfy/cli_args.py

@@ -112,6 +112,7 @@ class LatentPreviewMethod(enum.Enum):
 vram_group.add_argument("--novram", action="store_true", help="When lowvram isn't enough.")
 vram_group.add_argument("--cpu", action="store_true", help="To use the CPU for everything (slow).")
 
+parser.add_argument("--default-hashing-function", type=str, choices=['md5', 'sha1', 'sha256', 'sha512'], default='sha256', help="Allows you to choose the hash function to use for duplicate filename / contents comparison. Default is sha256.")
 
 parser.add_argument("--disable-smart-memory", action="store_true", help="Force ComfyUI to agressively offload to regular ram instead of keeping models in vram when it can.")
 parser.add_argument("--deterministic", action="store_true", help="Make pytorch use slower deterministic algorithms when it can. Note that this might not make images deterministic in all cases.")

node_helpers.py

@@ -1,3 +1,7 @@
+import hashlib
+
+from comfy.cli_args import args
+
 from PIL import ImageFile, UnidentifiedImageError
 
 def conditioning_set_values(conditioning, values={}):
@@ -22,3 +26,12 @@ def pillow(fn, arg):
  if prev_value is not None:
  ImageFile.LOAD_TRUNCATED_IMAGES = prev_value
  return x
+
+def hasher():
+ hashfuncs = {
+ "md5": hashlib.md5,
+ "sha1": hashlib.sha1,
+ "sha256": hashlib.sha256,
+ "sha512": hashlib.sha512
+ }
+ return hashfuncs[args.default_hashing_function]

server.py

@@ -25,6 +25,7 @@
 from comfy.cli_args import args
 import comfy.utils
 import comfy.model_management
+import node_helpers
 from app.frontend_management import FrontendManager
 from app.user_manager import UserManager
 
@@ -161,10 +162,12 @@ def get_dir_by_type(dir_type):
  return type_dir, dir_type
 
  def compare_image_hash(filepath, image):
+ hasher = node_helpers.hasher()
+
  # function to compare hashes of two images to see if it already exists, fix to #3465
  if os.path.exists(filepath):
- a = hashlib.sha256()
- b = hashlib.sha256()
+ a = hasher()
+ b = hasher()
  with open(filepath, "rb") as f:
  a.update(f.read())
  b.update(image.file.read())