From 99684540b2c6d05ebb3b21163da0a5e314a07052 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期一, 30 九月 2024 01:11:50 +0800
Subject: [PATCH] bug修复
---
third_data/third_blocks_manager.py | 10 ++++++++++
1 files changed, 10 insertions(+), 0 deletions(-)
diff --git a/third_data/third_blocks_manager.py b/third_data/third_blocks_manager.py
index a94ea1c..46343d8 100644
--- a/third_data/third_blocks_manager.py
+++ b/third_data/third_blocks_manager.py
@@ -5,6 +5,7 @@
from db.mysql_data_delegate import Mysqldb
from utils import middle_api_protocol
+from utils.ths_industry_util import ThsCodeIndustryManager
SOURCE_TYPE_KPL = 1 # 涓滄柟璐㈠瘜
SOURCE_TYPE_TDX = 2 # 閫氳揪淇�
@@ -21,6 +22,7 @@
# 浠ｇ爜鏉垮潡锛歿code:{1:{"b1","b2"},2:{"c1","c2"}}}
__code_source_blocks_dict = {}
__code_source_blocks_dict_origin = {}
+ __ths_industry = ThsCodeIndustryManager()
def __new__(cls, *args, **kwargs):
if not cls.__instance:
@@ -37,6 +39,11 @@
cls.__code_source_blocks_dict[result[0]] = {}
cls.__code_source_blocks_dict_origin[result[0]] = {}
blocks = set(result[2].split("銆�"))
+ if result[1] == SOURCE_TYPE_THS:
+ industry = cls.__ths_industry.get_industry(result[0])
+ if industry:
+ blocks.add(industry)
+
cls.__code_source_blocks_dict_origin[result[0]][result[1]] = blocks
cls.__code_source_blocks_dict[result[0]][result[1]] = BlockMapManager().filter_blocks(blocks)
@@ -94,6 +101,9 @@
# 鏇存柊缂撳瓨鏁版嵁
if code not in self.__code_source_blocks_dict:
self.__code_source_blocks_dict[code] = {}
+ if code not in self.__code_source_blocks_dict_origin:
+ self.__code_source_blocks_dict_origin[code] = {}
+
if blocks:
self.__code_source_blocks_dict[code][source_type] = BlockMapManager().filter_blocks(set(blocks))
self.__code_source_blocks_dict_origin[code][source_type] = set(blocks)
--
Gitblit v1.8.0