import asyncio

from datetime import date

from src.script import script
from src.dataset import create_dataset

if __name__ == "__main__":
    user_order = None
    while user_order not in ["1", "2"]:
        user_order = input("choose what are you want (1,2)\n 1. generate data\n 2. export data\n").strip()
    
    if user_order == "1":
        asyncio.run(script())
    else:
        start_date = date(*map(int, input("input start date in format YYYY-MM-DD\n").strip().split("-")))
        end_date = date(*map(int, input("input end date in format YYYY-MM-DD\n").strip().split("-")))
        asyncio.run(create_dataset(start_date, end_date))
        print("data is exported in data/dataset.csv")