Types Stop and StopTrip are now supported by board

2022-10-05 21:32:25 +02:00 · 2022-10-05 21:32:25 +02:00 · 6d447a64c5
commit 6d447a64c5
parent eee76fdff9
2 changed files with 6 additions and 0 deletions
--- a/pyrmv/methods/board_arrival.py
+++ b/pyrmv/methods/board_arrival.py
@ -49,6 +49,9 @@ def board_arrival(
        * BoardArrival: Instance of BoardArrival object.
    """

+    if (isinstance(direction, Stop) or isinstance(direction, StopTrip)):
+        direction = direction.id
+
    board_raw = raw_board_arrival(
        accessId=access_id,
        id=id,
--- a/pyrmv/methods/board_departure.py
+++ b/pyrmv/methods/board_departure.py
@ -49,6 +49,9 @@ def board_departure(
        * BoardDeparture: Instance of BoardDeparture object.
    """

+    if (isinstance(direction, Stop) or isinstance(direction, StopTrip)):
+        direction = direction.id
+
    board_raw = raw_board_departure(
        accessId=access_id,
        id=id,