Implement parsing operators

pyhafas/profile/base/helper/parse_leg.py

@@ -117,7 +117,8 @@ def parse_leg(
                     arrival.get('aPlatfS', arrival.get('aPltfR', arrival.get('aPltfS', {})).get('txt'))),
                 stopovers=leg_stopovers,
                 remarks=[self.parse_remark(common['remL'][msg['remX']], common)
-                         for msg in journey.get('msgL', {}) if msg.get('remX') is not None])
+                         for msg in journey.get('msgL', {}) if msg.get('remX') is not None],
+                operators=[op['name'] for op in common['opL']] if common['opL'] else None)
 
     def parse_legs(
             self: ProfileInterface,

pyhafas/types/fptf.py

@@ -213,6 +213,8 @@ class Leg(FPTFObject):
     :vartype stopovers: Optional[List[Stopover]]
     :ivar remarks: (optional) List of remarks
     :vartype remarks: List[Remark]
+    :ivar operators: (optional) The operators of the train
+    :vartype operators: List[str]
     """
 
     def __init__(
@@ -231,7 +233,8 @@ def __init__(
             arrival_delay: Optional[datetime.timedelta] = None,
             arrival_platform: Optional[str] = None,
             stopovers: Optional[List[Stopover]] = None,
-            remarks: Optional[List[Remark]] = None
+            remarks: Optional[List[Remark]] = None,
+            operators: Optional[List[str]] = None
     ):
         """
         FPTF `Leg` object
@@ -272,6 +275,7 @@ def __init__(
         if remarks is None:
             remarks = []
         self.remarks: List[Remark] = remarks
+        self.operators = operators
 
 
 class Journey(FPTFObject):

tests/db/parsing/journey_test.py

@@ -98,6 +98,7 @@ def test_db_journey_parsing():
                 priority=560,
                 trip_id=None
             )
-        ])]
+        ],
+        operators=['DB Regio AG NRW'])]
     )
     assert DBProfile().parse_journey_request(hafas_response) == correct_journey

tests/db/parsing/journeys_test.py

@@ -97,7 +97,8 @@ def test_db_journeys_parsing():
                     priority=560,
                     trip_id=None
                 )
-            ]
+            ],
+            operators=['DB Regio AG NRW']
         )],
     )]
     assert DBProfile().parse_journeys_request(hafas_response) == correct_journeys

tests/db/parsing/trip_test.py

@@ -398,6 +398,7 @@ def test_db_trips_parsing():
                 priority=560,
                 trip_id=None
             )
-        ]
+        ],
+        operators=['DB Regio AG NRW']
     )
     assert DBProfile().parse_trip_request(hafas_response) == correct_trip

tests/vsn/parsing/journeys_test.py

@@ -130,7 +130,8 @@ def test_vsn_journeys_parsing():
                             priority=899,
                             trip_id=None
                         )
-                    ]
+                    ],
+                    operators=['NordWestBahn', 'NordWestBahn']
                 )
             ]
         )