from flask import request

from flask_restful import Resource, reqparse
from flask_login import current_user
from sqlalchemy import desc

from applications.extensions import db
from applications.common.utils.http import fail_api, success_api, table_api
from applications.common.utils.rights import authorize
from applications.models import User, Role, Dept
from applications.models import AdminLog
from . import users_api


def get_current_user_logs():
    """ 获取当前用户日志 """
    log = AdminLog.query.filter_by(url='/passport/login').filter_by(uid=current_user.id).order_by(
        desc(AdminLog.create_time)).limit(10)
    return log


def is_user_exists(username):
    """ 判断用户是否存在 """
    res = User.query.filter_by(username=username).count()
    return bool(res)


def delete_by_id(_id):
    """ 删除用户 """
    user = User.query.filter_by(id=_id).first()
    roles_id = []
    for role in user.role:
        roles_id.append(role.id)
    roles = Role.query.filter(Role.id.in_(roles_id)).all()
    for r in roles:
        user.role.remove(r)
    res = User.query.filter_by(id=_id).delete()
    db.session.commit()
    return res


def batch_remove(ids):
    """ 批量删除 """
    for _id in ids:
        delete_by_id(_id)


def update_user_role(_id, roles_list):
    user = User.query.filter_by(id=_id).first()
    roles_id = []
    for role in user.role:
        roles_id.append(role.id)
    roles = Role.query.filter(Role.id.in_(roles_id)).all()
    for r in roles:
        user.role.remove(r)
    roles = Role.query.filter(Role.id.in_(roles_list)).all()
    for r in roles:
        user.role.append(r)
    db.session.commit()


@users_api.resource('/add')
class Users(Resource):
    """用户列表数据操作"""

    # @authorize("admin:user:add", log=True)
    # def get(self):
    #     roles = Role.query.all()
    #     return make_response(render_template('users/add.html', roles=roles))

    @authorize("admin:user:add", log=True)
    def post(self):
        """新建单个用户"""
        parser = reqparse.RequestParser()
        parser.add_argument("roleIds", type=str, dest='role_ids')
        parser.add_argument("username", type=str, required=True, help="用户名不能为空")
        parser.add_argument("realName", type=str, required=True, help="真实姓名不能为空", dest='real_name')
        parser.add_argument("password", type=str, required=True, help="密码不得为空")

        res = parser.parse_args()

        role_ids = res.role_ids.split(',')

        if is_user_exists(res.username):
            return fail_api(msg="用户已经存在")

        user = User()
        user.username = res.username
        user.realname = res.real_name
        user.set_password(res.password)
        db.session.add(user)
        db.session.commit()

        """ 增加用户角色 """
        user = User.query.filter_by(id=user.id).first()
        roles = Role.query.filter(Role.id.in_(role_ids)).all()
        for r in roles:
            user.role.append(r)
        db.session.commit()

        return success_api(msg="增加成功", code=0)


@users_api.resource('/<user_id>')
class CURDUser(Resource):
    """修改用户数据"""

    # @authorize("admin:user:edit", log=True)
    # def get(self, user_id):
    #     #  获取编辑用户信息
    #     user = User.query.filter_by(id=user_id).first()
    #     roles = Role.query.all()
    #     checked_roles = []
    #     for r in user.role:
    #         checked_roles.append(r.id)
    #     return make_response(
    #         render_template('users/edit_users.html', user=user, roles=roles, checked_roles=checked_roles))

    @authorize("admin:user:remove", log=True)
    def delete(self, user_id):
        # 删除用户
        res = delete_by_id(user_id)
        if not res:
            return fail_api(msg="删除失败")
        return success_api(msg="删除成功")

    @authorize("admin:user:edit", log=True)
    def put(self, user_id):
        parser = reqparse.RequestParser()
        parser.add_argument('roleIds', type=str, dest='role_ids')
        parser.add_argument('userId', type=str, dest='user_id')
        parser.add_argument('username', type=str)
        parser.add_argument('realName', type=str, dest='real_name')
        parser.add_argument('deptId', type=str, dest='dept_id')

        res = parser.parse_args()
        role_ids = res.role_ids.split(',')

        # 更新用户数据
        User.query.filter_by(id=user_id).update({'username': res.username,
                                                 'realname': res.real_name,
                                                 'dept_id': res.dept_id})
        db.session.commit()

        update_user_role(user_id, role_ids)

        return success_api(msg="更新成功")


@users_api.resource('/batchRemove')
class UserBatchRemove(Resource):
    """批量删除"""

    @authorize("admin:user:remove", log=True)
    def delete(self):
        ids = request.form.getlist('ids[]')
        batch_remove(ids)
        return success_api(msg="批量删除成功")


@users_api.resource('/enable')
class UserEnable(Resource):
    def put(self):
        # 启用或者禁用用户 enable disable

        parser = reqparse.RequestParser()
        parser.add_argument('userId', type=int, required=True, dest='user_id')
        parser.add_argument('operate', type=int, required=True, dest='operate', choices=[0, 1])

        res = parser.parse_args()

        if res.operate == 1:
            user = User.query.filter_by(id=res.user_id).update({"enable": res.operate})
            message = success_api(msg="启动成功")
        else:
            user = User.query.filter_by(id=res.user_id).update({"enable": res.operate})
            message = success_api(msg="禁用成功")
        if user:
            db.session.commit()
        else:
            return fail_api(msg="出错啦")
        return message


@users_api.resource('/data')
class UsersData(Resource):
    @authorize("admin:user:main", log=True)
    def get(self):
        parser = reqparse.RequestParser()
        parser.add_argument('page', type=int, default=1)
        parser.add_argument('limit', type=int, default=10)
        parser.add_argument('realName', type=str, dest='real_name')
        parser.add_argument('username', type=str)
        parser.add_argument('deptId', type=int, dest='dept_id', default=0)

        res = parser.parse_args()

        filters = []

        if res.real_name:
            filters.append(User.realname.like('%' + res.real_name + '%'))
        if res.username:
            filters.append(User.username.like('%' + res.username + '%'))
        if res.dept_id:
            filters.append(User.dept_id == res.dept_id)

        paginate = User.query.filter(*filters).paginate(page=res.page,
                                                        per_page=res.limit,
                                                        error_out=False)

        dept_name = lambda dept_id: Dept.query.filter_by(id=dept_id).first().dept_name if dept_id else ""
        user_data = [{
            'id': item.id,
            'username': item.username,
            'realname': item.realname,
            'enable': item.enable,
            'create_at': item.create_at,
            'update_at': item.update_at,
            'dept': dept_name(item.dept_id),
        } for item in paginate.items]

        return table_api(data=user_data, count=paginate.total, code=0)