import time
import datetime
from app import base_info
from app import plate_info
from app import league_info
from app.utils import data_util


def main_func(date_str):
    base_info_url = 'https://live.500.com/?e=' + date_str
    # 获取基本信息
    info_grid = base_info.gen_by_date_str(base_info_url, date_str)
    for info in info_grid:
        time.sleep(3)  # 等待几秒，防止请求过快
        # 获取欧洲指数信息,竞彩及威廉
        J, K, L, tmp1, tmp2, tmp3 = plate_info.gen_europe_plate_by_url(info[9])

        time.sleep(3)  # 等待几秒，防止请求过快
        # 获取竞彩让球指数信息
        N, O, P = plate_info.gen_rq_plate_by_url(info[10])

        time.sleep(3)  # 等待几秒，防止请求过快

        info[9] = J
        info[10] = K
        info[11] = L
        info[12] = ""  # 分割线
        info.append(N)
        info.append(O)
        info.append(P)
        # 根据赔率算出总的概率all_sp_percent，这部分肯定高于100%，所以下面是去除抽水之后，机构真实的胜平负概率
        all_sp_percent = 1/J+1/K+1/L
        info.append(1/(J*all_sp_percent))
        info.append(1/(K*all_sp_percent))
        info.append(1/(L*all_sp_percent))
        print(f"{info[0]}，{info[1]}，{info[2]}，{info[3]} VS {info[4]}  已于{str(datetime.datetime.now())}处理完成")

    # 构建出球队联赛近6场的相对状态 以及 主客胜平负的概率
    info_grid = league_info.gen_WDL_prop(info_grid)
    # 将数据追加到保存到result中
    data_util.save_append_to_csv("data/result.xlsx", info_grid)
    time.sleep(3)


if __name__ == '__main__':
    # 建议用一个即可，随用随改
    date_str_arr = ['2024-07-20', '2024-07-21']
    for date_str in date_str_arr:
        main_func(date_str)