| | |
|---|
| | | from datetime import datetime |
|---|
| | | from app.api import pwd_context |
|---|
| | | from app.config.const import RAGFLOW, BISHENG, DIFY, USER_STATSU_DELETE, ROLE_STATUS_ON, DEPT_STATUS_ON, \ |
|---|
| | | Dialog_STATSU_ON |
|---|
| | | Dialog_STATSU_ON, USER_STATSU_ON |
|---|
| | | from app.models import RoleModel, GroupModel, AgentType, role_resource_table, DialogModel, OrganizationModel, \ |
|---|
| | | ResourceModel |
|---|
| | | from app.models.menu_model import WebMenuModel, MenuCapacityModel |
|---|
| | |
|---|
| | | |
|---|
| | | async def edit_user_data(db, user_id, email, phone, login_name, roles, groups): |
|---|
| | | try: |
|---|
| | | print(121221) |
|---|
| | | print(roles) |
|---|
| | | user = db.query(UserModel).filter(UserModel.id == user_id).first() |
|---|
| | | user.email = email |
|---|
| | | user.phone = phone |
|---|
| | |
|---|
| | | user.email = email |
|---|
| | | user.updated_at = datetime.now() |
|---|
| | | user.roles = [db.get(RoleModel, roleId) for roleId in roles] |
|---|
| | | # user.groups = [db.get(GroupModel, groupId) for groupId in groups] |
|---|
| | | user.groups = [db.get(GroupModel, groupId) for groupId in groups] |
|---|
| | | db.commit() |
|---|
| | | print(4343) |
|---|
| | | except Exception as e: |
|---|
| | | logger.error(e) |
|---|
| | | db.rollback() |
|---|
| | |
|---|
| | | async def edit_user_pwd(db, user_id, current_user_id, new_password="basic123456"): |
|---|
| | | try: |
|---|
| | | user = db.query(UserModel).filter(UserModel.id == user_id).first() |
|---|
| | | # pwd = user.decrypted_password() |
|---|
| | | # for app in AppRegisterDao(db).get_apps(): |
|---|
| | | # if app.get("id") == RAGFLOW: |
|---|
| | | # token = await get_new_token(db, user_id, app.get("id")) |
|---|
| | | # ragflow_service = RagflowService(settings.fwr_base_url) |
|---|
| | | # await ragflow_service.set_user_password(token, pwd, new_password) |
|---|
| | | # elif app.get("id") == BISHENG: |
|---|
| | | # token = await get_bisheng_token(db, current_user_id) |
|---|
| | | # bisheng_service = BishengService(settings.sgb_base_url) |
|---|
| | | # await bisheng_service.change_password_public(token, user.username, pwd, new_password) |
|---|
| | | # else: |
|---|
| | | # logger.error("注册未知应用：{}".format(app.get("id"))) |
|---|
| | | user.hashed_password = pwd_context.hash(new_password) |
|---|
| | | # user.password = user.encrypted_password(new_password) |
|---|
| | | db.commit() |
|---|
| | | except Exception as e: |
|---|
| | | logger.error(e) |
|---|
| | |
|---|
| | | |
|---|
| | | await role_resource(role_set, permissions, user.roles) |
|---|
| | | for ogt in user.organizations: |
|---|
| | | print(ogt.name) |
|---|
| | | if ogt.status != DEPT_STATUS_ON: |
|---|
| | | continue |
|---|
| | | print(ogt.name) |
|---|
| | | if ogt.roles: |
|---|
| | | print(9877) |
|---|
| | | await role_resource(role_set, permissions,ogt.roles) |
|---|
| | | parent_ogt = ogt.parent |
|---|
| | | while parent_ogt: |
|---|
| | |
|---|
| | | break |
|---|
| | | # print(permissions.keys()) |
|---|
| | | def get_parent(parent, permissions): |
|---|
| | | # print(parent.id) |
|---|
| | | # nonlocal permissions |
|---|
| | | if parent.id not in permissions: |
|---|
| | | |
|---|
| | | permissions[parent.id] = parent # .to_router_dict() |
|---|
| | |
|---|
| | | menu_list = db.query(WebMenuModel.id, WebMenuModel.title, WebMenuModel.describe, WebMenuModel.icon, WebMenuModel.desc,WebMenuModel.rank, |
|---|
| | | WebMenuModel.img, MenuCapacityModel.capacity_id, MenuCapacityModel.capacity_type, MenuCapacityModel.chat_id.label("agentId"), MenuCapacityModel.chat_type).outerjoin( |
|---|
| | | MenuCapacityModel, WebMenuModel.id == MenuCapacityModel.menu_id).outerjoin( |
|---|
| | | DialogModel, MenuCapacityModel.capacity_id == DialogModel.id).filter(DialogModel.status=="1").all() |
|---|
| | | DialogModel, MenuCapacityModel.capacity_id == DialogModel.id).filter(DialogModel.status==USER_STATSU_ON).all() |
|---|
| | | |
|---|
| | | for menu in menu_list: |
|---|
| | | menu_dict[menu.id] = menu_dict.get(menu.id, []) + [menu] |
|---|
| | |
|---|
| | | parent_ogt = parent_ogt.parent |
|---|
| | | else: |
|---|
| | | break |
|---|
| | | |
|---|
| | | # tmp_dit = {} |
|---|
| | | # for permission in permissions.values(): |
|---|
| | | # tmp_dit[permission["parentId"]] = tmp_dit.get(permission["parentId"], []) + [permission] |
|---|
| | | # |
|---|
| | | # def get_child(parent_id): |
|---|
| | | # res = permissions[parent_id] |
|---|
| | | # res["children"] = [get_child(i["id"]) for i in tmp_dit.get(parent_id, [])] |
|---|
| | | # return res |
|---|
| | | |
|---|
| | | # res["menus"] = [get_child(i["id"]) for i in tmp_dit.get(parent_id, [])] |
|---|
| | | res["menus"] = list(permissions.values()) |
|---|
