行业查询接口
This commit is contained in:
parent
2d3ff44c25
commit
199e7d8d3d
|
@ -33,3 +33,31 @@ def UPDATE_DATA(param1, param2, param3):
|
|||
"""
|
||||
collection = CLIENT['宏观'][param1]
|
||||
collection.update_one(param2, {"$set": param3}, upsert=True)
|
||||
|
||||
|
||||
def FIND_DATA(param1, param2):
|
||||
"""
|
||||
根据查询条件查询符合条件的所有数据
|
||||
Parameters:
|
||||
param1: str 数据集
|
||||
param2: obj 查询条件
|
||||
Returns:
|
||||
records: list 符合条件的数据
|
||||
"""
|
||||
collection = CLIENT['宏观'][param1]
|
||||
records = list(collection.find(param2, {'_id': False}))
|
||||
return records
|
||||
|
||||
|
||||
def FUZZY_FIND_DATA(param1, param2):
|
||||
"""
|
||||
模糊查询
|
||||
Parameters:
|
||||
param1: str 数据集
|
||||
param2: obj 查询条件
|
||||
Returns:
|
||||
records: list 符合条件的数据
|
||||
"""
|
||||
collection = CLIENT['宏观'][param1]
|
||||
records = list(collection.find(param2, {'_id': False}))
|
||||
return records
|
||||
|
|
|
@ -1,5 +1,21 @@
|
|||
from Manage.industry.industry_db import UPDATE_DATA
|
||||
from Manage.industry.industry_utils import dict_fields_map
|
||||
from Manage.industry.industry_db import UPDATE_DATA, FUZZY_FIND_DATA
|
||||
from Manage.industry.industry_utils import dict_fields_map, sub_dict
|
||||
|
||||
|
||||
def search_industry_impl(param):
|
||||
"""
|
||||
搜索行业列表
|
||||
Parameters:
|
||||
param: str 搜索条件
|
||||
Returns:
|
||||
result: list 行业名称
|
||||
"""
|
||||
param = "" if type(param) is not str else param # 强制模糊搜索条件为str类型
|
||||
records = FUZZY_FIND_DATA('行业宏观分析', {'行业': {'$regex': param}})
|
||||
result = []
|
||||
for record in records:
|
||||
result.append(sub_dict(record, ['行业'])['行业'])
|
||||
return result
|
||||
|
||||
|
||||
def primary_industry_update_data_impl(param):
|
||||
|
|
|
@ -1,6 +1,6 @@
|
|||
import time
|
||||
|
||||
from Manage.industry.industry_impl import primary_industry_update_data_impl
|
||||
from Manage.industry.industry_impl import primary_industry_update_data_impl, search_industry_impl
|
||||
|
||||
|
||||
class PrimaryIndustry:
|
||||
|
@ -14,6 +14,9 @@ class PrimaryIndustry:
|
|||
self.scale_distribution = None
|
||||
self.update_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
|
||||
|
||||
def search_industry(self):
|
||||
return search_industry_impl(self.name)
|
||||
|
||||
def update_analysis_data(self):
|
||||
"""
|
||||
更新行业分析数据
|
||||
|
|
|
@ -6,16 +6,19 @@ from Manage.industry.industry_obj import PrimaryIndustry
|
|||
industry_route = Blueprint('/manage/industry', __name__)
|
||||
|
||||
|
||||
@industry_route.route('/search')
|
||||
@industry_route.route('/search', methods=['POST'])
|
||||
@verify_token
|
||||
@authority_scope(['admin', 'developer', 'analysts', 'operator '])
|
||||
@authority_scope(['admin', 'developer', 'analysts', 'operator'])
|
||||
def search_route(**kwargs):
|
||||
pass
|
||||
pi = PrimaryIndustry()
|
||||
pi.name = request.json['行业']
|
||||
result = pi.search_industry()
|
||||
return {"info": "查询成功", "result": result}
|
||||
|
||||
|
||||
@industry_route.route('/get_primary_industry_data')
|
||||
@verify_token
|
||||
@authority_scope(['admin', 'developer', 'analysts', 'operator '])
|
||||
@authority_scope(['admin', 'developer', 'analysts', 'operator'])
|
||||
def get_primary_industry_data_route(**kwargs):
|
||||
pass
|
||||
|
||||
|
|
|
@ -9,3 +9,15 @@ def dict_fields_map(param1, param2):
|
|||
for item in param1.items():
|
||||
result[param2[item[0]]] = item[1]
|
||||
return result
|
||||
|
||||
|
||||
def sub_dict(param1, param2):
|
||||
"""
|
||||
获取字典的子集
|
||||
Parameters:
|
||||
param1: 原字典
|
||||
param2: 子集字段
|
||||
Returns:
|
||||
子集
|
||||
"""
|
||||
return dict((key, value) for key, value in param1.items() if key in param2)
|
||||
|
|
Loading…
Reference in New Issue