Commit f5b3584b by dong

fix202302014

parent 5f68cf03
...@@ -37,7 +37,8 @@ class IndustryChain(db.Model): ...@@ -37,7 +37,8 @@ class IndustryChain(db.Model):
icon_url1 = db.Column(db.String(255), comment='选中时行业图标') icon_url1 = db.Column(db.String(255), comment='选中时行业图标')
status = db.Column(db.Integer, doc='启用状态1启用,2禁用', comment='启用状态1启用,2禁用') # 启用状态1启用,2禁用 status = db.Column(db.Integer, doc='启用状态1启用,2禁用', comment='启用状态1启用,2禁用') # 启用状态1启用,2禁用
industry_type = db.Column(db.Integer, comment='行业类型(0是产业集群,1是上游行业,2是中游行业,3是下游行业)') industry_type = db.Column(db.Integer, comment='行业类型(0是产业集群,1是上游行业,2是中游行业,3是下游行业)')
relate_id = db.Column(db.Integer, comment='关系id(关联与哪个产业环节') relate_id = db.Column(db.Integer, comment='关系id(关联哪个产业环节')
chain_id = db.Column(db.Integer, comment='关系id(关联哪个产业链')
enterprise = db.relationship('Enterprise', backref=db.backref('enterprise')) enterprise = db.relationship('Enterprise', backref=db.backref('enterprise'))
enterprise_num = db.Column(db.Integer, comment='相关全企业表数量') enterprise_num = db.Column(db.Integer, comment='相关全企业表数量')
......
...@@ -137,6 +137,34 @@ def get_count(industry_id, chain_name, flag): ...@@ -137,6 +137,34 @@ def get_count(industry_id, chain_name, flag):
Enterprise.c_type == industry_id, Enterprise.c_type == industry_id,
).count() ).count()
return company_count return company_count
# def get_count(industry_id, chain_name, flag):
# company_count = ''
# if flag == 1: # 晋城
# if chain_name in ["钢铁", "光机电", "煤层气", "装备制造", "铸造",
# "煤化工", "新材料", "绿色建材", "医药", "丝麻纺织服装",
# "现代服务业", "数字经济", "文化旅游"]:
# company_count = Company.query.filter(
# # Company.product_all.like('%{}%'.format(chain_name)),
# Company.c_type == industry_id
# ).count()
# else:
# company_count = Company.query.filter(
# Company.product_all.like('%{}%'.format(chain_name)),
# Company.c_type == industry_id,
# ).count()
# if flag == 2: # 全国
# if chain_name in ["钢铁", "光机电", "煤层气", "装备制造", "铸造",
# "煤化工", "新材料", "绿色建材", "医药", "丝麻纺织服装",
# "现代服务业", "数字经济", "文化旅游"]:
# chain_obj = IndustryChain.query.get(id=industry_id)
# company_count = chain_obj.enterprise_num
# else:
# chain_obj = IndustryChain.query.filter(
# IndustryChain.industry_name == chain_name,
# IndustryChain.chain_id == industry_id,
# ).first()
# company_count = chain_obj.enterprise_num
# return company_count
# def find_up_thr1(industry_type, industry_id): # def find_up_thr1(industry_type, industry_id):
......
...@@ -23,62 +23,59 @@ def handle(): ...@@ -23,62 +23,59 @@ def handle():
# 读取industry_chain数据表的产业名称 # 读取industry_chain数据表的产业名称
num = 0 num = 0
for industry_obj in industry_obj_list: for industry_obj in industry_obj_list:
# # 清空原有关联企业
# industry_obj.enterprise = []
# db.session.commit()
industry_name = industry_obj.industry_name industry_name = industry_obj.industry_name
industry_id = industry_obj.id industry_id = industry_obj.id
chain_id = industry_obj.chain_id
# 读取全国企业表,模糊匹配product_all字段 # 读取全国企业表,模糊匹配product_all字段
if industry_name in ["钢铁", "光机电", "煤层气", "装备制造", "铸造", if industry_name in ["钢铁", "光机电", "煤层气", "装备制造", "铸造",
"煤化工", "新材料", "绿色建材", "医药", "丝麻纺织服装", "煤化工", "新材料", "绿色建材", "医药", "丝麻纺织服装",
"现代服务业", "数字经济", "文化旅游"]: "现代服务业", "数字经济", "文化旅游"]:
print("正在查询【{}】的相关企业......".format(industry_name)) print("正在查询【{}】的相关企业......".format(industry_name))
enterprise_list = Enterprise.query.filter(Enterprise.c_type == industry_id).all() enterprise_list = Enterprise.query.filter(Enterprise.c_type == industry_id).count()
else: else:
print("正在查询【{}】的相关企业......".format(industry_name)) print("正在查询【{}】的相关企业......".format(industry_name))
enterprise_list = Enterprise.query.filter(Enterprise.product_all.like("%{}%".format(industry_name)), enterprise_list = Enterprise.query.filter(Enterprise.product_all.like("%{}%".format(industry_name)),
Enterprise.c_type == industry_id).all() Enterprise.c_type == chain_id).count()
if not enterprise_list: if not enterprise_list:
continue enterprise_list = 0
# 清空原有关联企业 # 清空原有关联企业
enterprise_list_num = len(enterprise_list) # enterprise_list_num = len(enterprise_list)
industry_obj.enterprise_num = enterprise_list_num industry_obj.enterprise_num = enterprise_list
industry_obj.enterprise = [] # industry_obj.enterprise = []
db.session.commit() db.session.commit()
enterprise_num = 0 # enterprise_num = 0
for enterprise_obj in enterprise_list: # for enterprise_obj in enterprise_list:
industry_obj.enterprise.append(enterprise_obj) # industry_obj.enterprise.append(enterprise_obj)
enterprise_num += 1 # enterprise_num += 1
enterprise_list_num -= 1 # enterprise_list_num -= 1
if enterprise_num >= 500: # if enterprise_num >= 100:
db.session.commit() # db.session.commit()
enterprise_num = 0 # enterprise_num = 0
continue # continue
elif enterprise_list_num == 0 and enterprise_num == 400: # # elif enterprise_list_num == 0 and enterprise_num == 400:
db.session.commit() # # db.session.commit()
enterprise_num = 0 # # enterprise_num = 0
continue # # continue
elif enterprise_list_num == 0 and enterprise_num == 300: # # elif enterprise_list_num == 0 and enterprise_num == 300:
db.session.commit() # # db.session.commit()
enterprise_num = 0 # # enterprise_num = 0
continue # # continue
elif enterprise_list_num == 0 and enterprise_num == 200: # # elif enterprise_list_num == 0 and enterprise_num == 200:
db.session.commit() # # db.session.commit()
enterprise_num = 0 # # enterprise_num = 0
continue # # continue
elif enterprise_list_num == 0 and enterprise_num == 100: # # elif enterprise_list_num == 0 and enterprise_num == 100:
db.session.commit() # # db.session.commit()
enterprise_num = 0 # # enterprise_num = 0
continue # # continue
else: # else:
pass # pass
db.session.commit() # db.session.commit()
# continue # continue
print("产业链【{}】已完成".format(industry_name)) print("产业链【{}】已完成, 数量【{}】个".format(industry_name, enterprise_list))
continue continue
# enterprise_num += 1 # enterprise_num += 1
......
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