bug修复

Administrator

2024-09-29 f8ff0e2645846ad96edb99045d7e642f97c8b7bb

 third_data/third_blocks_manager.py

@@ -4,7 +4,8 @@
from itertools import combinations

from db.mysql_data_delegate import Mysqldb
from third_data import kpl_api
from third_data import kpl_api, block_web_api
from utils import middle_api_protocol

SOURCE_TYPE_KPL = 1  # 东方财富
SOURCE_TYPE_TDX = 2  # 通达信
@@ -190,8 +191,31 @@
        return self.__block_map.keys()


def load_if_less(codes):
    """
    加载
    @param codes:
    @return:
    """
    for code in codes:
        source_blocks = CodeThirdBlocksManager().get_source_blocks_origin(code)
        if source_blocks is None:
            source_blocks = {}
        all_source = {SOURCE_TYPE_EASTMONEY, SOURCE_TYPE_TDX, SOURCE_TYPE_THS}
        sources = all_source - source_blocks.keys()
        for source in sources:
            try:
                blocks = middle_api_protocol.request(middle_api_protocol.get_third_blocks(code, source))
                if blocks:
                    CodeThirdBlocksManager().set_blocks(code, blocks, source)
            except:
                pass


if __name__ == '__main__':
    code = "000761"
    blocks = kpl_api.getCodeJingXuanBlocks(code, jx=False)
    blocks = set([b[1] for b in blocks])
    print(CodeThirdBlocksManager().get_intersection_blocks_info(code, blocks))
    try:
        blocks = middle_api_protocol.request(middle_api_protocol.get_third_blocks(code, 4))
        print(blocks)
    except Exception as e:
        print(e)