增加功能

zhaoqingang

2024-11-13 de45d8dc19cd16b85344a3c3bbb243a12e8811a0

 app/service/resource.py

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

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

from Log import logger
from app.models import ResourceModel


async def get_resource_tree(db):
@@ -26,4 +24,42 @@
        logger.error(e)
        db.rollback()
        return False
    return True


async def create_menu(db, menu_name, component, path, order_num, perms, menu_type, description, parent_id, status, icon):
    try:
        dept_model = ResourceModel(id=str(uuid.uuid4()),name=menu_name, url=component,path=path,perms=perms,description=description,status=status, icon=icon,seq=order_num, resource_type_id=menu_type)
        if parent_id:
            dept_model.parent = db.get(ResourceModel, parent_id)
        db.add(dept_model)
        db.commit()
        db.refresh(dept_model)
    except Exception as e:
        logger.error(e)
        db.rollback()
        return False
    return True


async def edit_resource_data(db, menu_id, menu_name, component, path, order_num, perms, menu_type, description, parent_id, status, icon):
    try:
        dept_model = db.query(ResourceModel).filter(ResourceModel.id == menu_id).first()
        dept_model.name = menu_name
        dept_model.url = component
        dept_model.path = path
        dept_model.perms = perms
        dept_model.status = status
        dept_model.icon = icon
        dept_model.seq = order_num
        dept_model.description = description
        dept_model.resource_type_id = menu_type
        if parent_id:
            dept_model.parent = db.get(ResourceModel, parent_id)
        db.commit()
        db.refresh(dept_model)
    except Exception as e:
        logger.error(e)
        db.rollback()
        return False
    return True