more info on station selection

JonasJoKuJonas · Mar 29, 2024 · dd1f90a · dd1f90a
1 parent c18da07
commit dd1f90a
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 21 deletions.
diff --git a/custom_components/trias/config_flow.py b/custom_components/trias/config_flow.py
@@ -211,27 +211,29 @@ async def async_step_search_station(
 
             stop_points = {}
 
-            if "Location" in data["Location"]:
-                stop_point_names = [
-                    "search",
-                    data["Location"]["Location"]["StopPoint"]["StopPointName"]["Text"],
-                ]
-                stop_points = {
-                    data["Location"]["Location"]["StopPoint"]["StopPointName"][
-                        "Text"
-                    ]: data["Location"]["Location"]["StopPoint"]["StopPointRef"]
-                }
-
-            else:
-                stop_point_names = ["search"]
-                self._stop_points = {}
-                for stop in data["Location"]:
-                    stop_point_names.append(
-                        stop["Location"]["StopPoint"]["StopPointName"]["Text"]
+            if "Location" in data:
+                locations = data["Location"]
+
+                if isinstance(locations, list):  # Handling multiple locations
+                    for stop in locations:
+                        _LOGGER.warning(stop)
+                        stop_point_name = (
+                            stop["Location"]["LocationName"]["Text"]
+                            + ", "
+                            + stop["Location"]["StopPoint"]["StopPointName"]["Text"]
+                        )
+                        stop_point_id = stop["Location"]["StopPoint"]["StopPointRef"]
+                        stop_points[stop_point_name] = stop_point_id
+                else:  # Handling single location
+                    stop_point_name = (
+                        locations["Location"]["LocationName"]["Text"]
+                        + ", "
+                        + locations["Location"]["StopPoint"]["StopPointName"]["Text"]
                     )
-                    stop_points[
-                        stop["Location"]["StopPoint"]["StopPointName"]["Text"]
-                    ] = stop["Location"]["StopPoint"]["StopPointRef"]
+                    stop_point_id = locations["Location"]["StopPoint"]["StopPointRef"]
+                    stop_points[stop_point_name] = stop_point_id
+
+            stop_point_names = ["search"] + list(stop_points.keys())
 
             self._search_data["stop_points"] = stop_points
 

diff --git a/custom_components/trias/trias_client/client.py b/custom_components/trias/trias_client/client.py
@@ -250,7 +250,7 @@ def location_information_request(
                 probability = float(
                     result["LocationInformationResponse"]["Location"]["Probability"]
                 )
-                if probability < 0.75:
+                if probability < 0.75 and probability > 0.0:
                     found = result["LocationInformationResponse"]["Location"][
                         "Location"
                     ]["StopPoint"]["StopPointName"]["Text"]