113 lines
3.2 KiB
Python
113 lines
3.2 KiB
Python
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))
|
|
client_admin = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format('root', 'RYIHrqml#LSW6#!*', '116.63.130.34', 27020))
|
|
client_tfse_021 = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format('root', 'UTlC9cCoglD1cI1*', '116.63.130.34', 27021))
|
|
|
|
|
|
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 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
|
|
|
|
|
|
def insert_data_to_tfse(param1, param2, param3):
|
|
"""
|
|
插入数据
|
|
Parameters:
|
|
param1: string 数据库
|
|
param2: string 数据集
|
|
param3: dict 数据
|
|
Returns:
|
|
-
|
|
"""
|
|
collection = client_tfse[param1][param2]
|
|
collection.insert_one(param3)
|
|
|
|
|
|
def conserve_data_to_tfse(param1, param2, param3, param4):
|
|
"""
|
|
保存数据
|
|
Parameters:
|
|
param1: string 数据库
|
|
param2: string 数据集
|
|
param3: string cid
|
|
param4: object 需要保存的数据
|
|
Returns:
|
|
-
|
|
"""
|
|
collection = client_tfse[param1][param2]
|
|
collection.update_one(param3, {"$set": param4}, upsert=True)
|
|
|
|
|
|
def update_data(param1, param2, param3, param4):
|
|
"""
|
|
更新数据
|
|
Parameters:
|
|
param1: string 数据库
|
|
param2: string 数据集
|
|
param3: string 查询条件
|
|
param4: dict 需要更新的数据
|
|
Returns:
|
|
-
|
|
"""
|
|
collection = client_tfse[param1][param2]
|
|
res = collection.update_one(param3, {"$set": param4})
|
|
info = res.raw_result['updatedExisting']
|
|
return info
|
|
|
|
|
|
def UPSERT_SINGLE_DATA(param1, param2, param3, param4):
|
|
"""
|
|
插入单条数据
|
|
Parameters:
|
|
param1: str 数据库
|
|
param2: str 数据表
|
|
param3: dict 查询条件
|
|
param4: dict 更新或新插入的数据
|
|
"""
|
|
collection = client_tfse_021[param1][param2]
|
|
collection.update_one(param3, {"$set": param4}, upsert=True)
|
|
|
|
|
|
def UPDATE_SINGLE_DATA(param1, param2, param3, param4):
|
|
"""
|
|
插入单条数据
|
|
Parameters:
|
|
param1: str 数据库
|
|
param2: str 数据表
|
|
param3: dict 查询条件
|
|
param4: dict 更新或新插入的数据
|
|
"""
|
|
collection = client_tfse_021[param1][param2]
|
|
res = collection.update_one(param3, {"$set": param4})
|
|
info = res.raw_result['updatedExisting']
|
|
return info
|