from flask import Blueprint, request

from common.security.APIAuth import api_secret
from user.User import UserManage

user_route = Blueprint('user', __name__)


@user_route.route('/create_user', methods=['POST'])
@api_secret
def create_user_route():
    """
    新建用户
    """
    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