Add ability to sync more than one workout with number_of_workouts option

Author: gazpachoking

config.yaml.example

@@ -8,4 +8,5 @@ recovery_pace_adjust: [0, 0]
 very_easy_pace_adjust: [0, 0]
 easy_pace_adjust: [0, 0]
 fast_pace_adjust: [0, 0]
-extreme_pace_adjust: [0, 0]
+extreme_pace_adjust: [0, 0]
+number_of_workouts: 1

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "trainaspower"
-version = "0.5.0"
+version = "0.5.1"
 description = "Convert TrainAsOne plans to power and upload to Final Surge for use with Stryd pod."
 authors = ["Chase Sterling <[email protected]>"]
 license = "MIT"

trainaspower/main.py

@@ -1,3 +1,4 @@
+from itertools import islice
 from pathlib import Path
 import sys
 
@@ -54,14 +55,14 @@ def main():
     finalsurge.login(config.finalsurge_email, config.finalsurge_password)
     stryd.login(config.stryd_email, config.stryd_password)
     try:
-        wo = trainasone.get_next_workout(config)
+        for wo in islice(trainasone.get_next_workouts(config), config.number_of_workouts):
+            finalsurge.add_workout(wo)
     except trainasone.FindWorkoutException as exc:
         with open(directory / exc.filename, "w", encoding="utf-8") as f:
             f.write(exc.html)
         logger.opt(exception=True).debug("Error")
         logger.error(f"Could not load next Train as One workout. Created {exc.filename} for debugging.")
         sys.exit(1)
-    finalsurge.add_workout(wo)
 
 
 if __name__ == "__main__":

trainaspower/models.py

@@ -25,6 +25,7 @@ class Config(BaseModel):
     easy_pace_adjust: Tuple[Union[float, int], Union[float, int]] = (0, 0)
     fast_pace_adjust: Tuple[Union[float, int], Union[float, int]] = (0, 0)
     extreme_pace_adjust: Tuple[Union[float, int], Union[float, int]] = (0, 0)
+    number_of_workouts: int = 1
 
     class Config:
         extra = 'forbid'

trainaspower/trainasone.py

@@ -1,3 +1,4 @@
+import datetime
 import re
 from typing import Generator
 
@@ -35,23 +36,27 @@ def login(email, password) -> None:
         raise Exception("Failed to login to Train as One")
 
 
-def get_next_workout(config) -> models.Workout:
+def get_next_workouts(config) -> Generator[models.Workout, None, None]:
     logger.info("Fetching next TrainAsOne workout.")
     r = tao_session.get("https://beta.trainasone.com/calendarView")
+    found = False
     try:
         upcoming = r.html.find(".today, .future")
         for day in upcoming:
             if day.find(".workout"):
-                break
-        else:
+                date = dateparser.parse(day.find(".title", first=True).text)
+                workout_url = day.find(".workout a", first=True).absolute_links.pop()
+                yield get_workout(workout_url, date, config)
+                found = True
+        if not found:
             raise Exception("Next tao workout not found.")
-        date = dateparser.parse(day.find(".title", first=True).text)
-        workout_url = day.find(".workout a", first=True).absolute_links.pop()
     except Exception as exc:
         raise FindWorkoutException(
             f"Error finding next TaO workout: {exc.args[0]}", "taocalendar.html", r.text
         ) from exc
 
+
+def get_workout(workout_url: str, date: datetime.date, config: models.Config) -> models.Workout:
     r = tao_session.get(workout_url)
     try:
         workout_html = r.html