DigitalMgnt
/
dataplatform
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

changes 数据库配置

This commit is contained in:
P3ngSaM 2022-11-19 12:31:11 +08:00
parent 9505d88529
commit 3f1f46f2f0
6 changed files with 21 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
APPData/Model/CompanyModel.py
View File

@ -50,7 +50,7 @@ class Business(Base):
def dict(self): def dict(self):
mongodb = get_mongodb() mongodb = get_mongodb()
item = mongodb.find_data_by_id(dbname="WR_DataBase_v2_test", sheet="C3.1_工商信息", _id=self.obj_id) item = mongodb.find_data_by_id(dbname="WR_DataBase", sheet="C3.1_工商信息", _id=self.obj_id)
return item return item
@ -64,7 +64,7 @@ class Shareholder(Base):
def dict(self): def dict(self):
mongodb = get_mongodb() mongodb = get_mongodb()
item = mongodb.find_data_by_id(dbname="WR_DataBase_v2_test", sheet="C3.2_股东信息", _id=self.obj_id) item = mongodb.find_data_by_id(dbname="WR_DataBase", sheet="C3.2_股东信息", _id=self.obj_id)
return item.get("股东列表") return item.get("股东列表")
@ -78,7 +78,7 @@ class Lawsuit(Base):
def list(self): def list(self):
mongodb = get_mongodb() mongodb = get_mongodb()
item = mongodb.find_data_by_id(dbname="WR_DataBase_v2_test", sheet="C9.6_法律诉讼", _id=self.obj_id) item = mongodb.find_data_by_id(dbname="WR_DataBase", sheet="C9.6_法律诉讼", _id=self.obj_id)
item.pop('企业ID') item.pop('企业ID')
return item return item
@ -93,7 +93,7 @@ class Dishonest(Base):
def list(self): def list(self):
mongodb = get_mongodb() mongodb = get_mongodb()
item = mongodb.find_data_by_id(dbname="WR_DataBase_v2_test", sheet="C9.5_失信人", _id=self.obj_id) item = mongodb.find_data_by_id(dbname="WR_DataBase", sheet="C9.5_失信人", _id=self.obj_id)
item.pop('企业ID') item.pop('企业ID')
return item return item
@ -109,7 +109,7 @@ class Balance(Base):
def list(self): def list(self):
mongodb = get_mongodb() mongodb = get_mongodb()
item = mongodb.find_data_by_id(dbname="WR_DataBase_v2_test", sheet="C6.1_资产负债表", _id=self.obj_id) item = mongodb.find_data_by_id(dbname="WR_DataBase", sheet="C6.1_资产负债表", _id=self.obj_id)
item.pop('企业ID') item.pop('企业ID')
return item return item
@ -125,7 +125,7 @@ class Income(Base):
def list(self): def list(self):
mongodb = get_mongodb() mongodb = get_mongodb()
item = mongodb.find_data_by_id(dbname="WR_DataBase_v2_test", sheet="C6.2_利润表", _id=self.obj_id) item = mongodb.find_data_by_id(dbname="WR_DataBase", sheet="C6.2_利润表", _id=self.obj_id)
item.pop('企业ID') item.pop('企业ID')
return item return item
@ -141,7 +141,7 @@ class CashFlow(Base):
def list(self): def list(self):
mongodb = get_mongodb() mongodb = get_mongodb()
item = mongodb.find_data_by_id(dbname="WR_DataBase_v2_test", sheet="C6.3_现金流量表", _id=self.obj_id) item = mongodb.find_data_by_id(dbname="WR_DataBase", sheet="C6.3_现金流量表", _id=self.obj_id)
item.pop('企业ID') item.pop('企业ID')
return item return item

2
APPData/Router/BusinessRouter.py
View File

