diff options
| author | jack1142 <[email protected]> | 2020-07-24 13:11:45 +0200 |
|---|---|---|
| committer | Rapptz <[email protected]> | 2020-07-25 09:59:40 -0400 |
| commit | bd9821320ec2288a4c3cfada980a85e78630f94f (patch) | |
| tree | 2ddd6b39b8961e56f56b008668fe2bff26f7d9ff /discord/http.py | |
| parent | Update message references in AutoShardedConnectionState (diff) | |
| download | discord.py-bd9821320ec2288a4c3cfada980a85e78630f94f.tar.xz discord.py-bd9821320ec2288a4c3cfada980a85e78630f94f.zip | |
Use a subclass rather than monkey-patching for the websocket
Diffstat (limited to 'discord/http.py')
| -rw-r--r-- | discord/http.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/discord/http.py b/discord/http.py index 4651490a..d91664c9 100644 --- a/discord/http.py +++ b/discord/http.py @@ -34,6 +34,7 @@ import weakref import aiohttp from .errors import HTTPException, Forbidden, NotFound, LoginFailure, GatewayNotFound +from .gateway import DiscordClientWebSocketResponse from . import __version__, utils log = logging.getLogger(__name__) @@ -113,7 +114,7 @@ class HTTPClient: def recreate(self): if self.__session.closed: - self.__session = aiohttp.ClientSession(connector=self.connector) + self.__session = aiohttp.ClientSession(connector=self.connector, ws_response_class=DiscordClientWebSocketResponse) async def ws_connect(self, url, *, compress=0): kwargs = { @@ -279,7 +280,7 @@ class HTTPClient: async def static_login(self, token, *, bot): # Necessary to get aiohttp to stop complaining about session creation - self.__session = aiohttp.ClientSession(connector=self.connector) + self.__session = aiohttp.ClientSession(connector=self.connector, ws_response_class=DiscordClientWebSocketResponse) old_token, old_bot = self.token, self.bot_token self._token(token, bot=bot) |