srbus.py

import questionary
from rich.console import Console

console = Console()

from modules import data, utils, stations, presets, settings, i18n

_ = i18n.getLocale

def mainMenu():
    console.clear()

    while 1 < 2:

        choices = [
            _("chooseStation"),
            _("choosePreset"),
            _("quickStationCheck"),
            _("settings"),
            _("exit"),
        ]

        console.rule(f"SrBus ({data.config['city']})")

        choice = questionary.select(_("chooseOption"), choices).ask()

        if choice == _("chooseStation"):
            stations.stationsMenu()

        elif choice == _("choosePreset"):
            presets.presetsMenu()

        elif choice == _("quickStationCheck"):
            stations.fastStationCheckMenu()

        elif choice == _("settings"):
            settings.settingsMenu()

        elif choice == _("exit"):
            console.clear()
            console.print(_("exitMsg"))
            exit()

        console.clear()


if __name__ == "__main__":
    i18n.updateLocale(data.config["locale"])

    if not data.config["city"] in data.getCities():
        settings.citySubmenu()

    mainMenu()