tfse-etl-api-v0.2/company/routes.py

32 lines
814 B
Python

from flask import Blueprint, request, Response
from common.scripts import verify_token
from company.scripts import *
company_route = Blueprint('company', __name__)
@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']
# 拉取企业数据
info1 = drag_company_data_request(company_name)
if info1 != "企业数据拉取成功":
return {"info": "数据拉取失败"}, 210
# 清洗基本信息数据
basic_info_etl(cid, company_name)
return {"info": "企业基本信息初始化成功"}