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, buy_strategy
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__":
    code = "000333"
    print(CodeTradeStateManager().get_trade_state(code,buy_strategy.STRATEGY_TYPE_LIMIT_UP))
    print(CodeTradeStateManager().get_trade_state(code, buy_strategy.STRATEGY_TYPE_RISE_HIGH))

    CodeTradeStateManager().set_trade_state(code, buy_strategy.STRATEGY_TYPE_LIMIT_UP,
                                            CodeTradeStateManager.TRADE_STATE_ALREADY_BUY)
    print(CodeTradeStateManager().get_trade_state(code,buy_strategy.STRATEGY_TYPE_LIMIT_UP))
    print(CodeTradeStateManager().get_trade_state(code, buy_strategy.STRATEGY_TYPE_RISE_HIGH))

    CodeTradeStateManager().set_trade_state(code, buy_strategy.STRATEGY_TYPE_RISE_HIGH,
                                            CodeTradeStateManager.TRADE_STATE_ALREADY_BUY)
    print(CodeTradeStateManager().get_trade_state(code,buy_strategy.STRATEGY_TYPE_LIMIT_UP))
    print(CodeTradeStateManager().get_trade_state(code, buy_strategy.STRATEGY_TYPE_RISE_HIGH))