Merge branch 'main' of https://git.miaig.dev/mia/hafas-terminal-app

backend.py

@@ -49,9 +49,49 @@ class HafasClient:
         res = self.stationRequest(stationString)
         return [(station["name"], station["extId"]) for station in res["svcResL"][0]["res"]["match"]["locL"]]
 
+    def _format_time(self, timeValue):
+        if not timeValue:
+            return None
+        return datetime.strptime(timeValue, "%H%M%S").strftime("%H:%M")
+
     def getArrDep(self, stationId, arrdep="DEP", count=1):
         res = self.arrDepRequest(stationId, arrdep, count)
-        return res["svcResL"][0]["res"]["jnyL"]
+        departures = []
+        time_key = "aTimeS" if arrdep == "ARR" else "dTimeS"
+        time_key_real = "aTimeR" if arrdep == "ARR" else "dTimeR"
+
+        for departure in res["svcResL"][0]["res"]["jnyL"]:
+            prod_list = departure.get("prodL") or []
+            prod_index = departure.get("prodX")
+            if isinstance(prod_index, int) and 0 <= prod_index < len(prod_list):
+                product = prod_list[prod_index]
+            else:
+                product = prod_list[0] if prod_list else {}
+            stop = departure.get("stbStop", {})
+
+            # Raw JSON access pattern:
+            # - Name: departure["dirTxt"]
+            # - Line: departure["prodL"][0]["nameS"] or departure["prodL"][0]["prodCtx"]["line"]
+            # - Departure/arrival time: departure["stbStop"]["dTimeS"] or ["aTimeS"]
+            departures.append({
+                "name": departure.get("dirTxt"),
+                "line": (
+                    product.get("nameS")
+                    or product.get("number")
+                    or product.get("name")
+                    or product.get("prodCtx", {}).get("line")
+                    or product.get("prodCtx", {}).get("name")
+                    or product.get("prodCtx", {}).get("nameS")
+                ),
+                "departure_time": self._format_time(stop.get(time_key)),
+                "departure_time_raw": stop.get(time_key),
+                "departure_time_real": self._format_time(stop.get(time_key_real)),
+                "departure_time_real_raw": stop.get(time_key_real),
+                "jid": departure.get("jid"),
+                "raw": departure,
+            })
+
+        return departures
 
     def getTrip(self, departure):
         res = self.tripRequest(departure)
@@ -74,6 +114,14 @@ class HafasClient:
         trip = self.getTrip(departures[0]["jid"])
         __import__('pprint').pprint(trip)
 
+    def moniterTest(self):
+        stationInput = input("Enter Station Name: ")
+        station = self.getStationNames(stationInput)
+        print(station)
+        # departures = self.getArrDep(station[0][1], arrdep="DEP", count=1)
+        departures = self.arrDepRequest(station[0][1], arrdep="DEP", count=1)
+        print(departures)
+
 
         # streq = self.stationRequest(station)["svcResL"][0]["res"]["match"]["locL"]
         #
@@ -91,4 +139,5 @@ class HafasClient:
 
 if __name__ == '__main__':
     client = HafasClient()
-    client.runTests()
+    # client.runTests()
+    client.moniterTest()