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 threading
import time
from logging import info
import fun
from logging import debug, info, warning
import commands
import core
import fun
import tasks
from state import client, start_time
from state import client
def prepare():
@ -42,19 +40,29 @@ async def on_message_edit(before, after):
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):
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 handler in handlers:
client.remove_listener(handler, event_type)
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_delete, "on_message_delete")
client.add_listener(on_message_edit, "on_message_edit")