Laurenz/gleiswechsel
1
0
Fork 0
mirror of https://github.com/kaaninchen/Gleiswechsel.git synced 2026-07-21 19:52:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Überspringe Verbindungen welche nicht ihre Ankunftszeit angeben

Browse Source
This commit is contained in:
Kaaninchen
2026-07-18 00:08:14 +02:00
parent 23f4134b8b
commit 31fea01875
2 changed files with 23 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/handlers.py
+11 -5
View File
@@ -7,7 +7,7 @@ from src.utils import random_connection, get_train_info, logger
current = None current = None
_scheduled_task: asyncio.Task | None = None _scheduled_task: asyncio.Task | None = None
async def rename_vc(bot: discord.Bot, scheduled = False) -> bool: async def rename_vc(bot: discord.Bot, scheduled=False) -> bool:
global current, train_name, train_info, _scheduled_task global current, train_name, train_info, _scheduled_task
guild = bot.get_guild(int(config["server"])) guild = bot.get_guild(int(config["server"]))
@@ -20,10 +20,11 @@ async def rename_vc(bot: discord.Bot, scheduled = False) -> bool:
logger(f"Es konnte kein VC mit der ID {config['vc']} auf dem Server gefunden werden") logger(f"Es konnte kein VC mit der ID {config['vc']} auf dem Server gefunden werden")
return False return False
while True:
current = random_connection() current = random_connection()
train_type = current['train'].split()[0] train_type = current['train'].split()[0]
print(train_type)
if train_type in ("IC", "ICE", "NJ", "ES", "DZ"): # Diese Züge haben ihre normalen Namen, nicht sowas wie RE RE7 sondern ICE 781 zb if train_type in ("IC", "ICE", "NJ", "ES", "DZ", "RJ"):
train = current['train'] train = current['train']
train_ID = current['train'].split()[1] train_ID = current['train'].split()[1]
else: else:
@@ -31,18 +32,23 @@ async def rename_vc(bot: discord.Bot, scheduled = False) -> bool:
train_ID = current['train_number'] train_ID = current['train_number']
train_info = get_train_info(station=current['station'], train_ID=train_ID, train_type=train_type) train_info = get_train_info(station=current['station'], train_ID=train_ID, train_type=train_type)
if train_info:
break
logger(f"Keine Ankunftszeit für {current['train']} verfügbar, versuche neue Verbindung...")
train_name = f"{train} nach {current['destination']}" train_name = f"{train} nach {current['destination']}"
if not scheduled and _scheduled_task and not _scheduled_task.done(): if not scheduled and _scheduled_task and not _scheduled_task.done():
_scheduled_task.cancel() _scheduled_task.cancel()
if train_info:
_scheduled_task = asyncio.create_task(_schedule_next_umstieg(bot, train_info["arrival"])) _scheduled_task = asyncio.create_task(_schedule_next_umstieg(bot, train_info["arrival"]))
print("-----------------------------------------") print("-----------------------------------------")
logger(f"Umstieg: {train_name} von {current['station']}") logger(f"Umstieg: {train_name} von {current['station']}")
logger(f"Wenn der Name nicht geändert wird bin ich im Cooldown") logger(f"Wenn der Name nicht geändert wird bin ich im Cooldown")
await channel.edit(name=f"{config['formatting']}{train_name}",) await channel.edit(name=f"{config['formatting']}{train_name}")
logger(f"Name geändert!") logger(f"Name geändert!")
return True return True
src/utils.py
-1
View File
@@ -54,7 +54,6 @@ def get_train_info(station, train_ID, train_type):
dep = data.get("departure", []) dep = data.get("departure", [])
arrival_iso = dep["route_post_diff"][-1]["sched_arr"] arrival_iso = dep["route_post_diff"][-1]["sched_arr"]
print(dep["operators"])
return { return {
"arrival": arrival_iso, "arrival": arrival_iso,
"operators": dep["operators"] "operators": dep["operators"]