This commit is contained in:
NotSoToothless
2026-05-25 22:48:57 -07:00
committed by GitHub
parent ac01217e16
commit 66e010c890
+13 -1
View File
@@ -18,8 +18,9 @@ from dotenv import load_dotenv
load_dotenv(dotenv_path=_HERE / ".env") load_dotenv(dotenv_path=_HERE / ".env")
# Imported after load_dotenv so STORAGE_VOL_PATH is read from .env. # Imported after load_dotenv so env vars are available.
from BOT.storage import init_tss_dbs # noqa: E402 from BOT.storage import init_tss_dbs # noqa: E402
from tss_ws import listen, _handle_game # noqa: E402
logging.basicConfig( logging.basicConfig(
level=logging.INFO, level=logging.INFO,
@@ -43,6 +44,16 @@ intents.messages = True
bot = commands.Bot(command_prefix="!", intents=intents) bot = commands.Bot(command_prefix="!", intents=intents)
async def _ws_task_loop() -> None:
"""Run the TSS WS listener forever, restarting on any error."""
while True:
try:
await listen(_handle_game)
except Exception as exc:
log.error(f"TSS WS listener crashed: {exc} — restarting in 10s")
await asyncio.sleep(10)
@bot.event @bot.event
async def on_ready(): async def on_ready():
log.info(f"logged in as {bot.user} (id={bot.user.id if bot.user else '?'})") log.info(f"logged in as {bot.user} (id={bot.user.id if bot.user else '?'})")
@@ -50,6 +61,7 @@ async def on_ready():
await bot.change_presence( await bot.change_presence(
activity=discord.CustomActivity(name="Soon...") activity=discord.CustomActivity(name="Soon...")
) )
asyncio.create_task(_ws_task_loop())
if __name__ == "__main__": if __name__ == "__main__":