Journey added

pyrmv/classes/Journey.py

@@ -1,2 +1,18 @@
+from pyrmv.classes.Stop import Stop
+from pyrmv.classes.Message import Message
+
 class Journey():
-    pass
+    def __init__(self, data: dict):
+        self.stops = []
+        self.direction = data["Directions"]["Direction"][0]["value"]
+        self.direction_flag = data["Directions"]["Direction"][0]["flag"]
+        self.messages = []
+
+        for stop in data["Stops"]["Stop"]:
+            self.stops.append(Stop(stop))
+
+        for message in data["Messages"]["Message"]:
+            self.messages.append(Message(message))
+
+    def __str__(self) -> str:
+        return f"Journey with total of {len(self.stops)} stops and {len(self.messages)} messages heading {self.direction} ({self.direction_flag})"

pyrmv/classes/Stop.py

@@ -8,17 +8,35 @@ class Stop():
 
         self.name = data["name"]
         self.id = data["id"]
+
         if "extId" in data:
             self.ext_id = data["extId"]
         else:
             self.ext_id = None
+
         if "description" in data:
             self.description = data["description"]
         else:
             self.description = None
+
         self.lon = data["lon"]
         self.lat = data["lat"]
 
+        if "routeIdx" in data:
+            self.route_index = data["routeIdx"]
+        else:
+            self.route_index = None
+
+        if "arrTrack" in data:
+            self.track_arrival = data["arrTrack"]
+        else:
+            self.track_arrival = None
+
+        if "depTrack" in data:
+            self.track_departure = data["depTrack"]
+        else:
+            self.track_departure = None
+
     def __str__(self) -> str:
         return f"Stop {self.name} at {self.lon}, {self.lat}"