瀏覽代碼

Unified user_service.py (#4606)

### What problem does this PR solve?

Unified user_service.py

### Type of change

- [x] Refactoring
tags/v0.16.0
Zhichang Yu 9 月之前
父節點
當前提交
240e7d7c22
No account linked to committer's email address
共有 1 個文件被更改,包括 8 次插入0 次删除
  1. 8
    0
      api/db/services/user_service.py

+ 8
- 0
api/db/services/user_service.py 查看文件

@@ -13,6 +13,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.
#
import hashlib
from datetime import datetime

import peewee
@@ -24,6 +25,7 @@ from api.db.db_models import User, Tenant
from api.db.services.common_service import CommonService
from api.utils import get_uuid, current_timestamp, datetime_format
from api.db import StatusEnum
from rag.settings import MINIO


class UserService(CommonService):
@@ -48,6 +50,12 @@ 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):

Loading…
取消
儲存