46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
|
from flask import Blueprint, request, Response
|
||
|
|
||
|
from CompanyUser.CompanyUser_impl import CompanyUserImpl
|
||
|
from Utils.ErrorUtil import ReturnConditionCheckFailed
|
||
|
from common.rsa import decrypt_data
|
||
|
from user.user_auth import check_block
|
||
|
|
||
|
company_user_route = Blueprint('user', __name__)
|
||
|
|
||
|
|
||
|
@company_user_route.route('/login', methods=['POST'])
|
||
|
@check_block
|
||
|
def login():
|
||
|
"""企业用户登录"""
|
||
|
try:
|
||
|
company_user = CompanyUserImpl()
|
||
|
company_user.email = request.json['email']
|
||
|
|
||
|
try:
|
||
|
company_user.pwd = decrypt_data(encrypt_msg=request.json['pwd'])
|
||
|
except Exception:
|
||
|
return {"info": "登录密码错误"}, 200
|
||
|
|
||
|
result = company_user.login()
|
||
|
return {"info": "登录成功", "result": result}, 200
|
||
|
except ReturnConditionCheckFailed as e:
|
||
|
return {"info": e.failed_info}, e.status_code
|
||
|
except KeyError:
|
||
|
return {"info": "参数异常"}, 400
|
||
|
|
||
|
|
||
|
@company_user_route.route('/register', methods=['POST'])
|
||
|
@check_block
|
||
|
def register():
|
||
|
"""企业用户注册"""
|
||
|
try:
|
||
|
req = request.json
|
||
|
company_user = CompanyUserImpl()
|
||
|
company_user.email = req['email']
|
||
|
company_user.pwd = req['pwd']
|
||
|
company_user.register(req['code'])
|
||
|
response = {"info": "注册成功"}, 200
|
||
|
return response
|
||
|
except ReturnConditionCheckFailed as e:
|
||
|
return {"info": e.failed_info}, e.status_code
|