rag-gateway/app/service/resource.py

import uuid

from app.Log import logger
from app.models import OrganizationModel, GroupModel, ResourceModel
from app.models.role_model import RoleModel



async def get_resource_tree(db):
    resource = db.query(ResourceModel).filter(ResourceModel.resource_id == None).order_by(ResourceModel.seq.desc()).all()

    return {"total": 0, "rows":  [dept.to_tree_select_json() for dept in resource]}


async def get_resource_info(db, menu_id: str):
    dept = db.query(ResourceModel).filter(ResourceModel.id.__eq__(menu_id)).first()

    return {"total": 0, "data":  dept.to_json()}


async def delete_resource_info(db, menu_id):
    try:
        db.query(ResourceModel).filter(ResourceModel.id.__eq__(menu_id)).delete()
        db.commit()
    except Exception as e:
        logger.error(e)
        db.rollback()
        return False
    return True