import time

import local_api
from gui import MainApp
from local_api.log_module.log import logger_print, printlog
from local_api.util import l1_data_api
from local_api.util.juejin_util import JueJinApi


class MyCallback(local_api.APICallback):
    def OnJueJinInited(self):
        printlog("掘金初始化")

    def OnTick(self, tick, blocks: set, limit_up_list: list):
        # printlog("OnTick:", tick)
        pass


def __get_target_codes():
    # TODO 需要获取目标票
    strategy_id, token = "95a982ce-fc2d-11ec-8ff3-0a0027000010", "77fa387b37e9bc1586ae1135827e2ccb3e7cd4e4"
    JueJinApi.init(strategy_id, token)
    results = JueJinApi().get_exchanges_codes("SHSE,SZSE", "sec_id,sec_type,sec_level")
    codes = set()
    for r in results:
        if r["sec_type"] != 1 or r["sec_level"] != 1:
            continue
        codes.add(r["sec_id"])
    return codes


if __name__ == '__main__':
    target_codes = __get_target_codes()
    l1_data_api.set_target_codes(target_codes)
    local_api.run("95a982ce-fc2d-11ec-8ff3-0a0027000010", "77fa387b37e9bc1586ae1135827e2ccb3e7cd4e4", target_codes,
                  MyCallback())
    app = MainApp()
    app.MainLoop()