Administrator
2024-06-17 827b86062f1089acf292943f764cdf4b63374288
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
import json
 
from code_attribute import history_k_data_util, target_codes_manager
from third_data import kpl_api, kpl_data_manager, kpl_block_util, kpl_util
from third_data.kpl_data_manager import KPLCodeJXBlockManager
from trade import huaxin_trade_api
from trade.trade_manager import CodeTradeStateManager
from utils import output_data_util, tool
 
 
def __split_codes(codes):
    """
    分离代码
    :param codes:
    :return:上证股票代码, 上证非股票代码, 深证代码
    """
    szse_codes = []
    sse_other_codes = []
    sse_stock_codes = []
    for code in codes:
        market_type = tool.get_market_type(code)
        if market_type == tool.MARKET_TYPE_SZSE:
            szse_codes.append(code.encode())
        elif market_type == tool.MARKET_TYPE_SSE:
            if tool.is_stock(code):
                sse_stock_codes.append(code.encode())
            else:
                sse_other_codes.append(code.encode())
    return sse_stock_codes, sse_other_codes, szse_codes
 
if __name__ == "__main__":
    underlying_codes = target_codes_manager.get_subscript_underlying_codes()
    print(__split_codes(underlying_codes))