import pandas as pd
import os
from os import path

from rd2l_pred.training_data_prep import list_format, modification, league_money, df_gen

if __name__ == "__main__":
    os.chdir("rd2l_pred/")
    data_type = "prediction"

    draft, captains = list_format("input")
    money = league_money(captains, data_type)
    result_money = money[list(money.keys())[0]]
    season = list(money.keys())[0]

    result_money.index.name = season

    os.chdir("../")
    result_money.to_csv("result_money.csv") 
    print("Saved?")