tfse-app-api-v0.2/company/company_db.py

54 lines
1.3 KiB
Python

import pymongo
MG_USER = "root"
MG_PWD = "sromitdTW569kC#M"
MG_HOST = "116.63.130.34"
MG_PORT = "27018"
CLIENT = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format(MG_USER, MG_PWD, MG_HOST, MG_PORT))
def FIND_DATA_IN_COMPANY(param1, param2):
"""
从企业数据库中查询数据
Parameters:
param1 str 数据表
param2 dict 查询条件
Returns:
record: 查询结果
"""
collection = CLIENT['企业'][param1]
data = collection.find(param2, {'_id': False})
record = list(data)
return record
def FIND_INDUSTRY_DATA_IN_RATING_INPUT(cid):
"""
从评价数据库中查询数据
Parameters:
cid str 企业ID
Returns:
record: 查询结果
"""
collection = CLIENT['评价']['综合评价填报']
data = collection.find({"企业ID": cid}, {'_id': False, "行业选择": 1, "评价年度": 1})
record = list(data)
return record
def FIND_DATA_IN_MACRO_INDUSTRY_ANALYSIS(industry):
"""
从宏观行业分析数据库中查询行业宏观分析信息
Parameters:
industry: str 行业
Returns:
record: 查询结果
"""
collection = CLIENT['宏观']['行业宏观分析']
data = collection.find({"行业": industry}, {'_id': False})
record = list(data)
if record:
return record[0]
else:
return False