2022-03-18 16:52:23 +08:00
|
|
|
import pymongo
|
|
|
|
|
from bson import ObjectId
|
|
|
|
|
from gridfs import GridFS
|
|
|
|
|
|
|
|
|
|
client = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format('root', 'sromitdTW569kC#M', '116.63.130.34', 27018))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def find_records(param1, param2, param3):
|
|
|
|
|
"""
|
|
|
|
|
查询数据库相关记录
|
|
|
|
|
Parameters:
|
|
|
|
|
param1: str 数据库名称
|
|
|
|
|
param2: str 数据集名称
|
|
|
|
|
param3: obj 查询条件
|
|
|
|
|
Returns:
|
|
|
|
|
result: bool 执行结果
|
|
|
|
|
"""
|
|
|
|
|
collection = client[param1][param2]
|
|
|
|
|
res = list(collection.find(param3, {'_id': False}))
|
|
|
|
|
return res
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def insert_data(param1, param2, param3):
|
|
|
|
|
"""
|
|
|
|
|
插入一条数据
|
|
|
|
|
Parameters:
|
|
|
|
|
param1: str 数据库
|
|
|
|
|
param2: str 数据集
|
|
|
|
|
param3: obj 查询条件
|
|
|
|
|
Returns:
|
|
|
|
|
result: bool 执行结果
|
|
|
|
|
"""
|
|
|
|
|
collection = client[param1][param2]
|
|
|
|
|
collection.insert_one(param3)
|
|
|
|
|
return True
|
2022-03-21 14:10:00 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
|
def update_data(param1, param2, param3, param4):
|
|
|
|
|
"""
|
|
|
|
|
更新一条数据
|
|
|
|
|
Parameters:
|
|
|
|
|
param1: str 数据库
|
|
|
|
|
param2: str 数据集
|
|
|
|
|
param3: obj 查询条件
|
|
|
|
|
param4: obj 更新数据
|
|
|
|
|
Returns:
|
|
|
|
|
result: bool 更新结果
|
|
|
|
|
"""
|
|
|
|
|
collection = client[param1][param2]
|
|
|
|
|
res = collection.update_one(param3, {"$set": param4}, upsert=True)
|
|
|
|
|
info = res.raw_result['ok']
|
|
|
|
|
return info
|
