98 lines
2.3 KiB
Python
98 lines
2.3 KiB
Python
import pymongo
|
|
from bson import ObjectId
|
|
from gridfs import GridFS
|
|
|
|
client = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format('root', 'sromitdTW569kC#M', '116.63.130.34', 27018))
|
|
|
|
|
|
def delete_data(param1, param2, param3):
|
|
"""
|
|
根据查询条件删除一条文档
|
|
Parameters:
|
|
param1: str 数据库
|
|
param2: str 数据集
|
|
param3: obj 查询条件
|
|
Returns:
|
|
res: True 执行完成
|
|
"""
|
|
collection = client[param1][param2]
|
|
collection.delete_one(param3)
|
|
return True
|
|
|
|
|
|
def find_data(param1, param2, param3):
|
|
"""
|
|
根据查询条件查询符合条件的所有数据
|
|
Parameters:
|
|
param1: str 数据库
|
|
param2: str 数据集
|
|
param3: obj 查询条件
|
|
Returns:
|
|
res: list 符合条件的数据
|
|
"""
|
|
collection = client[param1][param2]
|
|
res = list(collection.find(param3, {'_id': False}))
|
|
return res
|
|
|
|
|
|
def update_user_info_by_cid(cid, data):
|
|
"""
|
|
更新用户信息
|
|
Parameters:
|
|
cid: 企业ID
|
|
data: 更新数据
|
|
Returns:
|
|
info: 信息
|
|
"""
|
|
collection = client['用户']['用户信息']
|
|
res = collection.update_one({"企业ID": cid}, {"$set": data})
|
|
info = res.raw_result['updatedExisting']
|
|
return info
|
|
|
|
|
|
def update_data(param1, param2, param3, param4):
|
|
"""
|
|
更新用户信息
|
|
Parameters:
|
|
param1: str 数据库
|
|
param2: str 数据集
|
|
param3: obj 查询条件
|
|
param4: obj 更新数据
|
|
Returns:
|
|
info: 更新结果
|
|
"""
|
|
collection = client[param1][param2]
|
|
res = collection.update_one(param3, {"$set": param4})
|
|
info = res.raw_result['updatedExisting']
|
|
return info
|
|
|
|
|
|
def find_user_info_by_cid(cid):
|
|
"""
|
|
更新用户信息
|
|
Parameters:
|
|
cid: 企业ID
|
|
Returns:
|
|
info: 查询结果
|
|
"""
|
|
collection = client['用户']['用户信息']
|
|
data = collection.find({"企业ID": cid}, {'_id': False})
|
|
record = list(data)
|
|
return record
|
|
|
|
|
|
def delete_pdf(param1, param2, param3):
|
|
"""
|
|
根据id删除报告/证书
|
|
Parameters:
|
|
param1: 数据库
|
|
param2: 数据集
|
|
param3: 文件fid
|
|
Returns:
|
|
record: 删除结果
|
|
"""
|
|
database = param1
|
|
collection = param2
|
|
fs = GridFS(client[database], collection)
|
|
fs.delete(ObjectId(param3))
|