43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
from pyrmv.classes.Gis import Gis
|
|
from pyrmv.classes.Message import Message
|
|
from pyrmv.classes.Stop import StopTrip
|
|
from isodate import parse_duration
|
|
|
|
class Leg():
|
|
"""Trip leg object."""
|
|
|
|
def __init__(self, data: dict):
|
|
|
|
self.origin = StopTrip(data["Origin"])
|
|
self.destination = StopTrip(data["Destination"])
|
|
|
|
if "GisRef" in data:
|
|
self.gis = Gis(data["GisRef"]["ref"], data["GisRoute"])
|
|
else:
|
|
self.gis = None
|
|
|
|
self.index = data["idx"]
|
|
self.name = data["name"]
|
|
self.type = data["type"]
|
|
|
|
if "direction" in data:
|
|
self.direction = data["direction"]
|
|
else:
|
|
self.direction = None
|
|
|
|
self.messages = []
|
|
if "Messages" in data:
|
|
for message in data["Messages"]["Message"]:
|
|
self.messages.append(Message(message))
|
|
|
|
if "number" in data:
|
|
self.number = data["number"]
|
|
else:
|
|
self.number = None
|
|
|
|
self.duration = parse_duration(data["duration"])
|
|
|
|
if "dist" in data:
|
|
self.distance = data["dist"]
|
|
else:
|
|
self.distance = None |