163 lines
4.2 KiB
Python
163 lines
4.2 KiB
Python
import pymongo
|
||
|
||
DB_HOST = "116.63.130.34"
|
||
DB_PASS = "UTlC9cCoglD1cI1*"
|
||
DB_USER = "root"
|
||
DB_PORT = "27021"
|
||
client = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format(DB_USER, DB_PASS, DB_HOST, DB_PORT))
|
||
|
||
|
||
def FIND_SINGLE_COLUMN_BY_REQ(param1, param2, param3, param4):
|
||
"""
|
||
查询符合条件的第一条数据的某个制定字段值
|
||
Parameters:
|
||
param1: str 数据库
|
||
param2: str 数据表
|
||
param3: dict 查询条件
|
||
param4: str 一个指定查询字段
|
||
Returns:
|
||
record: 查询结果为空，返回None; 查询结果正常，返回查询结果的第一条数据;
|
||
"""
|
||
collection = client[param1][param2]
|
||
column = {**{'_id': False}, **{param4: 1}}
|
||
record = list(collection.find(param3, column))
|
||
return None if record == [] else record[0][param4]
|
||
|
||
|
||
def FIND_SINGLE_DATA_BY_REQ(param1, param2, param3, param4):
|
||
"""
|
||
查询符合条件的第一条数据
|
||
Parameters:
|
||
param1: str 数据库
|
||
param2: str 数据表
|
||
param3: dict 查询条件
|
||
param4: list 查询字段
|
||
Returns:
|
||
record: 查询结果为空，返回False; 查询结果正常，返回查询结果的第一条数据;
|
||
"""
|
||
collection = client[param1][param2]
|
||
columns = {**{'_id': False}, **dict(zip(param4, [1] * len(param4)))}
|
||
record = list(collection.find(param3, columns))
|
||
return False if record == [] else record[0]
|
||
|
||
|
||
def FIND_ALL_DATA_BY_REQ(param1, param2, param3, param4):
|
||
"""
|
||
查询符合条件的所有数据
|
||
Parameters:
|
||
param1: str 数据库
|
||
param2: str 数据表
|
||
param3: dict 查询条件
|
||
param4: list 查询字段
|
||
Returns:
|
||
record: list 查询结果
|
||
"""
|
||
collection = client[param1][param2]
|
||
columns = {**{'_id': False}, **dict(zip(param4, [1] * len(param4)))}
|
||
record = list(collection.find(param3, columns))
|
||
return record
|
||
|
||
|
||
def FIND_RATING_RECORDS(param):
|
||
"""
|
||
查询评价记录
|
||
Parameters:
|
||
param: 查询条件
|
||
Returns:
|
||
record: 查询结果
|
||
"""
|
||
collection = client['企业数据']['评价记录']
|
||
data = collection.find(param, {'_id': False})
|
||
record = list(data)
|
||
return record
|
||
|
||
|
||
def FIND_CC_RATING_PROCESS_DATA(param1, param2):
|
||
"""
|
||
查询流程记录
|
||
Parameters:
|
||
param1: 数据集名称
|
||
param2: 查询条件
|
||
Returns:
|
||
record: 查询结果
|
||
"""
|
||
collection = client['评价'][param1]
|
||
data = collection.find(param2, {'_id': False})
|
||
record = list(data)
|
||
return record
|
||
|
||
|
||
def find_rating_inputs(param):
|
||
"""
|
||
查询填报信息记录
|
||
Parameters:
|
||
param: 查询条件
|
||
Returns:
|
||
records: 查询结果
|
||
"""
|
||
collection = client['评价']['综合评价填报']
|
||
data = collection.find(param, {'_id': False})
|
||
record = list(data)
|
||
return record
|
||
|
||
|
||
def INSERT_COMPREHENSIVE_INPUT_DATA(data):
|
||
"""
|
||
插入一条新综合填报
|
||
Parameters:
|
||
data: 新综合填报数据
|
||
Returns:
|
||
-
|
||
"""
|
||
collection = client['评价']['综合评价填报']
|
||
collection.insert_one(data)
|
||
|
||
|
||
def UPDATE_COMPREHENSIVE_CREDIT_INPUT_DATA(rid, data):
|
||
"""
|
||
更新综合信用填报数据
|
||
Parameters:
|
||
rid: 评价ID
|
||
data: 综合填报数据
|
||
Returns:
|
||
res: None
|
||
"""
|
||
collection = client['评价']['综合评价填报']
|
||
collection.update_one({"评价ID": rid}, {"$set": data})
|
||
|
||
|
||
def find_risk_data(rid):
|
||
"""
|
||
查找评价所需的风险数据
|
||
"""
|
||
collection = client['评价']['风险数据']
|
||
data = collection.find({"评价ID": rid}, {'_id': False})
|
||
record = list(data)
|
||
return record
|
||
|
||
|
||
def update_rating_records(find, data):
|
||
"""
|
||
更新评价记录
|
||
Parameters:
|
||
find 查询条件
|
||
data 更新数据
|
||
Returns:
|
||
-
|
||
"""
|
||
collection = client['评价']['评价记录']
|
||
collection.update_one(find, {"$set": data})
|
||
|
||
|
||
def UPSERT_SINGLE_DATA(param1, param2, param3, param4):
|
||
"""
|
||
插入单条数据
|
||
Parameters:
|
||
param1: str 数据库
|
||
param2: str 数据表
|
||
param3: dict 查询条件
|
||
param4: dict 更新或新插入的数据
|
||
"""
|
||
collection = client[param1][param2]
|
||
collection.update_one(param3, {"$set": param4}, upsert=True)
|