from db.mysql_data_delegate import Mysqldb
from third_data.history_k_data_util import HistoryKDatasUtils
from utils import tool


class KPLLimitUpDataUtil:
    @classmethod
    def get_latest_block_infos(cls, min_day=tool.date_sub(tool.get_now_date_str(), 180), code=None,
                               max_day=tool.get_now_date_str()):
        """
        @param min_day: 默认获取180天之前的
        @param code: 代码
        @return: 最近的涨停板块信息
        """
        sql = f"SELECT r.`_code`, r.`_day`, r.`_hot_block_name`, r.`_blocks`,  r.`_open` FROM `kpl_limit_up_record` r WHERE r.`_day`>='{min_day}' and r.`_day` <'{max_day}'"

        if code:
            sql += f" AND _code='{code}'"
        sql += " order by _create_time"
        mysqldb = Mysqldb()
        results = mysqldb.select_all(sql)
        return results

    @classmethod
    def get_latest_block_infos_by_day(cls, day=tool.get_now_date_str()):
        sql = f"SELECT * FROM `kpl_limit_up_record` r WHERE r.`_day`='{day}'"
        mysqldb = Mysqldb()
        results = mysqldb.select_all(sql)
        return results


if __name__ == '__main__':
    # 删除不是交易日期的数据
    datas = KPLLimitUpDataUtil.get_latest_block_infos()
    len(datas)