diff options
| author | Nadir Chowdhury <[email protected]> | 2021-02-25 02:26:51 +0000 |
|---|---|---|
| committer | GitHub <[email protected]> | 2021-02-24 21:26:51 -0500 |
| commit | 63ec23bac24ab62633bdb8fd19b93ecd3fddba7c (patch) | |
| tree | 14ad9433dddf4c056c292a07e2c0e74b79942cdd /discord/gateway.py | |
| parent | Fix NameError with invoked_parents (diff) | |
| download | discord.py-63ec23bac24ab62633bdb8fd19b93ecd3fddba7c.tar.xz discord.py-63ec23bac24ab62633bdb8fd19b93ecd3fddba7c.zip | |
Code optimisations and refactoring via Sourcery
Diffstat (limited to 'discord/gateway.py')
| -rw-r--r-- | discord/gateway.py | 13 |
1 files changed, 4 insertions, 9 deletions
diff --git a/discord/gateway.py b/discord/gateway.py index f13276a1..210a8822 100644 --- a/discord/gateway.py +++ b/discord/gateway.py @@ -422,16 +422,11 @@ class DiscordWebSocket: if type(msg) is bytes: self._buffer.extend(msg) - if len(msg) >= 4: - if msg[-4:] == b'\x00\x00\xff\xff': - msg = self._zlib.decompress(self._buffer) - msg = msg.decode('utf-8') - self._buffer = bytearray() - else: - return - else: + if len(msg) < 4 or msg[-4:] != b'\x00\x00\xff\xff': return - + msg = self._zlib.decompress(self._buffer) + msg = msg.decode('utf-8') + self._buffer = bytearray() msg = json.loads(msg) log.debug('For Shard ID %s: WebSocket Event: %s', self.shard_id, msg) |