Added rounding to distance, prevented end_time from NoneType as Times…

…tamps need to be actual timestamps
edwork · Jan 6, 2023 · 6508a1d · 6508a1d
1 parent 13d99e2
commit 6508a1d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/custom_components/peloton/__init__.py b/custom_components/peloton/__init__.py
@@ -250,16 +250,16 @@ def compile_quant_data(
             "End Time",
             datetime.fromtimestamp(workout_stats_summary["end_time"], user_timezone)
             if "end_time" in workout_stats_summary
-            and workout_stats_summary["end_time"] is not None
-            else None,
+            and workout_stats_summary["end_time"] is not None or not 'null'
+            else datetime.fromtimestamp(workout_stats_summary["start_time"], user_timezone),
             None,
             SensorDeviceClass.TIMESTAMP,
             SensorStateClass.MEASUREMENT,
             "mdi:timer-sand-complete",
         ),
         PelotonStat(
             "Duration",
-            duration_sec / 60
+            round((duration_sec / 60), 2)
             if (
                 (duration_sec := workout_stats_summary.get("ride", {}).get("duration"))
                 and duration_sec is not None