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 = "507bb60a-a919-11ee-b6ad-0ae0afd621cd", "6c1dbe95191fb77cced9d805cb9c853805551ddb"
    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("507bb60a-a919-11ee-b6ad-0ae0afd621cd", "6c1dbe95191fb77cced9d805cb9c853805551ddb", target_codes,
                  MyCallback())
    app = MainApp()
    app.MainLoop()