from strategy import strategy_manager
from strategy.strategy_variable import StockVariables


# 统计当日的平均溢价率
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")
    strategy_manager.low_suction_strtegy