@ -28,7 +28,7 @@ def func01(name: str, db: Session = Depends(get_db)):
data = CompanyCrud.create_company_by_name(db=db, name=name) data = CompanyCrud.create_company_by_name(db=db, name=name)
cid = data.id cid = data.id
# 查询数据库是否有此条数据 # 查询数据库是否有此条数据
search_data = mongodb.find_data("WR_DataBase_v2_test", 'C3.1_工商信息', {"企业ID": cid}, []) search_data = mongodb.find_data("WR_DataBase", 'C3.1_工商信息', {"企业ID": cid}, [])
if search_data: if search_data:
return {"info": "Data already exists", "Datail": search_data} return {"info": "Data already exists", "Datail": search_data}
# 拉取天眼查数据到mongodb # 拉取天眼查数据到mongodb

2
APPData/Router/DishonestRouter.py
View File

@ -28,7 +28,7 @@ def func01(name: str, db: Session = Depends(get_db)):
data = CompanyCrud.create_company_by_name(db=db, name=name) data = CompanyCrud.create_company_by_name(db=db, name=name)
cid = data.id cid = data.id
# 查询数据库是否有此条数据 # 查询数据库是否有此条数据
search_data = mongodb.find_all_data("WR_DataBase_v2_test", 'C9.5_失信人', {"企业ID": cid}, []) search_data = mongodb.find_all_data("WR_DataBase", 'C9.5_失信人', {"企业ID": cid}, [])
if search_data: if search_data:
return {"info": "Data already exists", "Datail": search_data} return {"info": "Data already exists", "Datail": search_data}
# 拉取天眼查数据到mongodb # 拉取天眼查数据到mongodb

2
APPData/Router/LawsuitRouter.py
View File

@ -28,7 +28,7 @@ def func01(name: str, db: Session = Depends(get_db)):
data = CompanyCrud.create_company_by_name(db=db, name=name) data = CompanyCrud.create_company_by_name(db=db, name=name)
cid = data.id cid = data.id
# 查询数据库是否有此条数据 # 查询数据库是否有此条数据
search_data = mongodb.find_data("WR_DataBase_v2_test", 'C9.6_法律诉讼', {"企业ID": cid}, []) search_data = mongodb.find_data("WR_DataBase", 'C9.6_法律诉讼', {"企业ID": cid}, [])
if search_data: if search_data:
return {"info": "Data already exists", "Datail": search_data} return {"info": "Data already exists", "Datail": search_data}
# 拉取天眼查数据到mongodb # 拉取天眼查数据到mongodb

18
APPData/Router/ShareholderRouter.py
View File

