Improve train type detection and add bus replacement text

This commit is contained in:
Julian Metzler 2022-05-14 00:22:23 +02:00
parent d24c75f3f4
commit 57ee285ca9
2 changed files with 10 additions and 0 deletions

9
run.py
View File

@ -58,6 +58,15 @@ def update_display(display, dbi):
parts = train['train'].split()
train_type = parts[0]
if train_type == "Bus":
display.info_2.set(BUS_REPLACEMENT_TEXT)
if not train_type[1].isdigit():
# Train number starts with a letter
train_type = parts[0] = parts[1].rstrip("0123456789")
parts[1] = parts[1][len(parts[0]):]
if train_type == "RE":
display.info_2.set(NO_STOP_EVERYWHERE_TEXT)

View File

@ -41,6 +41,7 @@ SMALL_DELAY_TEXT = ""
UNSPECIFIC_DELAY_TEXT = "Über 60 Min. später"
DO_NOT_BOARD_TEXT = "Nicht einsteigen"
NO_STOP_EVERYWHERE_TEXT = "Hält nicht überall"
BUS_REPLACEMENT_TEXT = "Bitte Ansage beachten"
MAP_HOUR = dict([(i, f"Abfahrt {i}") for i in range(24)] + [(i+30, f"Ankunft {i}") for i in range(24)])