from huaxin_client import l1_subscript_codes_manager
from strategy import strategy_manager
from strategy.strategy_variable import StockVariables

# 统计当日的平均溢价率
from third_data.kpl_block_manager import KPLCodeJXBlocksManager


def statistic_average(path):
    rate_list = []
    with open(path, mode='r', encoding='utf-8') as f:
        lines = f.readlines()
        for line in lines:
            if line.find('回测结果：') < 0:
                continue
            if line.find('当日盈亏：') < 0:
                continue
            r = round(float(line.split("当日盈亏：")[1].split("，")[0].replace("%", "")), 2)
            rate_list.append(r)
    print("平均利润率：", round(sum(rate_list) / len(rate_list), 2))
    print("总利润率：", round(sum(rate_list), 2), "总买票数量：", len(rate_list))


if __name__ == "__main__":
    # print("======3个票涨停之后买+开盘价>=-3")
    # statistic_average(r"C:\Users\Administrator\Desktop\3个票涨停之后买.txt")
    # print("======3个票涨停之后买+不限开盘涨幅+3个涨停之后大单打折")
    # statistic_average(r"C:\Users\Administrator\Desktop\3个票涨停之后买_不限开盘涨幅.txt")
    codes = set()
    codes_sh, codes_sz = l1_subscript_codes_manager.get_codes()
    codes |= set([x.decode() for x in codes_sh])
    codes |= set([x.decode() for x in codes_sz])
    KPLCodeJXBlocksManager('2025-06-17', codes).start_download_blocks()
    # target_block = {"石油石化", "天然气", "化工"}
    # for code in code_blocks:
    #     blocks = code_blocks.get(code)
    #     if len(blocks & target_block) == len(target_block):
    #         print(code, blocks)