fix handling for timestrings without date

FaserF · Feb 26, 2025 · 3468994 · 3468994
1 parent b56c8ca
commit 3468994
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/custom_components/db_infoscreen/sensor.py b/custom_components/db_infoscreen/sensor.py
@@ -51,13 +51,17 @@ def format_departure_time(self, departure_time):
             departure_time = datetime.fromtimestamp(departure_time)
             _LOGGER.debug("Converted departure time from timestamp: %s", departure_time)
 
-        elif isinstance(departure_time, str):  # Falls der Zeitstempel als String geliefert wird
+        elif isinstance(departure_time, str):
             try:
                 departure_time = datetime.strptime(departure_time, "%Y-%m-%d %H:%M:%S")
                 _LOGGER.debug("Converted departure time from string: %s", departure_time)
             except ValueError:
-                _LOGGER.warning("Unable to parse departure time from string: %s", departure_time)
-                return None
+                try:
+                    departure_time = datetime.strptime(f"{datetime.now().date()} {departure_time}", "%Y-%m-%d %H:%M")
+                    _LOGGER.debug("Converted departure time from time string: %s", departure_time)
+                except ValueError:
+                    _LOGGER.warning("Unable to parse departure time from string: %s", departure_time)
+                    return None
 
         if isinstance(departure_time, datetime):
             _LOGGER.debug("Checking departure time date: %s", departure_time.date())