from applications.extensions import db
from applications.models import Dept, DeptSchema
from applications.models import User
from applications.common.curd import model_to_dicts


def get_dept_dict():
    dept = Dept.query.order_by(Dept.sort).all()
    res = model_to_dicts(Schema=DeptSchema, model=dept)
    return res


def save_dept(req):
    address = req.get("address")
    deptName = req.get("deptName")
    email = req.get("email")
    leader = req.get("leader")
    parentId = req.get("parentId")
    phone = req.get("phone")
    sort = req.get("sort")
    status = req.get("status")
    dept = Dept(
        parent_id=parentId,
        dept_name=deptName,
        sort=sort,
        leader=leader,
        phone=phone,
        email=email,
        status=status,
        address=address
    )
    r = db.session.add(dept)
    db.session.commit()
    return r


def get_dept_by_id(_id):
    """根据 id 获取部门"""
    d = Dept.query.filter_by(id=_id).first()
    return d


def enable_status(_id):
    """ 启用权限 """
    enable = 1
    d = Dept.query.filter_by(id=_id).update({"status": enable})
    if d:
        db.session.commit()
        return True
    return False


def disable_status(_id):
    """ 停用权限 """
    enable = 0
    d = Dept.query.filter_by(id=_id).update({"status": enable})
    if d:
        db.session.commit()
        return True
    return False


def update_dept(json):
    """ 更新部门信息 """
    _id = json.get("deptId"),
    data = {
        "dept_name": json.get("deptName"),
        "sort": json.get("sort"),
        "leader": json.get("leader"),
        "phone": json.get("phone"),
        "email": json.get("email"),
        "status": json.get("status"),
        "address": json.get("address")
    }
    d = Dept.query.filter_by(id=_id).update(data)
    if not d:
        return False
    db.session.commit()
    return True


def remove_dept(_id):
    d = Dept.query.filter_by(id=_id).delete()
    if not d:
        return False
    User.query.filter_by(dept_id=_id).update({"dept_id": None})
    db.session.commit()
    return True