changes 数据库配置
This commit is contained in:
parent
9505d88529
commit
3f1f46f2f0
|
@ -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
|
||||||
|
|
||||||
|
|
|
@ -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
|
||||||
|
|
|
@ -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
|
||||||
|
|
|
@ -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
|
||||||
|
|
|
@ -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)
|
||||||
|
|
|
@ -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": "山西省",
|
||||||
|
|
Loading…
Reference in New Issue