From 7696a566e1b19b3a9367922ff5c03d286294079e Mon Sep 17 00:00:00 2001 From: Rapptz Date: Tue, 29 Dec 2015 01:21:10 -0500 Subject: [PATCH] Raise ClientException if an unexpected websocket close happens --- discord/client.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/discord/client.py b/discord/client.py index 0f8df2d5d..e77d3b71a 100644 --- a/discord/client.py +++ b/discord/client.py @@ -698,13 +698,17 @@ class Client: Raises ------- 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() while not self.is_closed: msg = yield from self.ws.recv() if msg is None: + if self.connection is None: + raise ClientException('Unexpected websocket closure received') + if self.ws.close_code == 1012: yield from self.redirect_websocket(self.gateway) continue