From cc9af9326aacc478039089b7977e62219ce241c5 Mon Sep 17 00:00:00 2001
From: P3ngSaM <61768364+P3ngSaM@users.noreply.github.com>
Date: Wed, 8 Jun 2022 14:20:26 +0800
Subject: [PATCH] =?UTF-8?q?update=20=E4=BF=9D=E5=AD=98=E4=BC=81=E4=B8=9A?=
 =?UTF-8?q?=E4=B8=BB=E8=A6=81=E4=BA=BA=E5=91=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 routes.py | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/routes.py b/routes.py
index 4221970..f15b8b0 100644
--- a/routes.py
+++ b/routes.py
@@ -3,7 +3,7 @@ import json
 from flask import Blueprint, request
 
 from common import verify_token
-from tyc import save_full_data, company_baseinfo, corporate_shareholders
+from tyc import save_full_data, company_baseinfo, corporate_shareholders, principal_officer
 
 tyc = Blueprint('tyc', __name__)
 
@@ -38,6 +38,18 @@ def upload_shareholders_data():
     """保存企业股东"""
     name = json.loads(request.data)['企业名称']
     res = corporate_shareholders(name)
+    if res == '保存成功':
+        return {"info": res}, 200
+    else:
+        return {"info": res}, 210
+
+
+@tyc.route('/main_staff', methods=['POST'])
+@verify_token
+def upload_shareholders_data():
+    """保存主要人员"""
+    name = json.loads(request.data)['企业名称']
+    res = principal_officer(name)
     if res == '保存成功':
         return {"info": res}, 200
     else: