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):
    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": False,
            "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)

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


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()