output/code_info_output.py | ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史 | |
trade/buy_radical/block_special_codes_manager.py | ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史 |
output/code_info_output.py
@@ -16,6 +16,7 @@ from l2.l2_data_util import L2DataUtil from third_data.code_plate_key_manager import KPLCodeJXBlockManager from third_data.kpl_data_constant import LimitUpCodesBlockRecordManager from trade.buy_radical.block_special_codes_manager import BlockSpecialCodesManager from utils import global_util, tool from log_module import log, log_export from l2 import l2_data_manager, l2_data_util, transaction_progress, l2_data_manager_new, code_price_manager @@ -79,8 +80,10 @@ if code_attribute.gpcode_manager.PauseBuyCodesManager().is_in_cache(code): code_extra_infos.append("暂不买") params["code"] = code special_blocks = BlockSpecialCodesManager().get_code_blocks(code) params["code_name"] = f"{gpcode_manager.get_code_name(code)} {code} ({','.join(code_extra_infos)})" if special_blocks: params["code_name"] += "@%s" % (','.join(special_blocks)) total_datas = l2_data_util.local_today_datas.get(code) if total_datas is None: total_datas = [] trade/buy_radical/block_special_codes_manager.py
@@ -10,6 +10,7 @@ @tool.singleton class BlockSpecialCodesManager: __block_codes_dict = {} __code_blocks_dict = {} def __init__(self): self.mysql = mysql_data.Mysqldb() @@ -19,13 +20,20 @@ results = self.mysql.select_all("select _block,_code from block_special_codes") temp_dict = {} for row in results: if row[0] not in temp_dict: temp_dict[row[0]] = set() temp_dict[row[0]].add(row[1]) block, code = row[0], row[1] if block not in temp_dict: temp_dict[block] = set() if code not in self.__code_blocks_dict: self.__code_blocks_dict[code] = set() self.__code_blocks_dict[code].add(block) temp_dict[block].add(code) self.__block_codes_dict = temp_dict def get_block_codes(self, block): return self.__block_codes_dict.get(block) def get_code_blocks(self, code): return self.__code_blocks_dict.get(code) def set_block_codes_list(self, datas): """ @@ -194,6 +202,7 @@ BlockSpecialCodesManager().set_block_codes_list(datas) return len(datas) if __name__ == "__main__": # print(datas) update_block_special_codes() print( BlockSpecialCodesManager().get_code_blocks("002582"))