整合企业主页接口
This commit is contained in:
parent
9c5b7a0e7a
commit
71c572b667
|
@ -1,21 +1,16 @@
|
||||||
from company.scripts import basic_info_script, general_rating_script
|
from company.scripts import find_company_data_scripts
|
||||||
|
|
||||||
|
|
||||||
class CompanyIndex:
|
class CompanyIndex:
|
||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self.cid = None
|
self.cid = None
|
||||||
|
self.table = None
|
||||||
|
|
||||||
def basic_info_obj(self):
|
def get_company_data_obj(self):
|
||||||
result = basic_info_script(self.cid)
|
result = find_company_data_scripts(self.table, self.cid)
|
||||||
if result:
|
|
||||||
return "查询成功", result[0]
|
|
||||||
else:
|
|
||||||
return "无数据", {}
|
|
||||||
|
|
||||||
def general_rating_obj(self):
|
|
||||||
result = general_rating_script(self.cid)
|
|
||||||
if result:
|
if result:
|
||||||
|
result[0].pop('企业ID')
|
||||||
return "查询成功", result[0]
|
return "查询成功", result[0]
|
||||||
else:
|
else:
|
||||||
return "无数据", {}
|
return "无数据", {}
|
||||||
|
|
|
@ -3,29 +3,16 @@ import pymongo
|
||||||
client = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format('root', 'sromitdTW569kC#M', '116.63.130.34', 27018))
|
client = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format('root', 'sromitdTW569kC#M', '116.63.130.34', 27018))
|
||||||
|
|
||||||
|
|
||||||
def find_basic_info(param):
|
def find_data_in_company(param1, param2):
|
||||||
"""
|
"""
|
||||||
Notes
|
从企业数据库中查询数据
|
||||||
Parameters:
|
Parameters:
|
||||||
param: desc
|
param1 str 数据表
|
||||||
|
param2 dict 查询条件
|
||||||
Returns:
|
Returns:
|
||||||
res: desc
|
res: desc
|
||||||
"""
|
"""
|
||||||
collection = client['企业']['公司基本信息']
|
collection = client['企业'][param1]
|
||||||
data = collection.find(param, {'_id': False})
|
data = collection.find(param2, {'_id': False})
|
||||||
record = list(data)
|
|
||||||
return record
|
|
||||||
|
|
||||||
|
|
||||||
def find_general_rating_info(param):
|
|
||||||
"""
|
|
||||||
Notes
|
|
||||||
Parameters:
|
|
||||||
param: desc
|
|
||||||
Returns:
|
|
||||||
res: desc
|
|
||||||
"""
|
|
||||||
collection = client['企业']['综合评价分析']
|
|
||||||
data = collection.find(param, {'_id': False})
|
|
||||||
record = list(data)
|
record = list(data)
|
||||||
return record
|
return record
|
||||||
|
|
|
@ -1,4 +1,4 @@
|
||||||
from flask import Blueprint, request, Response
|
from flask import Blueprint, request
|
||||||
|
|
||||||
from common.auth import verify_token
|
from common.auth import verify_token
|
||||||
from company.CompanyIndex import CompanyIndex
|
from company.CompanyIndex import CompanyIndex
|
||||||
|
@ -6,25 +6,9 @@ from company.CompanyIndex import CompanyIndex
|
||||||
company_route = Blueprint('company', __name__)
|
company_route = Blueprint('company', __name__)
|
||||||
|
|
||||||
|
|
||||||
@company_route.route('/basic_info', methods=['GET'])
|
@company_route.route('/company_index', methods=['POST'])
|
||||||
@verify_token
|
@verify_token
|
||||||
def basic_info_route(**kwargs):
|
def company_page(**kwargs):
|
||||||
"""
|
|
||||||
企业基本信息
|
|
||||||
Parameters:
|
|
||||||
param desc
|
|
||||||
Returns:
|
|
||||||
res: desc
|
|
||||||
"""
|
|
||||||
company_index = CompanyIndex()
|
|
||||||
company_index.cid = kwargs['cid']
|
|
||||||
info, result = company_index.basic_info_obj()
|
|
||||||
return {"info": info, "result": result}
|
|
||||||
|
|
||||||
|
|
||||||
@company_route.route('/general_rating', methods=['GET'])
|
|
||||||
@verify_token
|
|
||||||
def general_rating_route(**kwargs):
|
|
||||||
"""
|
"""
|
||||||
|
|
||||||
Parameters:
|
Parameters:
|
||||||
|
@ -34,5 +18,6 @@ def general_rating_route(**kwargs):
|
||||||
"""
|
"""
|
||||||
company_index = CompanyIndex()
|
company_index = CompanyIndex()
|
||||||
company_index.cid = kwargs['cid']
|
company_index.cid = kwargs['cid']
|
||||||
info, result = company_index.general_rating_obj()
|
company_index.table = request.json['table']
|
||||||
|
info, result = company_index.get_company_data_obj()
|
||||||
return {"info": info, "result": result}
|
return {"info": info, "result": result}
|
||||||
|
|
|
@ -1,11 +1,5 @@
|
||||||
from company.db import find_basic_info, find_general_rating_info
|
from company.db import find_data_in_company
|
||||||
|
|
||||||
|
|
||||||
def basic_info_script(cid):
|
def find_company_data_scripts(table, cid):
|
||||||
data = find_basic_info({"企业ID": cid})
|
return find_data_in_company(table, {"企业ID": cid})
|
||||||
return data
|
|
||||||
|
|
||||||
|
|
||||||
def general_rating_script(cid):
|
|
||||||
data = find_general_rating_info({"企业ID": cid})
|
|
||||||
return data
|
|
||||||
|
|
Loading…
Reference in New Issue