diff --git a/TestForAdmin/route.py b/TestForAdmin/route.py index 71e4bb6..b5575d3 100644 --- a/TestForAdmin/route.py +++ b/TestForAdmin/route.py @@ -1,6 +1,7 @@ from flask import Blueprint, request, Response +from werkzeug.security import generate_password_hash -from CompanyUser.CompanyUserImpl import CompanyVerifyThreeFactorsImpl +from CompanyUser.CompanyUserImpl import CompanyVerifyThreeFactorsImpl, CompanyUserImpl, EmailVerifyCodeRecordImpl from Rating.Credit.CreditImpl import CreditRating from Rating.Esg.EsgImpl import EsgRating from TestForAdmin.APIAuth import api_secret @@ -47,3 +48,51 @@ def general_rating(): return {"info": e.failed_info}, e.status_code except KeyError: return {"info": "参数异常"}, 400 + + +@test_admin_route.route('/verify_email', methods=['POST']) +@api_secret +def verify_email(): + """ + 新注册企业 + 1.调用发送注册邮箱,获取验证码 + 2.使用邮箱、密码、验证码进行注册 + 3.使用认证接口进行认证、初始化企业数据 + """ + try: + req = request.json + # 发送验证码 + email_verify = EmailVerifyCodeRecordImpl() + email_verify.email = req['email'] + email_verify.v_type = 'register' + email_verify.send_email() + return {"info": "邮件已发送"}, 200 + except ReturnConditionCheckFailed as e: + e.log_error() + return {"info": e.failed_info}, e.status_code + except KeyError: + return {"info": "参数异常"}, 400 + + +@test_admin_route.route('/register', methods=['POST']) +@api_secret +def register(): + """ + 新注册企业 + 1.调用发送注册邮箱,获取验证码 + 2.使用邮箱、密码、验证码进行注册 + 3.使用认证接口进行认证、初始化企业数据 + """ + try: + req = request.json + # 注册 + company_user = CompanyUserImpl() + company_user.email = req['email'] + company_user.pwd = generate_password_hash("Fecr1988.") + company_user.register(req['code']) + return {"info": "注册成功"}, 200 + except ReturnConditionCheckFailed as e: + e.log_error() + return {"info": e.failed_info}, e.status_code + except KeyError: + return {"info": "参数异常"}, 400