Guard uses of the keep alive thread in case they're None

Fixes #5800
This commit is contained in:
Rapptz 2020-09-09 21:18:41 -04:00
parent e6017a2170
commit f8808dba9b

View File

@ -398,12 +398,14 @@ class DiscordWebSocket:
raise ReconnectWebSocket(self.shard_id) raise ReconnectWebSocket(self.shard_id)
if op == self.HEARTBEAT_ACK: if op == self.HEARTBEAT_ACK:
self._keep_alive.ack() if self._keep_alive:
self._keep_alive.ack()
return return
if op == self.HEARTBEAT: if op == self.HEARTBEAT:
beat = self._keep_alive.get_payload() if self._keep_alive:
await self.send_as_json(beat) beat = self._keep_alive.get_payload()
await self.send_as_json(beat)
return return
if op == self.HELLO: if op == self.HELLO: