From 129d52dff2189d863e490dc3313d5295170ee056 Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com>
Date: Wed, 28 May 2025 10:13:47 +0800
Subject: [PATCH] fix: common (#3154)

---
 apps/common/utils/common.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/apps/common/utils/common.py b/apps/common/utils/common.py
index b57639194..af0e0ed4f 100644
--- a/apps/common/utils/common.py
+++ b/apps/common/utils/common.py
@@ -280,7 +280,10 @@ def bulk_create_in_batches(model, data, batch_size=1000):
         model.objects.bulk_create(batch)
 
 
-def get_sha256_hash(_v: str):
+def get_sha256_hash(_v: str | bytes):
     sha256 = hashlib.sha256()
-    sha256.update(_v.encode())
+    if isinstance(_v, str):
+        sha256.update(_v.encode())
+    else:
+        sha256.update(_v)
     return sha256.hexdigest()