From 2c30502b50fbff67060be9cdb2156bd604f5ac4a Mon Sep 17 00:00:00 2001 From: Zomatree Date: Tue, 13 Jun 2023 04:17:14 +0100 Subject: [PATCH] Raise an error if the token is invalid --- revolt/websocket.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/revolt/websocket.py b/revolt/websocket.py index 3af088a..99f8065 100755 --- a/revolt/websocket.py +++ b/revolt/websocket.py @@ -6,6 +6,7 @@ import time from copy import copy from typing import TYPE_CHECKING, Callable, NamedTuple, cast +from .errors import RevoltError from . import utils from .channel import GroupDMChannel, TextChannel, VoiceChannel from .enums import RelationshipType @@ -102,8 +103,9 @@ class WebsocketHandler: async def handle_event(self, payload: BasePayload) -> None: event_type = payload["type"].lower() logger.debug("Recieved event %s %s", event_type, payload) + try: - if event_type != "ready": + if event_type not in ["ready", "notfound"]: await self.ready.wait() func = getattr(self, f"handle_{event_type}") @@ -115,6 +117,9 @@ class WebsocketHandler: async def handle_authenticated(self, _: BasePayload) -> None: logger.info("Successfully authenticated") + async def handle_notfound(self, _: BasePayload) -> None: + raise RevoltError("Invalid token") + async def handle_ready(self, payload: ReadyEventPayload) -> None: for user_payload in payload["users"]: user = self.state.add_user(user_payload)