import logging
import os
from IPython.display import display, Markdown
from decouple import config
from trainerroad.Utils.Str import SUCCESS_MARKDOWN_WARNING, FAIL_MARKDOWN_WARNING, OUTPUT_PATH_MARKDOWN_WARNING
from trainerroad.Utils.Str import TRAINERROAD_USER, TRAINERROAD_PASSWORD
from trainerroad.Controller.Zwift import Zwift
os.environ[TRAINERROAD_USER] = ""
os.environ[TRAINERROAD_PASSWORD] = ""
START_DATE = "" # By default today's date, date format = "MM-DD-YY" For ex: "12-24-2021"
END_DATE = "" # By default today's date plus 3 years = date format = "MM-DD-YY" For ex: "12-24-2023"
INCLUDE_DATE = False # if True Download the whole calendar if False, get unique workouts
COMPRESS_ZIP = True
zwift_manager = Zwift(username=config(TRAINERROAD_USER), password=config(TRAINERROAD_PASSWORD))
success = await zwift_manager.export_training_plan(start_date=START_DATE, include_date=INCLUDE_DATE, end_date=END_DATE,
compress=COMPRESS_ZIP)
display(Markdown(SUCCESS_MARKDOWN_WARNING)) if success else display(
Markdown(FAIL_MARKDOWN_WARNING))
display(Markdown(OUTPUT_PATH_MARKDOWN_WARNING.format(zwift_manager.zipfile))) if success else display(
Markdown(""))