from copy import deepcopy

from cancel_strategy.s_l_h_cancel_strategy import LCancelBigNumComputer
from db.redis_manager_delegate import RedisUtils
from l2.cancel_buy_strategy import NewGCancelBigNumComputer, JCancelBigNumComputer
from log_module import log_export
import l2


def test_g_cancel():
    NewGCancelBigNumComputer().test()


def test_j():
    code = '603778'
    pos_list = log_export.get_l2_process_position(code)
    l2.l2_data_util.load_l2_data(code)
    total_datas = deepcopy(l2.l2_data_util.local_today_datas[code])
    l2.l2_data_util.local_today_num_operate_map.get(code).clear()
    l2.l2_data_util.local_today_buyno_map.get(code).clear()
    l2.l2_data_util.local_today_canceled_buyno_map.get(code).clear()
    l2.l2_data_util.local_today_datas[code] = []
    for p in pos_list:
        if p[0] >= 234:
            break
        add_datas = total_datas[p[0]:p[1] + 1]
        l2.l2_data_util.local_today_datas[code].extend(add_datas)
        l2.l2_data_util.load_num_operate_map(l2.l2_data_util.local_today_num_operate_map, code, add_datas)
        l2.l2_data_util.load_buy_no_map(l2.l2_data_util.local_today_buyno_map, code, add_datas)
        l2.l2_data_util.load_sell_no_map(l2.l2_data_util.local_today_sellno_map, code, add_datas)
        l2.l2_data_util.load_canceled_buy_no_map(l2.l2_data_util.local_today_canceled_buyno_map, code, add_datas)
        if p[0] > 71:
            JCancelBigNumComputer().set_real_place_order_index(code, 71, 19, False)
        result = JCancelBigNumComputer().need_cancel(code, p[0], p[1])
        # print(add_datas[-1]['val']['time'], result)
        if result and result[0]:
            print(result)


def test_l():
    LCancelBigNumComputer().test()


if __name__ == '__main__':
    test_l()
    RedisUtils.run_loop()