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

Timeout when doing initial connection.

Browse Source
This commit is contained in:
Rapptz
2017-04-22 02:14:06 -04:00
parent b6ac856868
commit 3cfebc4605
3 changed files with 25 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
discord/client.py
View File

@ -342,17 +342,18 @@ class Client:
@asyncio.coroutine
def _connect(self):
self.ws = yield from DiscordWebSocket.from_client(self, shard_id=self.shard_id)
coro = DiscordWebSocket.from_client(self, shard_id=self.shard_id)
self.ws = yield from asyncio.wait_for(coro, timeout=180.0, loop=self.loop)
while True:
try:
yield from self.ws.poll_event()
except ResumeWebSocket as e:
log.info('Got a request to RESUME the websocket.')
self.ws = yield from DiscordWebSocket.from_client(self, shard_id=self.shard_id,
session=self.ws.session_id,
sequence=self.ws.sequence,
resume=True)
coro = DiscordWebSocket.from_client(self, shard_id=self.shard_id,
session=self.ws.session_id,
sequence=self.ws.sequence,
resume=True)
self.ws = yield from asyncio.wait_for(coro, timeout=180.0, loop=self.loop)
@asyncio.coroutine
def connect(self, *, reconnect=True):