dify 文档智能

zhaoqingang

2024-12-17 e6c5e89f09637b8d9ebca6895d781663f12646d6

 app/service/role.py

@@ -1,6 +1,7 @@
import uuid

from Log import logger
from app.models import OrganizationModel
from app.models.resource_model import ResourceModel
from app.models.role_model import RoleModel

@@ -13,7 +14,7 @@
    if keyword:
        query = query.filter(RoleModel.name.like('%{}%'.format(keyword)))
    total = query.count()
    roles = query.order_by(RoleModel.id.desc()).limit(page_size).offset(
    roles = query.order_by(RoleModel.created_at.desc()).limit(page_size).offset(
        (page_index - 1) * page_size).all()
    return {"total": total, "rows":  [role.to_json() for role in roles]}

@@ -54,7 +55,7 @@
    return True


async def edit_role_resource(db, role_id:str, role_name:str, description:str,role_key:str, data_scope:int, resources:list, edit_type:int):
async def edit_role_resource(db, role_id:str, role_name:str, description:str,role_key:str, data_scope:int, resources:list, edit_type:int, depts:list):
    try:
        role = db.query(RoleModel).filter(RoleModel.id == role_id).first()
        if edit_type == 1:
@@ -65,8 +66,10 @@
                role.roleKey = role_key
            if data_scope:
                role.dataScope = data_scope
        else:
        elif edit_type == 2:
            role.resources = [db.get(ResourceModel, resourcesId) for resourcesId in resources]
        elif edit_type == 3:
            role.organizations = [db.get(OrganizationModel, deptId) for deptId in depts]
        db.add(role)
        db.commit()
    except Exception as e: