From f72f322ed4ab31852969f261823fbeb7f6fd5e3f Mon Sep 17 00:00:00 2001
From: P3ngSaM <61768364+P3ngSaM@users.noreply.github.com>
Date: Tue, 15 Mar 2022 10:56:16 +0800
Subject: [PATCH] =?UTF-8?q?update=20=E6=A0=B9=E6=8D=AE=E4=BC=81=E4=B8=9A?=
 =?UTF-8?q?=E7=B1=BB=E5=9E=8B=EF=BC=8C=E5=AD=98=E5=82=A8=E4=BC=81=E4=B8=9A?=
 =?UTF-8?q?=E7=9B=B8=E5=85=B3=E8=82=A1=E4=B8=9C=E7=9A=84=E8=82=A1=E4=B8=9C?=
 =?UTF-8?q?=E4=BF=A1=E6=81=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tyc.py | 54 ++++++++++++++++++++++++++++++++++++------------------
 1 file changed, 36 insertions(+), 18 deletions(-)

diff --git a/tyc.py b/tyc.py
index 037a3ed..fe85b3a 100644
--- a/tyc.py
+++ b/tyc.py
@@ -309,23 +309,41 @@ def holding_enterprise(h_id, h_name):
 
 # 保存股东数据
 def shareholder_information(name):
-    # 判断企业大股东类型
-    result = find_db_data(db='公司背景', col='企业股东', query={'企业名称': name})
-    if len(result) > 0:
-        if result[0]['企业股东']['result'][0]['type'] == 1:
-            # 1-公司
-            company_name = result[0]['企业股东']['result'][0]['name']
-            # 股东为公司，需要再次调用基本信息/企业股东天眼查接口
-            res = company_baseinfo(company_name)
-            if res == '保存成功':
-                res = corporate_shareholders(company_name)
-        else:
-            # 2-自然人
-            h_id = result[0]['企业股东']['result'][0]['id']
-            h_name = result[0]['企业股东']['result'][0]['name']
-            res = holding_enterprise(h_id, h_name)
-
-        return res
+    basic_info = find_db_data(db='公司背景', col='基本信息', query={'企业名称': name})[0]
+    if '上市' in basic_info['基本信息']['companyOrgType']:
+        # 判断企业大股东类型
+        result = find_db_data(db='公司背景', col='十大股东', query={'企业名称': name})
+        if len(result) > 0:
+            if result[0]['十大股东']['holderList'][0]['cType'] == 1:
+                # 1-公司
+                company_name = result[0]['十大股东']['holderList'][0]['name']
+                # 股东为公司，需要再次调用基本信息/企业股东天眼查接口
+                res = company_baseinfo(company_name)
+                if res == '保存成功':
+                    res = corporate_shareholders(company_name)
+            else:
+                # 2-自然人
+                h_id = result[0]['十大股东']['holderList'][0]['id']
+                h_name = result[0]['十大股东']['holderList'][0]['name']
+                res = holding_enterprise(h_id, h_name)
+            return res
+    else:
+        # 判断企业大股东类型
+        result = find_db_data(db='公司背景', col='企业股东', query={'企业名称': name})
+        if len(result) > 0:
+            if result[0]['企业股东']['result'][0]['type'] == 1:
+                # 1-公司
+                company_name = result[0]['企业股东']['result'][0]['name']
+                # 股东为公司，需要再次调用基本信息/企业股东天眼查接口
+                res = company_baseinfo(company_name)
+                if res == '保存成功':
+                    res = corporate_shareholders(company_name)
+            else:
+                # 2-自然人
+                h_id = result[0]['企业股东']['result'][0]['id']
+                h_name = result[0]['企业股东']['result'][0]['name']
+                res = holding_enterprise(h_id, h_name)
+            return res
 
 
 # 保存全量数据
@@ -375,5 +393,5 @@ def save_full_data(name):
 
 
 if __name__ == '__main__':
-    r = corporate_shareholders('浪潮软件股份有限公司')
+    r = shareholder_information('大族激光科技产业集团股份有限公司')
     print(r)
\ No newline at end of file