from flask import render_template, request, jsonify, make_response
from flask_restful import Resource, reqparse

from . import rights_bp, rights_api
from ._utils import select_power_dict, get_power_dict, batch_remove_power
from ...common.utils.http import success_api, fail_api
from ...common.utils.rights import authorize
from ...extensions import db
from ...models import Power, Role

parser_power = reqparse.RequestParser(bundle_errors=True)
parser_power.add_argument('icon', type=str)
parser_power.add_argument('openType', type=str, dest='open_type')
parser_power.add_argument('parentId', type=str, dest='parent_id')
parser_power.add_argument('powerCode', type=str, dest='power_code')
parser_power.add_argument('powerName', type=str, dest='power_name')
parser_power.add_argument('powerType', type=str, dest='power_type')
parser_power.add_argument('powerUrl', type=str, dest='power_url')
parser_power.add_argument('sort', type=int, dest='sort')


@rights_bp.get('/')
@authorize("admin:power:main", log=True)
def index():
    return render_template('admin/rights/main.html')


@rights_bp.get('/data')
@authorize("admin:power:main", log=True)
def data():
    power_data = get_power_dict()

    res = {
        "data": power_data
    }
    return jsonify(res)


@rights_bp.get('/selectParent')
@authorize("admin:power:main", log=True)
def select_parent():
    """获取选择父节点"""
    power_data = select_power_dict()
    res = {
        "status": {"code": 200, "message": "默认"},
        "data": power_data

    }
    return jsonify(res)


# 启用权限
@rights_bp.put('/enable')
@authorize("admin:power:edit", log=True)
def enable():
    parser = reqparse.RequestParser(bundle_errors=True)
    parser.add_argument('powerId', dest='power_id', type=int, required=True)
    parser.add_argument('operate', type=int, required=True, choices=[0, 1])

    res = parser.parse_args()

    power = Power.query.filter_by(id=res.power_id).update({"enable": res.operate})
    if power:
        db.session.commit()
        message = "启用成功" if res.operate else '禁用成功'
        return success_api(msg=message)
    else:
        return fail_api(msg="出错啦")


# 批量删除
@rights_bp.delete('/batchRemove')
@authorize("admin:power:remove", log=True)
def batch_remove_view():
    ids = request.form.getlist('ids[]')
    batch_remove_power(ids)
    return success_api(msg="批量删除成功")


@rights_api.resource('/power')
class AddRight(Resource):
    @authorize("admin:power:add", log=True)
    def get(self):
        """获取增加视图"""
        return make_response(render_template('admin/rights/add.html'))

    @authorize("admin:power:add", log=True)
    def post(self):
        res = parser_power.parse_args()
        power = Power(
            icon=res.icon,
            open_type=res.open_type,
            parent_id=res.parent_id,
            code=res.power_code,
            name=res.power_name,
            type=res.power_type,
            url=res.power_url,
            sort=res.sort,
            enable=1
        )
        try:
            db.session.add(power)
            db.session.commit()
        except Exception as e:
            print(e)
            return fail_api(msg='数据提交失败')

        return success_api(msg="成功")


@rights_api.resource('/power/<int:power_id>')
class RightsURD(Resource):
    @authorize("admin:power:edit", log=True)
    def get(self, power_id):
        power = Power.query.filter_by(id=power_id).first()
        icon = str(power.icon).split()
        if len(icon) == 2:
            icon = icon[1]
        else:
            icon = None
        return make_response(render_template('admin/rights/edit.html', power=power, icon=icon))

    @authorize("admin:power:remove", log=True)
    def delete(self, power_id):
        # 删除权限（目前没有判断父节点自动删除子节点）
        power = Power.query.filter_by(id=power_id).first()
        role_id_list = []
        roles = power.role
        for role in roles:
            role_id_list.append(role.id)
        roles = Role.query.filter(Role.id.in_(role_id_list)).all()
        for p in roles:
            power.role.remove(p)
        r = Power.query.filter_by(id=power_id).delete()
        db.session.commit()

        if r:
            return success_api(msg="删除成功")
        else:
            return fail_api(msg="删除失败")

    @authorize("admin:power:edit", log=True)
    def put(self, power_id):

        res = parser_power.parse_args()
        data = {
            "icon": res.icon,
            "open_type": res.open_type,
            "parent_id": res.parent_id,
            "code": res.power_code,
            "name": res.power_name,
            "type": res.power_type,
            "url": res.power_url,
            "sort": res.sort
        }
        power = Power.query.filter_by(id=power_id).update(data)
        db.session.commit()

        if not power:
            return fail_api(msg="更新权限失败")
        return success_api(msg="更新权限成功")