diff --git a/common/scripts.py b/common/scripts.py index 5dac4df..55a8702 100644 --- a/common/scripts.py +++ b/common/scripts.py @@ -1,8 +1,38 @@ +import os +import json import functools from flask import request -from setting import API_SECRET +from setting import API_SECRET, APP_NAME + + +def file_path(param): + """ + 返回文件真实路径 + 避免因为文件路径问题带来的麻烦 + Parameters: + param: 文件相对路径 + Returns: + path: 文件真实路径 + """ + abs_path = os.path.abspath(os.path.dirname(__file__)) + rel_path = abs_path[:abs_path.find(APP_NAME) + len(APP_NAME)] + path = os.path.abspath(rel_path + param) + return path + + +def read_json_file(param): + """ + 读取json文件 + Parameters: + param: json文件真实路径 + Returns: + result: json内容 + """ + with open(param, "r", encoding='utf-8') as f: + result = json.load(f) + return result def verify_token(func): diff --git a/company/db.py b/company/db.py index e69de29..19b24e3 100644 --- a/company/db.py +++ b/company/db.py @@ -0,0 +1,30 @@ +import pymongo + +client_tfse = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format('root', 'sromitdTW569kC#M', '116.63.130.34', 27018)) +client_tyc = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format('root', 'gP%40DwMSVd5Sh6EiH', '116.63.130.34', 27019)) + + +def find_data_in_tyc(param1, param2, param3): + """ + 根据企业ID查询企业信息 + Parameters: + param1: string 数据库 db + param2: string 数据集 collection + param3: dict 查询条件 例如 {"企业名称": "xxx有限公司"} + Returns: + record: 查询结果 + """ + collection = client_tyc[param1][param2] + data = collection.find(param3, {'_id': False}) + record = list(data) + return record + + +def insert_data_in_tfse(param): + """ + + Parameters: + param: desc + Returns: + res: desc + """ \ No newline at end of file diff --git a/company/scripts.py b/company/scripts.py index 1492d1b..02f130b 100644 --- a/company/scripts.py +++ b/company/scripts.py @@ -1,6 +1,10 @@ import json +import time import requests +from common.scripts import file_path, read_json_file +from company.db import find_data_in_tyc + def drag_company_data_request(company_name): """ @@ -29,4 +33,32 @@ def basic_info_etl(cid, company_name): company_name: 企业名称 Returns: - - """ \ No newline at end of file + """ + basic = read_json_file(file_path('/company/template/基本信息.json')) + basic['企业ID'] = cid + basic['更新日期'] = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()) + + basic_tyc = find_data_in_tyc('公司背景', '基本信息', {"企业名称": company_name}) + + basic['企业名称'] = basic_tyc[0]['企业名称'] + basic['工商信息']['企业状态'] = basic_tyc[0]['基本信息']['regStatus'] + basic['工商信息']['法定代表人'] = basic_tyc[0]['基本信息']['legalPersonName'] + basic['工商信息']['企业类型'] = basic_tyc[0]['基本信息']['companyOrgType'] + basic['工商信息']['纳税人识别号'] = basic_tyc[0]['基本信息']['taxNumber'] + basic['工商信息']['经营范围'] = basic_tyc[0]['基本信息']['businessScope'] + basic['工商信息']['注册资本'] = basic_tyc[0]['基本信息']['regCapital'] + basic['工商信息']['实缴资本'] = basic_tyc[0]['基本信息']['actualCapital'] + basic['工商信息']['注册地址'] = basic_tyc[0]['基本信息']['regLocation'] + basic['工商信息']['登记机关'] = basic_tyc[0]['基本信息']['regInstitute'] + basic['工商信息']['行业'] = basic_tyc[0]['基本信息']['industry'] + basic['工商信息']['人员规模'] = basic_tyc[0]['基本信息']['staffNumRange'] + basic['工商信息']['参保人数'] = basic_tyc[0]['基本信息']['socialStaffNum'] + basic['工商信息']['小微企业'] = "是" if basic_tyc[0]['基本信息']['regStatus'] == 1 else "否" + + + + pass + + +if __name__ == '__main__': + basic_info_etl('xxx', '德阳市德润达机械有限公司') \ No newline at end of file diff --git a/company/template/基本信息.json b/company/template/基本信息.json new file mode 100644 index 0000000..b243824 --- /dev/null +++ b/company/template/基本信息.json @@ -0,0 +1,32 @@ +{ + "企业ID": null, + "更新日期": null, + "工商信息": { + "法定代表人": null, + "企业类型": null, + "纳税人识别号": null, + "纳税人资质": null, + "经营范围": null, + "注册资本": null, + "实缴资本": null, + "注册地址": null, + "登记机关": null, + "行业": null, + "人员规模": null, + "参保人数": null, + "营业年限": null + }, + "股东信息": { + "股东": null, + "持股比例": null, + "收益比例": null, + "认缴出资额": null, + "认缴日期": null + }, + "主要成员": { + "姓名": null, + "职位": null, + "持股比例": null, + "收益比例": null + } +} \ No newline at end of file