2022-01-06 15:28:08 +08:00
|
|
|
import re
|
|
|
|
|
|
|
|
import pymongo
|
|
|
|
|
|
|
|
client = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format('root', 'sromitdTW569kC#M', '116.63.130.34', 27018))
|
|
|
|
|
|
|
|
|
2022-01-13 15:08:06 +08:00
|
|
|
def find_company_records(query, skip, limit):
|
2022-01-06 15:28:08 +08:00
|
|
|
"""
|
|
|
|
查询企业信息
|
|
|
|
Parameters:
|
|
|
|
query: 查询条件
|
2022-01-13 15:08:06 +08:00
|
|
|
skip: 默认0
|
|
|
|
limit: 查询条数
|
2022-01-06 15:28:08 +08:00
|
|
|
Returns:
|
|
|
|
record: 查询结果
|
|
|
|
"""
|
|
|
|
collection = client['用户']['用户信息']
|
2022-01-14 15:37:13 +08:00
|
|
|
if query:
|
|
|
|
data = collection.find({"企业ID": query}, {'企业ID': True, '企业名称': True, '_id': False}).skip(skip).limit(limit)
|
2022-01-06 15:28:08 +08:00
|
|
|
record = list(data)
|
2022-01-14 15:37:13 +08:00
|
|
|
if len(record) == 0:
|
|
|
|
data = collection.find({"企业名称": re.compile(query)}, {'企业ID': True, '企业名称': True, '_id': False}).skip(skip).limit(limit)
|
|
|
|
record = list(data)
|
|
|
|
return record
|
|
|
|
else:
|
|
|
|
data = collection.find({}, {'企业ID': True, '企业名称': True, '_id': False}).skip(skip).limit(limit)
|
|
|
|
record = list(data)
|
|
|
|
return record
|
2022-01-06 15:28:08 +08:00
|
|
|
|
|
|
|
|
|
|
|
def find_user_info(cid):
|
|
|
|
"""
|
|
|
|
查询用户信息
|
|
|
|
Parameters:
|
|
|
|
cid: 企业ID
|
|
|
|
Returns:
|
|
|
|
record: 查询结果
|
|
|
|
"""
|
|
|
|
collection = client['用户']['用户信息']
|
|
|
|
data = collection.find({"企业ID": cid}, {'企业ID': True, '企业名称': True, '已认证': True, '_id': False})
|
|
|
|
record = list(data)
|
|
|
|
return record
|
|
|
|
|
|
|
|
|
|
|
|
def find_company_basic_info(cid):
|
|
|
|
"""
|
|
|
|
查询公司基本信息
|
|
|
|
Parameters:
|
|
|
|
cid: 企业ID
|
|
|
|
Returns:
|
|
|
|
record: 查询结果
|
|
|
|
"""
|
|
|
|
collection = client['企业']['公司基本信息']
|
|
|
|
data = collection.find({"企业ID": cid}, {'_id': False})
|
|
|
|
record = list(data)
|
|
|
|
return record
|
|
|
|
|
|
|
|
|
|
|
|
def find_company_evaluate_info(cid):
|
|
|
|
"""
|
|
|
|
查询评价记录
|
|
|
|
Parameters:
|
|
|
|
cid: 企业ID
|
|
|
|
Returns:
|
|
|
|
record: 查询结果
|
|
|
|
"""
|
|
|
|
collection = client['评价']['评价记录']
|
|
|
|
data = collection.find({"企业ID": cid}, {'_id': False, '企业ID': False})
|
|
|
|
record = list(data)
|
|
|
|
return record
|
2022-01-18 15:35:11 +08:00
|
|
|
|
|
|
|
|
|
|
|
def update_company_info(cid, data):
|
|
|
|
"""
|
|
|
|
编辑企业信息
|
|
|
|
Parameters:
|
|
|
|
cid: 企业ID
|
|
|
|
data: 编辑后数据
|
|
|
|
Returns:
|
|
|
|
record: 编辑结果
|
|
|
|
"""
|
|
|
|
collection = client['企业']['公司基本信息']
|
|
|
|
res = collection.update_one({"企业ID": cid}, {"$set": data})
|
|
|
|
info = res.raw_result['updatedExisting']
|
|
|
|
return info
|