Add support for runback step. fix #25

pyproject.toml

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

trainaspower/finalsurge.py

@@ -58,20 +58,25 @@ def convert_step(step: models.Step, id_counter) -> dict:
         return convert_repeat(step, id_counter)
     if not isinstance(step, models.ConcreteStep):
         raise ValueError(f"unknown step type received {step.type}")
-    s = {
+    if step.length is None:
+        s = {"durationType": "OPEN"}
+    elif step.length.check("[time]"):
+        s = {
+            "durationType": "TIME",
+            "duration": str(timedelta(seconds=step.length.to("seconds").magnitude)),
+        }
+    else:
+        s = {
+            "durationType": "DISTANCE",
+            "durationDist": step.length.magnitude,
+            "distUnit": f"{step.length.units:~}",
+        }
+    s.update({
         "type": "step",
         "id": next(id_counter),
         "name": None,
-        "durationType": "TIME" if step.length.check("[time]") else "DISTANCE",
-        "duration": (
-            str(timedelta(seconds=step.length.to("seconds").magnitude))
-            if step.length.check("[time]")
-            else 0
-        ),
         "targetAbsOrPct": "",
-        "durationDist": 0 if step.length.check("[time]") else step.length.magnitude,
         "data": [],
-        "distUnit": f"{step.length.units:~}" if step.length.check("[length]") else "mi",
         "target": [
             {
                 "targetType": "power",
@@ -94,7 +99,7 @@ def convert_step(step: models.Step, id_counter) -> dict:
         ],
         "intensity": step.type,
         "comments": None,
-    }
+    })
     return s
 
 

trainaspower/models.py

@@ -1,7 +1,7 @@
 import builtins
 import datetime
 from dataclasses import dataclass
-from typing import List, NamedTuple, Union, Tuple, Any
+from typing import List, NamedTuple, Union, Tuple, Any, Optional
 
 from loguru import logger
 from pint import UnitRegistry, Quantity
@@ -86,7 +86,7 @@ class Step:
 class ConcreteStep(Step):
     power_range: PowerRange
     pace_range: PaceRange
-    length: Quantity
+    length: Optional[Quantity]
 
 
 class RepeatStep(Step):

trainaspower/trainasone.py

@@ -1,5 +1,4 @@
 import datetime
-import re
 from typing import Generator
 
 import dateparser
@@ -128,8 +127,8 @@ def convert_steps(steps, config: models.Config, perceived_effort: bool) -> Gener
                 raise ValueError(
                     f"Unsupported target type {step['targetType']}. Please ensure that you have selected speed as \"Workout step target\" in the TAO settings under \"Garmin workout preferences\"")
 
-            if step["intensity"] == "WARMUP":
-                out_step.type = "WARMUP"
+            if step["intensity"] in ["WARMUP", "COOLDOWN"]:
+                out_step.type = step["intensity"]
             elif step["intensity"] in active_step_types:
                 if "targetValueLow" in step and step["targetValueLow"] == 0.0:
                     out_step.type = "REST"
@@ -141,6 +140,9 @@ def convert_steps(steps, config: models.Config, perceived_effort: bool) -> Gener
             if step["durationType"] == "DISTANCE":
                 out_step.length = round(
                     step["durationValue"]) * models.meter
+            elif step["durationType"] == "OPEN":
+                # Runback step
+                out_step.length = None
             else:
                 out_step.length = step["durationValue"] * models.second