From c4ea3dd38939117ee9df9383bb40d865242bfd7d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=8E=8B=E6=80=9D=E5=B7=9D?= <sichanwong0527@gmail.com>
Date: Fri, 29 Apr 2022 17:04:50 +0800
Subject: [PATCH] tmp

---
 CompanyData/CompanyDataImpl.py   | 26 +++++++++++++++++++------
 CompanyData/CompanyDataRoutes.py | 33 +++++++++++++-------------------
 2 files changed, 33 insertions(+), 26 deletions(-)

diff --git a/CompanyData/CompanyDataImpl.py b/CompanyData/CompanyDataImpl.py
index ec6def0..76875c0 100644
--- a/CompanyData/CompanyDataImpl.py
+++ b/CompanyData/CompanyDataImpl.py
@@ -56,12 +56,19 @@ class CompanyDataImpl(CompanyData):
 
             self.basic_info = basic_info
 
-            self.db_tfse.upsert_single_data(
-                "企业数据",
-                "企业数据_更新汇总",
-                {"企业ID": self.cid},
-                self.dict_to_save()
-            )
+            # self.db_tfse.upsert_single_data(
+            #     "企业数据",
+            #     "企业数据_更新汇总",
+            #     {"企业ID": self.cid},
+            #     self.dict_to_save()
+            # )
+            #
+            # self.db_tfse.upsert_single_data(
+            #     "企业数据",
+            #     "基本工商信息",
+            #     {"企业ID": self.cid},
+            #     self.dict_to_save()
+            # )
 
         def prepare_share_holders():
             """"""
@@ -142,6 +149,13 @@ class CompanyDataImpl(CompanyData):
                 self.dict_to_save()
             )
 
+            self.db_tfse.upsert_single_data(
+                "企业数据",
+                "企业数据_更新汇总",
+                {"企业ID": self.cid},
+                self.dict_to_save()
+            )
+
         def prepare_main_members():
             """"""
 
diff --git a/CompanyData/CompanyDataRoutes.py b/CompanyData/CompanyDataRoutes.py
index b5c04cf..b7bafb3 100644
--- a/CompanyData/CompanyDataRoutes.py
+++ b/CompanyData/CompanyDataRoutes.py
@@ -15,26 +15,19 @@ def basic_info():
     """
     初始化企业数据
     """
-    req = request.json
-    company_data = CompanyDataImpl()
-    company_data.cid = req['cid']
-    company_data.name = req['company_name']
-    company_data.drag_data_from_tyc_api()
-    company_data.prepare_company_init_data()
-    return {"info": "企业数据初始化成功"}, 200
-    # try:
-    #     req = request.json
-    #     company_data = CompanyDataImpl()
-    #     company_data.cid = req['cid']
-    #     company_data.name = req['company_name']
-    #     company_data.drag_data_from_tyc_api()
-    #     company_data.prepare_company_init_data()
-    #     return {"info": "企业数据初始化成功"}, 200
-    # except ReturnConditionCheckFailed as e:
-    #     e.log_error()
-    #     return {"info": e.failed_info}, e.status_code
-    # except KeyError:
-    #     return {"info": "参数错误"}, 400
+    try:
+        req = request.json
+        company_data = CompanyDataImpl()
+        company_data.cid = req['cid']
+        company_data.name = req['company_name']
+        # company_data.drag_data_from_tyc_api()
+        company_data.prepare_company_init_data()
+        return {"info": "企业数据初始化成功"}, 200
+    except ReturnConditionCheckFailed as e:
+        e.log_error()
+        return {"info": e.failed_info}, e.status_code
+    except KeyError:
+        return {"info": "参数错误"}, 400
 
 
 @company_route.route('/company_index_etl', methods=['POST'])