57 lines
1.4 KiB
Python
57 lines
1.4 KiB
Python
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_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
|