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

52 lines
1.4 KiB
Python
Raw Normal View History

2021-12-09 17:17:17 +08:00
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
2021-12-16 17:46:30 +08:00
def find_data_in_tfse(param1, param2, param3):
"""
根据企业ID查询企业信息
Parameters:
param1: string 数据库 db
param2: string 数据集 collection
param3: dict 查询条件 例如 {"企业名称": "xxx有限公司"}
Returns:
record: 查询结果
"""
collection = client_tfse[param1][param2]
data = collection.find(param3, {'_id': False})
record = list(data)
return record
2021-12-10 16:12:57 +08:00
def insert_data_to_tfse(param1, param2, param3):
2021-12-09 17:17:17 +08:00
"""
2021-12-10 16:12:57 +08:00
插入数据
2021-12-09 17:17:17 +08:00
Parameters:
2021-12-10 16:12:57 +08:00
param1: string 数据库
param2: string 数据集
param3: dict 数据
2021-12-09 17:17:17 +08:00
Returns:
2021-12-10 16:12:57 +08:00
-
"""
collection = client_tfse[param1][param2]
collection.insert_one(param3)