from flask import Blueprint, render_template, request, jsonify
from marshmallow import INCLUDE
from applications.models.admin_dept import DeptSchema
from applications.service.admin import dept_curd as dept_curd
from applications.service.common.response import success_api, fail_api
from applications.service.common.validate import check_data
from applications.service.route_auth import authorize

admin_dept = Blueprint('adminDept', __name__, url_prefix='/admin/dept')


@admin_dept.route('/')
@authorize("admin:dept:main", log=True)
def main():
    return render_template('admin/dept/main.html')


@admin_dept.route('/data')
@authorize("admin:dept:main", log=True)
def data():
    power_data = dept_curd.get_dept_dict()
    res = {
        "data": power_data
    }
    return jsonify(res)


@admin_dept.route('/add')
@authorize("admin:dept:add", log=True)
def add():
    return render_template('admin/dept/add.html')


@admin_dept.route('/tree')
@authorize("admin:dept:main", log=True)
def tree():
    power_data = dept_curd.get_dept_dict()
    res = {
        "status": {"code": 200, "message": "默认"},
        "data": power_data

    }
    return jsonify(res)


@admin_dept.route('/save', methods=['POST'])
@authorize("admin:dept:add", log=True)
def save():
    req = request.json
    check_data(DeptSchema(unknown=INCLUDE), req)
    dept_curd.save_dept(req)
    return success_api(msg="成功")


@admin_dept.route('/edit', methods=['GET', 'POST'])
@authorize("admin:dept:edit", log=True)
def edit():
    id = request.args.get("deptId")
    dept = dept_curd.get_dept_by_id(id)
    return render_template('admin/dept/edit.html', dept=dept)


# 启用
@admin_dept.route('/enable', methods=['PUT'])
@authorize("admin:dept:edit", log=True)
def enable():
    id = request.json.get('deptId')
    if id:
        res = dept_curd.enable_status(id)
        if not res:
            return fail_api(msg="出错啦")
        return success_api(msg="启用成功")
    return fail_api(msg="数据错误")


# 禁用
@admin_dept.route('/disable', methods=['PUT'])
@authorize("admin:dept:edit", log=True)
def disenable():
    id = request.json.get('deptId')
    if id:
        res = dept_curd.disable_status(id)
        if not res:
            return fail_api(msg="出错啦")
        return success_api(msg="禁用成功")
    return fail_api(msg="数据错误")


@admin_dept.route('/update', methods=['PUT'])
@authorize("admin:dept:edit", log=True)
def update():
    req = request.json
    check_data(DeptSchema(unknown=INCLUDE), req)
    res = dept_curd.update_dept(req)
    if not res:
        return fail_api(msg="更新失败")
    return success_api(msg="更新成功")


@admin_dept.route('/remove/<int:id>', methods=['DELETE'])
@authorize("admin:dept:remove", log=True)
def remove(id):
    res = dept_curd.remove_dept(id)
    if res:
        return success_api(msg="删除成功")
    else:
        return fail_api(msg="删除失败")