From e14d6ae441d1cb55789088768ad8e9c91e20ef08 Mon Sep 17 00:00:00 2001 From: Kevin Hu Date: Thu, 23 Jan 2025 18:56:02 +0800 Subject: [PATCH] Refactor. (#4612) ### What problem does this PR solve? ### Type of change - [x] Refactoring --- api/db/services/user_service.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/api/db/services/user_service.py b/api/db/services/user_service.py index 04789fd16..56a4352ce 100644 --- a/api/db/services/user_service.py +++ b/api/db/services/user_service.py @@ -50,12 +50,6 @@ class UserService(CommonService): else: return None - @classmethod - @DB.connection_context() - def user_gateway(cls, tenant_id): - hashobj = hashlib.sha256(tenant_id.encode("utf-8")) - return int(hashobj.hexdigest(), 16)%len(MINIO) - @classmethod @DB.connection_context() def save(cls, **kwargs): @@ -134,6 +128,12 @@ class TenantService(CommonService): if num == 0: raise LookupError("Tenant not found which is supposed to be there") + @classmethod + @DB.connection_context() + def user_gateway(cls, tenant_id): + hashobj = hashlib.sha256(tenant_id.encode("utf-8")) + return int(hashobj.hexdigest(), 16)%len(MINIO) + class UserTenantService(CommonService): model = UserTenant