From 34a567a6b32d177eece19336691b4ca592e96986 Mon Sep 17 00:00:00 2001
From: P3ngSaM <61768364+P3ngSaM@users.noreply.github.com>
Date: Mon, 25 Apr 2022 17:04:17 +0800
Subject: [PATCH] =?UTF-8?q?update=20testforadmin=E5=8C=85?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 TestForAdmin/route.py | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/TestForAdmin/route.py b/TestForAdmin/route.py
index 71e4bb6..64eee7f 100644
--- a/TestForAdmin/route.py
+++ b/TestForAdmin/route.py
@@ -1,6 +1,6 @@
 from flask import Blueprint, request, Response
 
-from CompanyUser.CompanyUserImpl import CompanyVerifyThreeFactorsImpl
+from CompanyUser.CompanyUserImpl import CompanyVerifyThreeFactorsImpl, CompanyUserImpl
 from Rating.Credit.CreditImpl import CreditRating
 from Rating.Esg.EsgImpl import EsgRating
 from TestForAdmin.APIAuth import api_secret
@@ -47,3 +47,22 @@ def general_rating():
         return {"info": e.failed_info}, e.status_code
     except KeyError:
         return {"info": "参数异常"}, 400
+
+
+@test_admin_route.route('/register', methods=['POST'])
+@api_secret
+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:
+        e.log_error()
+        return {"info": e.failed_info}, e.status_code
+    except KeyError:
+        return {"info": "参数异常"}, 400
\ No newline at end of file