From 08b5332c947c870ec6f28393fbfbf0f3d841bd16 Mon Sep 17 00:00:00 2001 From: Adrian Lehmann Date: Mon, 25 Dec 2023 17:05:28 +0100 Subject: [PATCH] Allow custom date for "get_timetable" --- deutsche_bahn_api/timetable_helper.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/deutsche_bahn_api/timetable_helper.py b/deutsche_bahn_api/timetable_helper.py index a2b67bf..4dfb7c1 100644 --- a/deutsche_bahn_api/timetable_helper.py +++ b/deutsche_bahn_api/timetable_helper.py @@ -44,9 +44,9 @@ def get_timetable_xml(self, hour: Optional[int] = None, date: Optional[datetime] .format(response.status_code, response.text)) return response.text - def get_timetable(self, hour: Optional[int] = None) -> list[Train]: + def get_timetable(self, hour: Optional[int] = None, date: Optional[datetime] = None) -> list[Train]: train_list: list[Train] = [] - trains = elementTree.fromstringlist(self.get_timetable_xml(hour)) + trains = elementTree.fromstringlist(self.get_timetable_xml(hour, date)) for train in trains: trip_label_object: dict[str, str] | None = None arrival_object: dict[str, str] | None = None