Merge pull request #72 from zaanposni/release/2.0.2

release/2.0.2
zaanposni · May 20, 2024 · b68108a · b68108a
2 parents 3b992e6 + 0fdf79e
commit b68108a
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setup(
     name="vvspy",
-    version="2.0.1",
+    version="2.0.2",
     license="MIT",
     description="API Wrapper for VVS (Verkehrsverbund Stuttgart)",
     author="zaanposni",

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",