diff --git a/src/data/operators.py b/src/data/operators.py index 08317c1..61e3076 100644 --- a/src/data/operators.py +++ b/src/data/operators.py @@ -1,22 +1,35 @@ db_slogans = ["Senk ju vor träwelling wis Deutsche Bahn.", "Bitte beachten Sie die umgekehrte Wagenreihung.", "Zurückbleiben bitte!", "Alle reden vom Wetter. Wir nicht.", "Die Bahn macht mobil.", "Grün abgefahren"] +db_allgemein = { + "logo": "https://marketingportal.extranet.deutschebahn.com/resource/blob/13602522/c53f806b9df966e144010b276af72dd2/Bild_09-data.png", + "color": 0xEC0016, + "slogan": db_slogans +} + +db_bawü = { + "logo": "https://upload.wikimedia.org/wikipedia/commons/thumb/4/4f/Bwegt_Logo.svg/960px-Bwegt_Logo.svg.png", + "color": 0xFFBF34, + "slogan": ["Nett hier", "Aber waren Sie schon mal in Baden-Württemberg?", "Wir können alles. Außer Hochdeutsch.", "Ausgesprochen international"] +} + +OPERATOR_ALIASES = { + "DB Regio AG Baden-Württemberg": db_bawü, + "DB Regio Stuttgart GmbH": db_bawü, + "DB Fernverkehr": db_allgemein +} + OPERATORS = { - # alles außer slogan ist ESSENZIELL für den Bot. - # Color ist Hex Color Code mit 0x am anfang. "fallback": { "unknown": True, "logo": "https://upload.wikimedia.org/wikipedia/commons/2/20/Bahn_aus_Zusatzzeichen_1024-15_A.png", "color": 0x000000 }, - "DB": { - "unknown": False, - "logo": "https://marketingportal.extranet.deutschebahn.com/resource/blob/13602522/c53f806b9df966e144010b276af72dd2/Bild_09-data.png", - "color": 0xEC0016, - "slogan": db_slogans - }, "Ostdeutsche Eisenbahn GmbH": { - "unknown": False, "logo": "https://www.odeg.de/fileadmin/user_upload/Unternehmensseite/presse/pressebilder/ODEG_Pressebilder_Logo-CMYK-JPG.jpg", "color": 0x00745C + }, + "Nederlandse Spoorwegen": { + "logo": "https://upload.wikimedia.org/wikipedia/commons/thumb/3/3d/Logo_NS.svg/960px-Logo_NS.svg.png", + "color": 0X00337F } } \ No newline at end of file diff --git a/src/embeds.py b/src/embeds.py index e0a4b22..1c45987 100644 --- a/src/embeds.py +++ b/src/embeds.py @@ -35,10 +35,7 @@ def build_info_embed() -> discord.Embed | None: arrival = format_iso_timestamp(info["arrival"]) departure = format_timestamp(conn['departure']) - if current_operator.split()[0] == "DB": - operator_infos = operator_metadata("DB") # die haben auch 500 tochterkonzerne da blickt keiner durch - else: - operator_infos = operator_metadata(current_operator) + operator_infos = operator_metadata(current_operator) embed = discord.Embed( title=handlers.train_name, diff --git a/src/handlers.py b/src/handlers.py index 111a292..3a86403 100644 --- a/src/handlers.py +++ b/src/handlers.py @@ -53,6 +53,7 @@ async def rename_vc(bot: discord.Bot, scheduled=False) -> bool: print("-----------------------------------------") logger(f"Umstieg: {train_name} von {current['station']}") + logger(f"Betreiber: {train_info['operators']}") logger(f"Wenn der Name nicht geändert wird bin ich im Cooldown") await channel.edit(name=f"{config['formatting']}{train_name}") logger(f"Name geändert!") @@ -61,9 +62,9 @@ async def rename_vc(bot: discord.Bot, scheduled=False) -> bool: async def _schedule_next_umstieg(bot, arrival_iso): try: -# arrival = datetime.fromisoformat(arrival_iso) -# wait_seconds = (arrival - datetime.now()).total_seconds() - wait_seconds = 30 # debug aktuell + arrival = datetime.fromisoformat(arrival_iso) + wait_seconds = (arrival - datetime.now()).total_seconds() + if wait_seconds > 0: remaining = str(timedelta(seconds=wait_seconds)) diff --git a/src/utils.py b/src/utils.py index 8d9c8a4..79a35cf 100644 --- a/src/utils.py +++ b/src/utils.py @@ -2,7 +2,7 @@ import requests import random from datetime import datetime from src.config import config -from src.data.operators import OPERATORS +from src.data.operators import OPERATORS, OPERATOR_ALIASES def random_connection(): while True: @@ -63,6 +63,9 @@ def operator_metadata(operator): if not operator: return OPERATORS["fallback"] + if operator in OPERATOR_ALIASES: + return OPERATOR_ALIASES[operator] + return OPERATORS.get(operator, OPERATORS["fallback"]) def logger(msg):