tfse-admin-api-v0.2/common/db.py

59 lines
1.5 KiB
Python

import pymongo
clients = {
"tfse": pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format('root', 'sromitdTW569kC#M', '116.63.130.34', 27018)),
"tfse_admin": pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format('root', 'RYIHrqml#LSW6#!*', '116.63.130.34', 27020)),
}
def find_data(param1, param2, param3, param4):
"""
根据查询条件查询符合条件的所有数据
Parameters:
param1: str 服务器
param2: str 数据库
param3: str 数据集
param4: obj 查询条件
Returns:
res: list 符合条件的数据
"""
collection = clients[param1][param2][param3]
res = list(collection.find(param4, {'_id': False}))
return res
def delete_data(param1, param2, param3, param4):
"""
根据查询条件删除一条文档
Parameters:
param1: str 服务器
param2: str 数据库
param3: str 数据集
param4: obj 查询条件
Returns:
res: True 执行完成
"""
collection = clients[param1][param2][param3]
collection.delete_one(param4)
return True
def insert_data(param1, param2, param3, param4):
"""
插入一条数据
Parameters:
param1: str 服务器
param2: str 数据库
param3: str 数据集
param4: obj 插入数据
Returns:
True 执行成功
False 执行失败
"""
try:
collection = clients[param1][param2][param3]
collection.insert_one(param4)
return True
except Exception:
return False