初始化企业数据

company/routes.py

@@ -6,15 +6,24 @@ from company.scripts import *
 company_route = Blueprint('company', __name__)
 
 
-@company_route.route('/basic_info', methods=['POST'])
+@company_route.route('/init_company_data', methods=['POST'])
 @verify_token
 def basic_info():
     """
-    基础信息清晰
+    初始化企业数据
+    包括 天眼查数据库、企业基本信息
     Parameters:
         company_name 企业名称
     Returns:
         res: desc
     """
+    cid = request.json['cid']
     company_name = request.json['company_name']
-    basic_info_etl(company_name)
+
+    # 拉取企业数据
+    info1 = drag_company_data_request(company_name)
+    if info1 != "企业数据拉取成功":
+        return {"info": "数据拉取失败"}, 210
+
+    # 清洗基本信息数据
+    basic_info_etl(cid, company_name)

company/scripts.py

@@ -1,9 +1,32 @@
-def basic_info_etl(param):
+import json
+import requests
+
+
+def drag_company_data_request(company_name):
+    """
+    从天眼查接口获取企业数据
+    Parameters:
+        company_name: 企业名称
+    Returns:
+        -
+    """
+    url = "http://139.9.249.34:51009/api/tyc/drag_data"
+    headers = {'token': "uzdq51N4!I0%HY4sCaQ!aeCSIDIVIdAM"}
+    data = {"企业名称": company_name}
+    res = requests.post(url=url, headers=headers, data=json.dumps(data))
+    if res.status_code == 200:
+        return "企业数据拉取成功"
+    else:
+        return "企业数据拉取失败"
+
+
+def basic_info_etl(cid, company_name):
     """
     根据企业名称，查询天眼查数据库
     将数据按规定格式存储到股交企业数据库中
     Parameters:
-        param: desc
+        cid: 企业ID
+        company_name: 企业名称
     Returns:
-        res: desc
+        -
     """