129 lines
3.1 KiB
Python
129 lines
3.1 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))
|
||
client_admin = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format('root', 'RYIHrqml#LSW6#!*', '116.63.130.34', 27020))
|
||
|
||
|
||
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 find_admin_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 insert_admin_data(param1, param2, param3):
|
||
"""
|
||
插入数据（admin）
|
||
Parameters:
|
||
param1: str 数据库
|
||
param2: str 数据集
|
||
param3: obj 查询条件
|
||
Returns:
|
||
res: list 符合条件的数据
|
||
"""
|
||
collection = client[param1][param2]
|
||
collection.insert_one(param3)
|
||
return True
|
||
|
||
|
||
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))
|