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.dept_id == user_id)
    return query.first()

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

    # 排除某些部门
    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()