profitroll/PythonRMV
profitroll/PythonRMV
1
1
Fork 0
Code Issues Pull Requests Actions Packages Releases 3 Wiki Activity

Fixed the case when stations is None

Browse Source
This commit is contained in:
Profitroll 2022-10-07 13:51:37 +02:00
parent 76774fc9cb
commit 94472ba945
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
pyrmv/classes/Client.py
View File

@ -198,12 +198,14 @@ class Client():
if isinstance(station_to, Stop): if isinstance(station_to, Stop):
station_to = station_to.ext_id station_to = station_to.ext_id
new_stations = [] if stations != None:
for stop in stations: new_stations = []
if isinstance(stop, Stop): for stop in stations:
new_stations.append(stop.ext_id) if isinstance(stop, Stop):
else: new_stations.append(stop.ext_id)
new_stations.append(stop) else:
new_stations.append(stop)
stations = new_stations
if search_mode == None: if search_mode == None:
search_mode = None search_mode = None
@ -235,7 +237,7 @@ class Client():
companies=companies, companies=companies,
lines=lines, lines=lines,
lineids=line_ids, lineids=line_ids,
stations=new_stations, stations=stations,
fromstation=station_from, fromstation=station_from,
tostation=station_to, tostation=station_to,
bothways=both_ways, bothways=both_ways,