AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-11 17:19:03 +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
# 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):