python-fastapi-mm-zhcs-yj-api/utils/ry_system_util.py

from models import *

def dept_id_get_dept_info(db,dept_id):
    query = db.query(SysDept)
    query = query.filter(SysDept.del_flag != '2')
    query = query.filter(SysDept.dept_id == dept_id)
    return query.first()

def user_id_get_user_info(db,user_id):
    query = db.query(SysUser)
    query = query.filter(SysUser.del_flag != '2')
    query = query.filter(SysUser.user_id == user_id)
    return query.first()

def user_id_get_user_roleIds(db,user_id):
    query = db.query(SysUserRole)
    # query = query.filter(SysUserRole.del_flag != '2')
    query = query.filter(SysUserRole.user_id == user_id)
    resutl = query.all()
    return [i.role_id for i in resutl]

def user_id_get_user_videoIds(db,user_id):
    query = db.query(SysUserVideo)
    # query = query.filter(SysUserRole.del_flag != '2')
    query = query.filter(SysUserVideo.user_id == user_id)
    resutl = query.all()
    return resutl

def role_list_to_dict(roles,role_id_list):
    return [
        {
            "roleId": role.role_id,
            "roleName": role.role_name,
            "roleKey": role.role_key,
            "roleSort": role.role_sort,
            "dataScope": role.data_scope,
            "menuCheckStrictly": role.menu_check_strictly,
            "deptCheckStrictly": role.dept_check_strictly,
            "status": role.status,
            "remark": role.remark,
            "createTime": role.create_time.strftime('%Y-%m-%d %H:%M:%S') if role.create_time else '',
            "flag": role.role_id in role_id_list,
            "superAdmin": False
        }
        for role in roles
    ]

def role_id_list_get_user_role(db,role_id_list):
    query = db.query(SysRole)
    # query = query.filter(SysUserRole.del_flag != '2')
    query = query.filter(SysRole.role_id.in_(role_id_list))
    resutl = query.all()
    return role_list_to_dict(resutl,role_id_list)

def get_role(db,role_id_list=[]):
    query = db.query(SysRole)
    query = query.filter(SysRole.del_flag != '2')
    return role_list_to_dict(query.all(),role_id_list)


def parent_id_get_exclude_dept_info(db,dept_id):
    query = db.query(SysDept)
    query = query.filter(SysDept.del_flag != '2')

    # 排除某些部门
    if dept_id:
        query = query.filter(SysDept.dept_id != dept_id)
    return query.all()

def parent_id_get_dept_info(db,parent_id):
    query = db.query(SysDept)
    query = query.filter(SysDept.del_flag != '2')
    query = query.filter(SysDept.parent_id == parent_id)
    return query.all()

def parent_id_get_menu_info(db,parent_id):
    query = db.query(SysMenu)
    query = query.filter(SysMenu.del_flag != '2')
    query = query.filter(SysMenu.parent_id == parent_id)
    return query.all()

def role_id_get_role_menus(db,role_id):
    query = db.query(SysRoleMenu)
    # query = query.filter(SysUserRole.del_flag != '2')
    query = query.filter(SysRoleMenu.role_id == role_id)
    resutl = query.all()
    return [i.menu_id for i in resutl]

def role_id_get_role_depts(db,role_id):
    query = db.query(SysRoleDept)
    # query = query.filter(SysUserRole.del_flag != '2')
    query = query.filter(SysRoleDept.role_id == role_id)
    resutl = query.all()
    return [i.dept_id for i in resutl]