from log_module import log_export, async_log_util
from log_module.log import logger_target_codes
from utils import tool, juejin_api


class TargetCodesManager:
    """
    目标交易代码管理
    """

    def __init__(self, day=tool.get_now_date_str()):
        self.__day = day
        self.__codes_info = []
        self.load_data()

    def load_data(self):
        codes_info = log_export.load_target_codes_info(self.__day)
        if codes_info:
            self.__codes_info = codes_info

    def get_codes_info(self):
        return self.__codes_info

    @classmethod
    def update_today_codes_info(cls):
        """
        更新今日的代码信息
        :return:
        """
        datas = juejin_api.JueJinApi.get_target_codes()
        if datas:
            # 过滤掉30的代码
            datas = [x for x in datas if 100 > x['pre_close'] > 2 and x['sec_id'].find('30') != 0]
        logger_target_codes.info(f"{datas}")


if __name__ == '__main__':
    TargetCodesManager.update_today_codes_info()