from flask import Blueprint, request

from common.security.APIAuth import api_secret, api_verification_code, verify_token
from user.User import UserManage, UserOperation

user_route = Blueprint('user', __name__)


@user_route.route('/create_user', methods=['POST'])
@verify_token
def create_user_route(**kwargs):
    """
    新建用户
    """
    um = UserManage()
    um.email = request.json['email']
    um.name = request.json['name']
    um.pwd = request.json['pwd']
    um.role = request.json['role']
    um.duty = request.json['duty']
    res = um.create_user()
    if res is True:
        return {"info": "新建用户成功"}, 200
    else:
        return {"info": res}, 200


@user_route.route('/send_vcode_to_user', methods=['POST'])
@api_secret
def send_vcode_to_user_route():
    """
    发送验证码
    """
    um = UserManage()
    um.email = request.json['email']
    res = um.send_vcode_to_user()
    if res is True:
        return {"info": "验证邮件已发送"}, 200
    else:
        return {"info": res}, 200


@user_route.route('/login', methods=['POST'])
@api_secret
@api_verification_code
def login_route():
    """
    登录
    """
    uo = UserOperation()
    uo.email = request.json['email']
    uo.pwd = request.json['pwd']
    res = uo.login()
    if type(res) == dict:
        return {"info": "登录成功", "result": res}, 200
    else:
        return {"info": res}, 200