From e5e2be29cfc033c976558a4d0b5fa73426818b90 Mon Sep 17 00:00:00 2001 From: zhaoqingang <zhaoqg0118@163.com> Date: 星期三, 26 二月 2025 11:30:48 +0800 Subject: [PATCH] 增加系统信息修改 --- app/service/user.py | 42 ++++++++++++++++++------------------------ 1 files changed, 18 insertions(+), 24 deletions(-) diff --git a/app/service/user.py b/app/service/user.py index ac3fddd..d11683e 100644 --- a/app/service/user.py +++ b/app/service/user.py @@ -1,11 +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, \ @@ -13,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): @@ -58,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 = await generate_password() + app_register = AppRegisterDao(db).get_apps() for app in app_register: if app["id"] == RAGFLOW: service = RagflowService(settings.fwr_base_url) @@ -82,26 +75,28 @@ 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,sync_flag=sync_flag) - pwd = user_model.encrypted_password(app_password) + # 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 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() @@ -332,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: @@ -341,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 -- Gitblit v1.8.0