zhaoqingang
2024-12-26 5eff7ec9d00fa835a76c354bad7cc7f284d28b11
新增用户增加rg数据同步
3个文件已修改
19 ■■■■ 已修改文件
app/api/user.py 8 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
app/service/service_token.py 2 ●●● 补丁 | 查看 | 原始文档 | blame | 历史
app/service/user.py 9 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
app/api/user.py
@@ -32,11 +32,11 @@
    pwd = user.pwd
    if not pwd:
        pwd = "basic123456"
    is_create = await create_user(db, user.userName, user.email, user.phone, user.loginName, pwd, user.roles,
    user_flag = await create_user(db, user.userName, user.email, user.phone, user.loginName, pwd, user.roles,
                                  user.groups, current_user.id)
    if not is_create:
        return Response(code=500, msg="user create failure", data={})
    return Response(code=200, msg="user create successfully", data={})
    # if not is_create:
    #     return Response(code=500, msg="user create failure", data={})
    return Response(code=200, msg="user create successfully", data={"userFlag": user_flag})
@user_router.put("/edit_user", response_model=Response)
app/service/service_token.py
@@ -118,4 +118,4 @@
            await ragflow_service.add_user_tenant(token, user.app_id,
                                                  u.email,
                                                  u.app_id)
    update_ragflow_user_tenant(sync_user.id)
    update_ragflow_user_tenant(user.app_id)
app/service/user.py
@@ -1,4 +1,5 @@
import time
import uuid
from datetime import datetime
from traceback import print_tb
@@ -84,10 +85,10 @@
        # 存储用户信息
        hashed_password = pwd_context.hash(password)
        sync_flag = str(uuid.uuid4())
        user_model = UserModel(username=user_name, hashed_password=hashed_password, email=email,
                               ## ragflow_id=ragflow_info.get("id"),bisheng_id=bisheng_info.get("user_id"),
                               phone=phone, login_name=login_name)
                               phone=phone, login_name=login_name,sync_flag=sync_flag)
        pwd = user_model.encrypted_password(app_password)
        user_model.roles = [db.get(RoleModel, roleId) for roleId in roles]
        user_model.password = pwd
@@ -104,8 +105,8 @@
    except Exception as e:
        logger.error(e)
        # db.rollback()
        return False
    return True
        return ''
    return sync_flag
async def edit_user_data(db, user_id, email, phone, login_name, roles, groups):