mirrors/discord.py
1
0
Fork 0
mirror of https://github.com/Rapptz/discord.py.git synced 2025-04-21 08:17:47 +00:00
Code Issues Projects Releases Wiki Activity

Re-raise exceptions from Client.start in Client.run.

Browse Source
This commit is contained in:
Rapptz 2017-05-20 12:58:29 -04:00
parent f3eb039c4b
commit 6e2ecbc167
1 changed files with 18 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
discord/client.py
View File

@ -454,17 +454,21 @@ class Client:
def _do_cleanup(self):
if self.loop.is_closed():
loop = self.loop
if loop.is_closed():
return # we're already cleaning up
self.loop.run_until_complete(self.close())
pending = asyncio.Task.all_tasks(loop=self.loop)
if loop.is_running():
loop.stop()
loop.run_until_complete(self.close())
pending = asyncio.Task.all_tasks(loop=loop)
if pending:
log.info('Cleaning up after %s tasks', len(pending))
gathered = asyncio.gather(*pending, loop=self.loop)
gathered = asyncio.gather(*pending, loop=loop)
try:
gathered.cancel()
self.loop.run_until_complete(gathered)
loop.run_until_complete(gathered)
# we want to retrieve any exceptions to make sure that
# they don't nag us about it being un-retrieved.
@ -472,7 +476,7 @@ class Client:
except:
pass
self.loop.close()
loop.close()
def run(self, *args, **kwargs):
"""A blocking call that abstracts away the `event loop`_
@ -506,25 +510,24 @@ class Client:
task = compat.create_task(self.start(*args, **kwargs), loop=loop)
def kill_loop_on_finish(fut):
try:
fut.result()
except:
pass # don't care
finally:
loop.stop()
def stop_loop_on_finish(fut):
loop.stop()
task.add_done_callback(kill_loop_on_finish)
task.add_done_callback(stop_loop_on_finish)
try:
loop.run_forever()
except KeyboardInterrupt:
pass
finally:
task.remove_done_callback(kill_loop_on_finish)
task.remove_done_callback(stop_loop_on_finish)
if is_windows:
self._do_cleanup()
if task.cancelled():
return None
return task.result()
# properties
def is_closed(self):