From 7dfd6d6b7c190b5cecc2959b6e24ac66137e0403 Mon Sep 17 00:00:00 2001 From: Luna Date: Sun, 2 Aug 2020 01:52:20 -0300 Subject: [PATCH] state_manager: make deletion schedule methods sync --- litecord/gateway/state_manager.py | 4 ++-- litecord/gateway/websocket.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/litecord/gateway/state_manager.py b/litecord/gateway/state_manager.py index 08a4114..e464cfd 100644 --- a/litecord/gateway/state_manager.py +++ b/litecord/gateway/state_manager.py @@ -247,11 +247,11 @@ class StateManager: state.ws.state = None state.ws = None - async def schedule_deletion(self, state: GatewayState): + def schedule_deletion(self, state: GatewayState): task = app.loop.create_task(self._future_cleanup(state)) self.tasks[state.session_id] = task - async def unschedule_deletion(self, state: GatewayState): + def unschedule_deletion(self, state: GatewayState): try: task = self.tasks.pop(state.session_id) except KeyError: diff --git a/litecord/gateway/websocket.py b/litecord/gateway/websocket.py index 9069102..e6b9161 100644 --- a/litecord/gateway/websocket.py +++ b/litecord/gateway/websocket.py @@ -836,7 +836,7 @@ class GatewayWebsocket: return await self.invalidate_session(False) # relink this connection - await self.app.state_manager.unschedule_deletion(state) + self.app.state_manager.unschedule_deletion(state) self.state = state state.ws = self