refactor: log messages on gateway connection state change

This commit is contained in:
Ryan 2025-02-04 17:04:32 -05:00
parent 81e30c7e70
commit 40cd8238dd
Signed by: ErrorNoInternet
GPG Key ID: 2486BFB7B1E6A4A3

View File

@ -1,14 +1,12 @@
import asyncio import asyncio
import threading import threading
import time from logging import debug, info, warning
from logging import info
import fun
import commands import commands
import core import core
import fun
import tasks import tasks
from state import client, start_time from state import client
def prepare(): def prepare():
@ -42,19 +40,29 @@ async def on_message_edit(before, after):
await core.on_message(after, edited=True) await core.on_message(after, edited=True)
async def on_ready():
info(f"logged in as {client.user} in {round(time.time() - start_time, 1)}s")
async def on_voice_state_update(member, before, after): async def on_voice_state_update(member, before, after):
await core.on_voice_state_update(member, before, after) await core.on_voice_state_update(member, before, after)
async def on_ready():
info(f"logged in as {client.user}")
async def on_connect():
debug("connected to the gateway!")
async def on_disconnect():
warning("disconnected from the gateway!")
for event_type, handlers in client.get_listeners().items(): for event_type, handlers in client.get_listeners().items():
for handler in handlers: for handler in handlers:
client.remove_listener(handler, event_type) client.remove_listener(handler, event_type)
client.add_listener(on_bulk_message_delete, "on_bulk_message_delete") client.add_listener(on_bulk_message_delete, "on_bulk_message_delete")
client.add_listener(on_connect, "on_connect")
client.add_listener(on_disconnect, "on_disconnect")
client.add_listener(on_message, "on_message") client.add_listener(on_message, "on_message")
client.add_listener(on_message_delete, "on_message_delete") client.add_listener(on_message_delete, "on_message_delete")
client.add_listener(on_message_edit, "on_message_edit") client.add_listener(on_message_edit, "on_message_edit")