matthew/enhanced-discord.py
1
0
Fork 0
Code Issues 4 Pull Requests 21 Projects Releases Wiki Activity

Raise ClientException if an unexpected websocket close happens

Browse Source
This commit is contained in:
Rapptz 2015-12-29 01:21:10 -05:00
parent 33dbd84398
commit 7696a566e1
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
discord/client.py
View File

@ -698,13 +698,17 @@ class Client:
Raises Raises
------- -------
ClientException ClientException
If this is called before :meth:`login` was invoked successfully. If this is called before :meth:`login` was invoked successfully
or when an unexpected closure of the websocket occurs.
""" """
yield from self._make_websocket() yield from self._make_websocket()
while not self.is_closed: while not self.is_closed:
msg = yield from self.ws.recv() msg = yield from self.ws.recv()
if msg is None: if msg is None:
if self.connection is None:
raise ClientException('Unexpected websocket closure received')
if self.ws.close_code == 1012: if self.ws.close_code == 1012:
yield from self.redirect_websocket(self.gateway) yield from self.redirect_websocket(self.gateway)
continue continue