rag-gateway/app/service/organization.py

import uuid

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



async def get_organization_list(db, deptName: str):
    query = db.query(OrganizationModel).filter(OrganizationModel.organization_id == None)
    if deptName:
        query = query.filter(OrganizationModel.name.like('%{}%'.format(deptName)))
    depts = query.order_by(OrganizationModel.seq.desc()).all()
    return {"total": query.count(), "rows":  [dept.to_json() for dept in depts]}


async def create_dept(db, dept_name, leader, phone, address, status, order_num, roles, groups,parent_id):
    try:
        dept_model = OrganizationModel(id=str(uuid.uuid4()),name=dept_name, address=address,leader=leader,phone=phone,seq=order_num,status=status)
        if parent_id:
            dept_model.parent = db.get(OrganizationModel, parent_id)
        if roles:
            dept_model.roles = [db.get(RoleModel, roleId) for roleId in roles]
        # if groups:
        #     dept_model.groups = [db.get(GroupModel, groupId) for groupId in groups]
        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_dept_data(db, dept_id, dept_name, leader, phone, address, status, order_num, roles, groups,parent_id):
    try:
        dept_model = db.query(OrganizationModel).filter(OrganizationModel.id == dept_id).first()
        dept_model.name = dept_name
        dept_model.address = address
        dept_model.phone = phone
        dept_model.leader = leader
        dept_model.status = status
        dept_model.seq = order_num
        # if parent_id:
        #     dept_model.parent = db.get(OrganizationModel, parent_id)
        if roles:
            dept_model.roles = [db.get(RoleModel, roleId) for roleId in roles]
        # if groups:
        #     dept_model.groups = [db.get(GroupModel, groupId) for groupId in groups]
        # 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_dept_parent(db, dept_id, parent_id, order_num):
    try:
        dept_model = db.query(OrganizationModel).filter(OrganizationModel.id == dept_id).first()
        dept_model.parent = db.get(OrganizationModel, parent_id)
        dept_model.seq = order_num
        db.commit()
        db.refresh(dept_model)
    except Exception as e:
        logger.error(e)
        db.rollback()
        return False
    return True


async def get_organization_info(db, dept_id: str):
    dept = db.query(OrganizationModel).filter(OrganizationModel.id.__eq__(dept_id)).first()

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


async def delete_organization_info(db, dept_id: str):
    try:
        db.query(OrganizationModel).filter(OrganizationModel.id.__eq__(dept_id)).delete()
        db.commit()
    except Exception as e:
        logger.error(e)
        db.rollback()
        return False
    return True