From 7e0d381523c5164295501a7b57e52a1d8b7fb79e Mon Sep 17 00:00:00 2001 From: profitroll Date: Thu, 6 Oct 2022 13:17:16 +0200 Subject: [PATCH] Board now support messages --- pyrmv/classes/Board.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/pyrmv/classes/Board.py b/pyrmv/classes/Board.py index dbbe3b2..a79b175 100644 --- a/pyrmv/classes/Board.py +++ b/pyrmv/classes/Board.py @@ -1,14 +1,16 @@ from datetime import datetime from pyrmv.methods.journey_detail import journey_detail from pyrmv.methods.stop_by_id import stop_by_id - +from pyrmv.classes.Message import Message class LineArrival(): def __init__(self, data, access_id: str): self.journey = journey_detail(access_id, data["JourneyDetailRef"]["ref"]) self.status = data["JourneyStatus"] - # self.messages = [] + self.messages = [] + for message in data["Messages"]["Message"]: + self.messages.append(Message(message)) self.name = data["name"] self.type = data["type"] self.stop_name = data["stop"] @@ -31,7 +33,9 @@ class LineDeparture(): def __init__(self, data, access_id: str): self.journey = journey_detail(access_id, data["JourneyDetailRef"]["ref"]) self.status = data["JourneyStatus"] - # self.messages = [] + self.messages = [] + for message in data["Messages"]["Message"]: + self.messages.append(Message(message)) self.name = data["name"] self.type = data["type"] self.stop_name = data["stop"]