correctly check for station instance

zaanposni · May 20, 2024 · b2b1cf1 · b2b1cf1
1 parent 3b992e6
commit b2b1cf1
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/vvspy/arrivals.py b/vvspy/arrivals.py
@@ -81,7 +81,7 @@ def get_arrivals(
         "type_dm": kwargs.get("type_dm", "any"),
         "anyObjFilter_dm": kwargs.get("anyObjFilter_dm", 2),
         "deleteAssignedStops": kwargs.get("deleteAssignedStops", 1),
-        "name_dm": station_id.value if type(station_id) == "<enum 'Station'>" else str(station_id),
+        "name_dm": station_id.value if isinstance(station_id, Station) else str(station_id),
         "mode": kwargs.get("mode", "direct"),
         "dmLineSelectionAll": kwargs.get("dmLineSelectionAll", 1),
         "useRealtime": kwargs.get("useRealtime", 1),  # live delay

diff --git a/vvspy/departures.py b/vvspy/departures.py
@@ -81,7 +81,7 @@ def get_departures(
         "type_dm": kwargs.get("type_dm", "any"),
         "anyObjFilter_dm": kwargs.get("anyObjFilter_dm", 2),
         "deleteAssignedStops": kwargs.get("deleteAssignedStops", 1),
-        "name_dm": station_id.value if type(station_id) == "<enum 'Station'>" else str(station_id),
+        "name_dm": station_id.value if isinstance(station_id, Station) else str(station_id),
         "mode": kwargs.get("mode", "direct"),
         "dmLineSelectionAll": kwargs.get("dmLineSelectionAll", 1),
         "useRealtime": kwargs.get("useRealtime", 1),  # live delay

diff --git a/vvspy/trip.py b/vvspy/trip.py
@@ -85,8 +85,8 @@ def get_trips(
         "language": kwargs.get("language", "de"),
         "locationServerActive": kwargs.get("locationServerActive", "1"),
         "macroWebTrip": kwargs.get("macroWebTrip", "true"),
-        "name_destination": destination_station_id.value if type(destination_station_id) == "<enum 'Station'>" else str(destination_station_id),
-        "name_origin": origin_station_id.value if type(origin_station_id) == "<enum 'Station'>" else str(origin_station_id),
+        "name_destination": destination_station_id.value if isinstance(destination_station_id, Station) else str(destination_station_id),
+        "name_origin": origin_station_id.value if isinstance(origin_station_id, Station) else str(origin_station_id),
         "noElevationProfile": kwargs.get("noElevationProfile", "1"),
         "noElevationSummary": kwargs.get("noElevationSummary", "1"),
         "outputFormat": "rapidJSON",