Kevin Hu
commited on
Commit
·
6889379
1
Parent(s):
a8c4af8
Refactor. (#4612)
Browse files### What problem does this PR solve?
### Type of change
- [x] Refactoring
api/db/services/user_service.py
CHANGED
|
@@ -50,12 +50,6 @@ class UserService(CommonService):
|
|
| 50 |
else:
|
| 51 |
return None
|
| 52 |
|
| 53 |
-
@classmethod
|
| 54 |
-
@DB.connection_context()
|
| 55 |
-
def user_gateway(cls, tenant_id):
|
| 56 |
-
hashobj = hashlib.sha256(tenant_id.encode("utf-8"))
|
| 57 |
-
return int(hashobj.hexdigest(), 16)%len(MINIO)
|
| 58 |
-
|
| 59 |
@classmethod
|
| 60 |
@DB.connection_context()
|
| 61 |
def save(cls, **kwargs):
|
|
@@ -134,6 +128,12 @@ class TenantService(CommonService):
|
|
| 134 |
if num == 0:
|
| 135 |
raise LookupError("Tenant not found which is supposed to be there")
|
| 136 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 137 |
|
| 138 |
class UserTenantService(CommonService):
|
| 139 |
model = UserTenant
|
|
|
|
| 50 |
else:
|
| 51 |
return None
|
| 52 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 53 |
@classmethod
|
| 54 |
@DB.connection_context()
|
| 55 |
def save(cls, **kwargs):
|
|
|
|
| 128 |
if num == 0:
|
| 129 |
raise LookupError("Tenant not found which is supposed to be there")
|
| 130 |
|
| 131 |
+
@classmethod
|
| 132 |
+
@DB.connection_context()
|
| 133 |
+
def user_gateway(cls, tenant_id):
|
| 134 |
+
hashobj = hashlib.sha256(tenant_id.encode("utf-8"))
|
| 135 |
+
return int(hashobj.hexdigest(), 16)%len(MINIO)
|
| 136 |
+
|
| 137 |
|
| 138 |
class UserTenantService(CommonService):
|
| 139 |
model = UserTenant
|