Commit 09f02e13 by dong

fix20221230

parent 78b6c728
......@@ -136,166 +136,166 @@ def open_dists():
# 搜索接口
@api_portraits.route("/EsSearch", methods=["POST"])
# @login_required
def search():
"""
获取用户信息
"""
# token = request.headers["token"]
# user = verify_token(token)
# user_id = user.id
# user_id = 5
# 参数获取
req_dict = request.get_json()
company = req_dict.get("company_name") # 搜索企业名称
f_name = req_dict.get("industry") # 所属行业
qualificat = req_dict.get("qualificat") # 企业资质
capital_id = req_dict.get("capital_id") # 注册资本
public_id = req_dict.get("public_id") # 上市状态
# area = req_dict.get("area") # 所在地区
province = req_dict.get("province")
city = req_dict.get("city")
district = req_dict.get("district")
yearid = req_dict.get("yearid") # 成立时间
roundid = req_dict.get("financ_id") # 融资轮次
status = req_dict.get("status") # 企业状态
scale_range = req_dict.get("scale_range") # 企业规模
entypeid = req_dict.get("entypeid") # 企业类型
page = req_dict.get("page") # 页码
per_page = req_dict.get("per_page") # 每页大小
# if area == []:
# area = ["", "", ""]
# province = area[0]
# city = area[1]
# district = area[2]
# 页数
if not page:
page = 1
page = int(page)
page = (page - 1) * per_page
args_query = dict() # 查询字段
args_term = dict() # 条件字典
args_query["company_name"] = "公司"
# 搜索框
if company:
args_query["company_name"] = company
if f_name:
args_term['f_name'] = f_name
# 公司资质
if qualificat:
if qualificat == 1:
args_term['isfive'] = 1
if qualificat == 2:
args_term['quoted_company'] = 1
if qualificat == 3:
args_term['fianacing'] = 1
if qualificat == 4:
args_term['high_new'] = 1
if qualificat == 5:
args_term['tbe'] = 1
if qualificat == 6:
args_term['dengl'] = 1
if qualificat == 7:
args_term['unicorn'] = 1
# 注册资本
if capital_id:
args_term['capital_id'] = capital_id
# 上市状态
if public_id:
args_term['public_id'] = public_id
# 省
if province:
args_term['province'] = province
# 市
if city:
args_term['city'] = city
# 区
if district:
args_term['district'] = district
# 成立时间
if yearid:
args_term['yearid'] = yearid
# 融资轮次
if roundid:
args_term['roundid'] = roundid
# 企业状态
if status:
args_term['status'] = status
# 企业规模
if scale_range:
args_term['scale_range'] = scale_range
# 公司类型
if entypeid:
args_term['entypeid'] = entypeid
body = create_body(page, per_page, args_query, args_term)
if not company:
del body["query"]["bool"]["must"][0]
try:
es = Elasticsearch([{'host': '39.100.39.50', 'port': 9200}])
data = es.search(index="jincheng_pictext_index_202202221626_29780d68", body=body)
data_list = []
print(data_list)
size = data["hits"]['total']
# 数据的列表添加进去
for i in data["hits"]['hits']:
if company:
# 判断高亮,然后添加
data_list.append({
"id": i["_source"]["mysql_id"],
"company": i["highlight"]["company_name"][0] if "company_name" in i["highlight"] else i["_source"][
"company_name"],
"area": i["_source"]["city"] + i["_source"]["district"],
"createtime": i["_source"]["build_date"],
"legal": i["_source"]["legal"],
"capital": i["_source"]["capital"],
"entype": i["_source"]["entype"],
"address": i["_source"]["address"],
"telephone": i["_source"]["telephone"],
"high_new": '高新技术企业' if i["_source"]["high_new"] == "1" else '',
"tbe": '科技型中小企业' if i["_source"]["tbe"] == "1" else '',
"fianacing": '融资企业' if i["_source"]["fianacing"] == "1" else '',
"quoted_company": '上市企业' if i["_source"]["quoted_company"] == "1" else '',
"dengl": '瞪羚企业' if i["_source"]["dengl"] == "1" else '',
"unicorn": '独角兽企业' if i["_source"]["unicorn"] == "1" else '',
"isfive": '500强企业' if i["_source"]["isfive"] == '1' else '',
})
else:
data_list.append({
"id": i["_source"]["mysql_id"],
"company": i["_source"]["company_name"],
"area": i["_source"]["city"] + i["_source"]["district"],
"createtime": i["_source"]["build_date"],
"legal": i["_source"]["legal"],
"capital": i["_source"]["capital"],
"entype": i["_source"]["entype"],
"address": i["_source"]["address"],
"telephone": i["_source"]["telephone"],
"high_new": '高新技术企业' if i["_source"]["high_new"] == "1" else '',
"tbe": '科技型中小企业' if i["_source"]["tbe"] == "1" else '',
"fianacing": '融资企业' if i["_source"]["fianacing"] == "1" else '',
"quoted_company": '上市企业' if i["_source"]["quoted_company"] == "1" else '',
"dengl": '瞪羚企业' if i["_source"]["dengl"] == "1" else '',
"unicorn": '独角兽企业' if i["_source"]["unicorn"] == "1" else '',
"isfive": '500强企业' if i["_source"]["isfive"] == '1' else '',
})
except Exception as e:
current_app.logger.error(e)
return jsonify(code=RET.DBERR, msg="es数据库错误")
data = {
"data": data_list, "size": size
}
return jsonify(code=RET.OK, msg="查找成功", data=data)
# @api_portraits.route("/EsSearch", methods=["POST"])
# # @login_required
# def search():
# """
# 获取用户信息
# """
# # token = request.headers["token"]
# # user = verify_token(token)
# # user_id = user.id
# # user_id = 5
# # 参数获取
# req_dict = request.get_json()
# company = req_dict.get("company_name") # 搜索企业名称
# f_name = req_dict.get("industry") # 所属行业
# qualificat = req_dict.get("qualificat") # 企业资质
# capital_id = req_dict.get("capital_id") # 注册资本
# public_id = req_dict.get("public_id") # 上市状态
# # area = req_dict.get("area") # 所在地区
# province = req_dict.get("province")
# city = req_dict.get("city")
# district = req_dict.get("district")
# yearid = req_dict.get("yearid") # 成立时间
# roundid = req_dict.get("financ_id") # 融资轮次
# status = req_dict.get("status") # 企业状态
# scale_range = req_dict.get("scale_range") # 企业规模
# entypeid = req_dict.get("entypeid") # 企业类型
#
# page = req_dict.get("page") # 页码
# per_page = req_dict.get("per_page") # 每页大小
#
# # if area == []:
# # area = ["", "", ""]
# # province = area[0]
# # city = area[1]
# # district = area[2]
# # 页数
# if not page:
# page = 1
# page = int(page)
# page = (page - 1) * per_page
#
# args_query = dict() # 查询字段
# args_term = dict() # 条件字典
#
# args_query["company_name"] = "公司"
# # 搜索框
# if company:
# args_query["company_name"] = company
#
# if f_name:
# args_term['f_name'] = f_name
# # 公司资质
# if qualificat:
# if qualificat == 1:
# args_term['isfive'] = 1
# if qualificat == 2:
# args_term['quoted_company'] = 1
# if qualificat == 3:
# args_term['fianacing'] = 1
# if qualificat == 4:
# args_term['high_new'] = 1
# if qualificat == 5:
# args_term['tbe'] = 1
# if qualificat == 6:
# args_term['dengl'] = 1
# if qualificat == 7:
# args_term['unicorn'] = 1
# # 注册资本
# if capital_id:
# args_term['capital_id'] = capital_id
# # 上市状态
# if public_id:
# args_term['public_id'] = public_id
# # 省
# if province:
# args_term['province'] = province
# # 市
# if city:
# args_term['city'] = city
# # 区
# if district:
# args_term['district'] = district
#
# # 成立时间
# if yearid:
# args_term['yearid'] = yearid
# # 融资轮次
# if roundid:
# args_term['roundid'] = roundid
# # 企业状态
# if status:
# args_term['status'] = status
# # 企业规模
# if scale_range:
# args_term['scale_range'] = scale_range
# # 公司类型
# if entypeid:
# args_term['entypeid'] = entypeid
#
# body = create_body(page, per_page, args_query, args_term)
# if not company:
# del body["query"]["bool"]["must"][0]
#
# try:
# es = Elasticsearch([{'host': '39.100.39.50', 'port': 9200}])
#
# data = es.search(index="jincheng_pictext_index_202202221626_29780d68", body=body)
# data_list = []
# print(data_list)
# size = data["hits"]['total']
#
# # 数据的列表添加进去
# for i in data["hits"]['hits']:
# if company:
# # 判断高亮,然后添加
# data_list.append({
# "id": i["_source"]["mysql_id"],
# "company": i["highlight"]["company_name"][0] if "company_name" in i["highlight"] else i["_source"][
# "company_name"],
# "area": i["_source"]["city"] + i["_source"]["district"],
# "createtime": i["_source"]["build_date"],
# "legal": i["_source"]["legal"],
# "capital": i["_source"]["capital"],
# "entype": i["_source"]["entype"],
# "address": i["_source"]["address"],
# "telephone": i["_source"]["telephone"],
# "high_new": '高新技术企业' if i["_source"]["high_new"] == "1" else '',
# "tbe": '科技型中小企业' if i["_source"]["tbe"] == "1" else '',
# "fianacing": '融资企业' if i["_source"]["fianacing"] == "1" else '',
# "quoted_company": '上市企业' if i["_source"]["quoted_company"] == "1" else '',
# "dengl": '瞪羚企业' if i["_source"]["dengl"] == "1" else '',
# "unicorn": '独角兽企业' if i["_source"]["unicorn"] == "1" else '',
# "isfive": '500强企业' if i["_source"]["isfive"] == '1' else '',
# })
# else:
# data_list.append({
# "id": i["_source"]["mysql_id"],
# "company": i["_source"]["company_name"],
# "area": i["_source"]["city"] + i["_source"]["district"],
# "createtime": i["_source"]["build_date"],
# "legal": i["_source"]["legal"],
# "capital": i["_source"]["capital"],
# "entype": i["_source"]["entype"],
# "address": i["_source"]["address"],
# "telephone": i["_source"]["telephone"],
# "high_new": '高新技术企业' if i["_source"]["high_new"] == "1" else '',
# "tbe": '科技型中小企业' if i["_source"]["tbe"] == "1" else '',
# "fianacing": '融资企业' if i["_source"]["fianacing"] == "1" else '',
# "quoted_company": '上市企业' if i["_source"]["quoted_company"] == "1" else '',
# "dengl": '瞪羚企业' if i["_source"]["dengl"] == "1" else '',
# "unicorn": '独角兽企业' if i["_source"]["unicorn"] == "1" else '',
# "isfive": '500强企业' if i["_source"]["isfive"] == '1' else '',
# })
# except Exception as e:
# current_app.logger.error(e)
# return jsonify(code=RET.DBERR, msg="es数据库错误")
# data = {
# "data": data_list, "size": size
# }
# return jsonify(code=RET.OK, msg="查找成功", data=data)
def body_qualificat(s, j):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment