code.py

import time

from config import config
from train_board import TrainBoard
from metro_api import MetroApi, MetroApiOnFireException

STATION_CODE = config["metro_station_code"]
TRAIN_GROUP = config["train_group"]
REFRESH_INTERVAL = config["refresh_interval"]


def refresh_trains() -> [dict]:
    try:
        return MetroApi.fetch_train_predictions(STATION_CODE, TRAIN_GROUP)
    except MetroApiOnFireException:
        print("WMATA Api is currently on fire. Trying again later ...")
        return None


train_board = TrainBoard(refresh_trains)

while True:
    train_board.refresh()
    time.sleep(REFRESH_INTERVAL)
    # Blink the trains
    train_board.clear()
    time.sleep(REFRESH_INTERVAL)