From f13239560d9a6888d3bf95699ed5fc10395b50ff Mon Sep 17 00:00:00 2001 From: zhaoqingang <zhaoqg0118@163.com> Date: 星期一, 24 二月 2025 18:57:51 +0800 Subject: [PATCH] 历史记录增加 --- app/service/user.py | 63 +++++++++++++++---------------- 1 files changed, 30 insertions(+), 33 deletions(-) diff --git a/app/service/user.py b/app/service/user.py index 42bb7fd..d11683e 100644 --- a/app/service/user.py +++ b/app/service/user.py @@ -1,10 +1,7 @@ import time +import uuid from datetime import datetime -from traceback import print_tb - from app.api import pwd_context -from app.api.dialog import dialog_list -from app.config.config import settings from app.config.const import RAGFLOW, BISHENG, DIFY, USER_STATSU_DELETE, ROLE_STATUS_ON, DEPT_STATUS_ON, \ Dialog_STATSU_ON from app.models import RoleModel, GroupModel, AgentType, role_resource_table, DialogModel, OrganizationModel, \ @@ -12,13 +9,7 @@ from app.models.menu_model import WebMenuModel, MenuCapacityModel from app.models.user_model import UserModel, UserAppModel from Log import logger -from app.service.auth import UserAppDao -from app.service.bisheng import BishengService -from app.service.v2.app_register import AppRegisterDao -from app.service.difyService import DifyService -from app.service.ragflow import RagflowService -from app.service.service_token import get_ragflow_token, get_bisheng_token, get_new_token, get_dify_token -from app.utils.password_handle import generate_password + async def get_user_list(db, page_index: int, page_size: int, keyword: str, role_key: str, user_id): @@ -57,10 +48,13 @@ async def create_user(db, user_name, email, phone, login_name, password, roles, groups, user_id): try: - app_register = AppRegisterDao(db).get_apps() + register_dict = {} token = "" - app_password = generate_password() + + """ + app_password = await generate_password() + app_register = AppRegisterDao(db).get_apps() for app in app_register: if app["id"] == RAGFLOW: service = RagflowService(settings.fwr_base_url) @@ -81,30 +75,33 @@ except Exception as e: logger.error(e) return False - + """ # 瀛樺偍鐢ㄦ埛淇℃伅 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) - pwd = user_model.encrypted_password(app_password) + phone=phone, login_name=login_name,sync_flag=sync_flag) + # pwd = user_model.encrypted_password(app_password) + # user_model.password = pwd user_model.roles = [db.get(RoleModel, roleId) for roleId in roles] - user_model.password = pwd if groups: user_model.groups = [db.get(GroupModel, groupId) for groupId in groups] user_model.creator = user_id db.add(user_model) db.commit() db.refresh(user_model) + """ u_id = user_model.id + user_app_dao = UserAppDao(db) for k, v in register_dict.items(): - await UserAppDao(db).update_and_insert_data(v.get("name"), pwd, v.get("email"), u_id, str(v.get("id")), k) + await user_app_dao.update_and_insert_data(v.get("name"), pwd, v.get("email"), u_id, str(v.get("id")), k) + """ 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): @@ -270,8 +267,9 @@ res = permissions[parent_id].to_router_dict() res["children"] = [get_child(i["id"]) for i in tmp_dit.get(parent_id, [])] return res - - return {"routers": [get_child(i["id"]) for i in tmp_dit.get(parent_id, [])]} + select_list = [get_child(i["id"]) for i in tmp_dit.get(parent_id, [])] + sorted_permission_list = sorted(select_list, key=lambda x: x['seq'], reverse=True) + return {"routers": sorted_permission_list} async def get_user_menus(db, user_id): @@ -329,8 +327,8 @@ knowledge_dict = {} user = db.query(UserModel).filter_by(id=user_id).first() parent_id = "" - # print(111111111111111) - async def role_resource(role_set, permissions, roles): + + async def role_resource_type(role_set, permissions, roles, type): nonlocal parent_id for role in roles: if role.id not in role_set: @@ -338,20 +336,19 @@ for r in role.resources: if r.resource_type_id != "1": - # if not r.resource_id: - # parent_id = r.id - # continue - permissions[r.id] = r.to_router_dict() + permission_dict = r.to_router_dict() + permission_dict["type"] = type + permissions[r.id] = permission_dict - await role_resource(role_set, permissions, user.roles) + await role_resource_type(role_set, permissions, user.roles, "user") for ogt in user.organizations: if ogt.roles: - await role_resource(role_set, permissions, ogt.roles) + await role_resource_type(role_set, permissions, ogt.roles, "org") parent_ogt = ogt.parent while parent_ogt: if parent_ogt.id not in dept_set: - await role_resource(role_set, permissions, parent_ogt.roles) + await role_resource_type(role_set, permissions, parent_ogt.roles, "org") dept_set.add(parent_ogt.id) parent_ogt = parent_ogt.parent @@ -382,7 +379,7 @@ async def get_user_dept(db, user_id): res = {} user = db.query(UserModel).filter_by(id=user_id).first() - res["rows"] = [i.to_dict() for i in user.organizations] + res["rows"] = [i.to_dict() for i in user.organizations if i.status == DEPT_STATUS_ON] return res -- Gitblit v1.8.0