python-openpublictransport 0.3.1__tar.gz → 0.3.2__tar.gz

{python_openpublictransport-0.3.1 → python_openpublictransport-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: python-openpublictransport
-Version: 0.3.1
+Version: 0.3.2
 Summary: Python library for public transport APIs (EFA, OTP2, TRIAS, FPTF, HAFAS, GTFS-RT)
 License: MIT
 Requires-Python: >=3.11

{python_openpublictransport-0.3.1 → python_openpublictransport-0.3.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "python-openpublictransport"
-version = "0.3.1"
+version = "0.3.2"
 description = "Python library for public transport APIs (EFA, OTP2, TRIAS, FPTF, HAFAS, GTFS-RT)"
 readme = "README.md"
 license = { text = "MIT" }

{python_openpublictransport-0.3.1 → python_openpublictransport-0.3.2}/src/openpublictransport/providers/hafas_mgate_base.py

@@ -327,11 +327,15 @@ class HafasMgateBaseProvider(BaseProvider):
         destination_id: str,
         departure_time: Optional[datetime] = None,
         results: int = 4,
+        arrival_time: Optional[datetime] = None,
     ) -> List[Dict[str, Any]]:
         """Plan journeys between two stops with the mgate ``TripSearch`` service.
 
         ``departure_time`` is converted to the network's time zone; without it
-        the search starts now. Returns one dict per connection::
+        the search starts now. ``arrival_time`` switches to an "arrive by"
+        search instead (like hafas-client's ``arrival`` option, ``outFrwd``
+        false): the connections that arrive by then. It wins over
+        ``departure_time``. Returns one dict per connection::
 
             {"duration_minutes": int, "transfers": int, "legs": [leg, ...]}
 
@@ -342,7 +346,7 @@ class HafasMgateBaseProvider(BaseProvider):
         (the unprefixed ones are real-time when the network reports it).
         """
         tz = ZoneInfo(self.timezone)
-        when = (departure_time or datetime.now(tz)).astimezone(tz)
+        when = (arrival_time or departure_time or datetime.now(tz)).astimezone(tz)
         svc = {
             "meth": "TripSearch",
             "req": {
@@ -350,7 +354,7 @@ class HafasMgateBaseProvider(BaseProvider):
                 "arrLocL": [self._location(destination_id)],
                 "outDate": when.strftime("%Y%m%d"),
                 "outTime": when.strftime("%H%M%S"),
-                "outFrwd": True,
+                "outFrwd": arrival_time is None,
                 "numF": results,
                 "getPasslist": False,
                 "getPolyline": False,

{python_openpublictransport-0.3.1 → python_openpublictransport-0.3.2}/src/python_openpublictransport.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: python-openpublictransport
-Version: 0.3.1
+Version: 0.3.2
 Summary: Python library for public transport APIs (EFA, OTP2, TRIAS, FPTF, HAFAS, GTFS-RT)
 License: MIT
 Requires-Python: >=3.11

{python_openpublictransport-0.3.1 → python_openpublictransport-0.3.2}/tests/test_german_regional_providers.py

@@ -202,3 +202,22 @@ async def test_search_trips_with_extid_and_empty_result():
 def test_german_category_labels(cat_out, cat_out_l, expected):
     prod = {"prodCtx": {"catOut": cat_out, "catOutL": cat_out_l}}
     assert VMTProvider(session=None)._transport_type(prod) == expected
+
+
+async def test_search_trips_arrive_by():
+    provider = NAHSHProvider(session=None)
+    provider._request = AsyncMock(return_value={"svcResL": [{"res": {}}]})
+
+    await provider.search_trips("A=1@L=1@", "A=1@L=2@", arrival_time=datetime(2026, 10, 17, 6, 0, tzinfo=ZoneInfo("UTC")))
+
+    req = json.loads(provider._request.call_args.kwargs["data"])["svcReqL"][0]["req"]
+    assert req["outFrwd"] is False
+    # 06:00 UTC is 08:00 in Berlin
+    assert (req["outDate"], req["outTime"]) == ("20261017", "080000")
+
+
+async def test_search_trips_departure_stays_forward():
+    provider = NAHSHProvider(session=None)
+    provider._request = AsyncMock(return_value={"svcResL": [{"res": {}}]})
+    await provider.search_trips("A=1@L=1@", "A=1@L=2@")
+    assert json.loads(provider._request.call_args.kwargs["data"])["svcReqL"][0]["req"]["outFrwd"] is True