首页智能体中心过滤掉自研类型

zhaoqingang

2025-04-11 e078028f1a8da34f3cc2cb5095b8e103a996f553

 app/service/resource.py

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

from Log import logger
from app.config.const import RESOURCE_STATUS_ON, RESOURCE_STATUS_OFF, RESOURCE_STATUS_DELETE
from app.models import ResourceModel


@@ -17,14 +18,28 @@


async def delete_resource_info(db, menu_id):
    delete_list = []
    dept_list = []
    next_dept_list = []
    base_dept = db.query(ResourceModel).filter(ResourceModel.id.__eq__(menu_id)).first()
    dept_list.append(base_dept)
    while dept_list:
        for dept in dept_list:
            delete_list.append(dept.id)
            for child_dept in dept.children:
                next_dept_list.append(child_dept)

        dept_list = next_dept_list
        next_dept_list = []

    try:
        db.query(ResourceModel).filter(ResourceModel.id.__eq__(menu_id)).delete()
        db.query(ResourceModel).filter(ResourceModel.id.in_(delete_list)).update({"status": RESOURCE_STATUS_DELETE})
        db.commit()
    except Exception as e:
        logger.error(e)
        db.rollback()
        return False
    return True
        return '服务异常 ！'
    return ""


async def create_menu(db, menu_name, component, path, order_num, perms, menu_type, description, parent_id, status, icon):
@@ -49,7 +64,7 @@
        dept_model.url = component
        dept_model.path = path
        dept_model.perms = perms
        dept_model.status = status
        # dept_model.status = status
        dept_model.icon = icon
        dept_model.seq = order_num
        dept_model.description = description
@@ -62,4 +77,40 @@
        logger.error(e)
        db.rollback()
        return False
    return True
    return True


async def edit_resource_status(db, menu_id, status):
    delete_list = []
    dept_list = []
    next_dept_list = []
    base_dept = db.query(ResourceModel).filter(ResourceModel.id.__eq__(menu_id)).first()
    if status == RESOURCE_STATUS_ON:
        if base_dept.parent and base_dept.parent.status != RESOURCE_STATUS_ON:
            return '上级节点状态异常！'
        try:
            base_dept.status = status
            db.commit()
        except Exception as e:
            logger.error(e)
            db.rollback()
            return '服务异常 ！'
    else:
        dept_list.append(base_dept)
        while dept_list:
            for dept in dept_list:
                delete_list.append(dept.id)
                for child_dept in dept.children:
                    next_dept_list.append(child_dept)

            dept_list = next_dept_list
            next_dept_list = []
        try:
            db.query(ResourceModel).filter(ResourceModel.id.in_(delete_list)).update(
                {"status": RESOURCE_STATUS_OFF})
            db.commit()
        except Exception as e:
            logger.error(e)
            db.rollback()
            return '服务异常 ！'
    return ""