Merge pull request #208 from gjohansson-ST/gj-20241122-7

Add OfficialLocationName to station search

Author: gjohansson-ST

pytrafikverket/trafikverket_train.py

@@ -37,7 +37,13 @@ async def async_get_train_station_from_signature(
             "TrainStation",
             "1.5",
             "rail.infrastructure",
-            ["AdvertisedLocationName", "LocationSignature", "Advertised", "Deleted"],
+            [
+                "AdvertisedLocationName",
+                "LocationSignature",
+                "Advertised",
+                "Deleted",
+                "OfficialLocationName",
+            ],
             [
                 FieldFilter(FilterOperation.EQUAL, "LocationSignature", signature),
             ],
@@ -71,10 +77,27 @@ async def async_search_train_stations(
             "TrainStation",
             "1.5",
             "rail.infrastructure",
-            ["AdvertisedLocationName", "LocationSignature", "Advertised", "Deleted"],
             [
-                FieldFilter(
-                    FilterOperation.LIKE, "AdvertisedLocationName", location_name
+                "AdvertisedLocationName",
+                "LocationSignature",
+                "Advertised",
+                "Deleted",
+                "OfficialLocationName",
+            ],
+            [
+                OrFilter(
+                    [
+                        FieldFilter(
+                            FilterOperation.LIKE,
+                            "AdvertisedLocationName",
+                            location_name,
+                        ),
+                        FieldFilter(
+                            FilterOperation.LIKE,
+                            "OfficialLocationName",
+                            location_name,
+                        ),
+                    ]
                 ),
                 FieldFilter(FilterOperation.EQUAL, "Advertised", "true"),
             ],