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

33 lines
792 B
Python
Raw Normal View History

2022-01-05 15:28:07 +08:00
import pymongo
client = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format('root', 'sromitdTW569kC#M', '116.63.130.34', 27018))
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
2022-01-05 15:28:07 +08:00
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