@ -6,7 +6,7 @@ from APPData.Crud import CompanyCrud, ShareholderCrud
from Utils.AccessControl.AccessUtil import rbac from Utils.AccessControl.AccessUtil import rbac
from Utils.DataBase.MongoHelperUtils import get_mongodb, get_tyc_mongodb from Utils.DataBase.MongoHelperUtils import get_mongodb, get_tyc_mongodb
from Utils.DataBase.SqlAlchemyUtils import get_db from Utils.DataBase.SqlAlchemyUtils import get_db
from Utils.TycApi.CommonUtil import CommonUtils from Utils.TycApi.CommonUtil import CommonUtils, TOKEN
from Utils.TycApi.Configure import DB_GSBJ from Utils.TycApi.Configure import DB_GSBJ
from Utils.TycApi.Shareholder import get_shareholer_data from Utils.TycApi.Shareholder import get_shareholer_data
@ -28,18 +28,18 @@ def func01(name: str, db: Session = Depends(get_db)):
data = CompanyCrud.create_company_by_name(db=db, name=name) data = CompanyCrud.create_company_by_name(db=db, name=name)
cid = data.id cid = data.id
# 查询数据库是否有此条数据 # 查询数据库是否有此条数据
search_data = mongodb.find_data("WR_DataBase_v2_test", 'C3.2_股东信息', {"企业ID": cid}, []) search_data = mongodb.find_data("WR_DataBase", 'C3.2_股东信息', {"企业ID": cid}, [])
if search_data: if search_data:
return {"info": "Data already exists", "Datail": search_data} return {"info": "Data already exists", "Datail": search_data}
# 获取企业类型，如果不存在则拉取工商信息 # 获取企业类型，如果不存在则拉取工商信息
company_type = mongodb.find_data("WR_DataBase_v2_test", 'C3.1_工商信息', {"企业ID": cid}, ["企业类型"]) company_type = mongodb.find_data("WR_DataBase", 'C3.1_工商信息', {"企业ID": cid}, ["企业类型"])
if not company_type: if not company_type:
url = "http://127.0.0.1:8009/api/business/pull?name={}" url = "http://127.0.0.1:8009/api/business/pull?name={}"
url = url.format(name) url = url.format(name)
# headers = {"token": token} # headers = {"token": token}
res = requests.get(url=url) res = requests.get(url=url)
if res.status_code == 200: if res.status_code == 200:
company_type = mongodb.find_data("WR_DataBase_v2_test", 'C3.1_工商信息', {"企业ID": cid}, ["企业类型"]) company_type = mongodb.find_data("WR_DataBase", 'C3.1_工商信息', {"企业ID": cid}, ["企业类型"])
else: else:
raise HTTPException(status_code=202, detail="Failed to get company type") raise HTTPException(status_code=202, detail="Failed to get company type")
company_type = company_type.get('企业类型', None) company_type = company_type.get('企业类型', None)
@ -108,14 +108,14 @@ def func02(cid: str, db: Session = Depends(get_db)):
raise HTTPException(status_code=204, detail="Company Not Found") raise HTTPException(status_code=204, detail="Company Not Found")
name = db_item.name name = db_item.name
# 获取企业类型，如果不存在则拉取工商信息 # 获取企业类型，如果不存在则拉取工商信息
company_type = mongodb.find_data("WR_DataBase_v2_test", 'C3.1_工商信息', {"企业ID": cid}, ["企业类型"]) company_type = mongodb.find_data("WR_DataBase", 'C3.1_工商信息', {"企业ID": cid}, ["企业类型"])
if not company_type: if not company_type:
url = "http://127.0.0.1:8009/api/business/pull?name={}" url = "http://test.fecribd.com:5001/api/business/pull?name={}"
url = url.format(name) url = url.format(name)
# headers = {"token": token} headers = {"token": TOKEN}
res = requests.get(url=url) res = requests.get(url=url, headers=headers)
if res.status_code == 200: if res.status_code == 200:
company_type = mongodb.find_data("WR_DataBase_v2_test", 'C3.1_工商信息', {"企业ID": cid}, ["企业类型"]) company_type = mongodb.find_data("WR_DataBase", 'C3.1_工商信息', {"企业ID": cid}, ["企业类型"])
else: else:
raise HTTPException(status_code=401, detail="Failed to get company type") raise HTTPException(status_code=401, detail="Failed to get company type")
company_type = company_type.get('企业类型', None) company_type = company_type.get('企业类型', None)

2
Utils/TycApi/CommonUtil.py
View File

@ -4,6 +4,8 @@ import yaml
import random import random
import time import time
TOKEN = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VyX2luZm8iOnsidWlkIjoiVUlEMDAwMSIsImVtYWlsIjoiZmVjcmliZEBmZWNyLmNvbS5jbiIsIm5hbWUiOiJyb290Iiwicm9sZSI6Ilx1N2JhMVx1NzQwNlx1NTQ1OCIsImRlcGFydG1lbnQiOiJcdTY1NzBcdTViNTdcdTUzMTZcdTkwZTgiLCJyb2xlX2lkIjoiUk9MRTAxIiwiZGVwYXJ0bWVudF9pZCI6IkQwMDEifSwiZXhwIjoxNjgzNDc2OTA4fQ.8girdw3n0WDktRuK0aSgGor10eb11nIFvRJqUtPZum4"
p_map = { p_map = {
"heb": "河北省", "heb": "河北省",
"sx": "山西省", "sx": "山西省",