AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-14 06:55:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

Unified user_service.py (#4606)

Browse Source 
### What problem does this PR solve?

Unified user_service.py

### Type of change

- [x] Refactoring
This commit is contained in:
Zhichang Yu 2025-01-23 15:49:21 +08:00 committed by GitHub
parent 52fa8bdcf3
commit 240e7d7c22
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
api/db/services/user_service.py
View File

@ -13,6 +13,7 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
# #
import hashlib
from datetime import datetime from datetime import datetime
import peewee import peewee
@ -24,6 +25,7 @@ from api.db.db_models import User, Tenant
from api.db.services.common_service import CommonService from api.db.services.common_service import CommonService
from api.utils import get_uuid, current_timestamp, datetime_format from api.utils import get_uuid, current_timestamp, datetime_format
from api.db import StatusEnum from api.db import StatusEnum
from rag.settings import MINIO
class UserService(CommonService): class UserService(CommonService):
@ -48,6 +50,12 @@ class UserService(CommonService):
else: else:
return None 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 @classmethod
@DB.connection_context() @DB.connection_context()
def save(cls, **kwargs): def save(cls, **kwargs):