nocturn9x
/
giambio
mirror of https://github.com/nocturn9x/giambio.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Backup

This commit is contained in:
nocturn9x 2020-07-17 09:39:22 +00:00
parent 77268f63bf
commit 52ba52f4ca
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
giambio/_managers.py
View File

@ -10,6 +10,9 @@ class TaskManager(object):
"""Object constructor""" """Object constructor"""
self.scheduler = scheduler self.scheduler = scheduler
self.tasks = self.scheduler.tasks
self.asleep = self.scheduler.paused
self.event_waiting = self.scheduler.event_waiting
self.values = {} self.values = {}
async def __aenter__(self): async def __aenter__(self):
@ -22,9 +25,9 @@ class TaskManager(object):
errors = [] errors = []
for task in itertools.chain( for task in itertools.chain(
self.scheduler.tasks.copy(), self.tasks.copy(),
self.scheduler.paused.items(), self.asleep.items(),
self.scheduler.event_waiting.values(), self.event_waiting.values(),
): ):
await task.cancel() await task.cancel()
try: try:
@ -44,7 +47,7 @@ class TaskManager(object):
while True: while True:
tasks = itertools.chain( tasks = itertools.chain(
self.scheduler.tasks.copy(), self.scheduler.paused.items() self.tasks.copy(), self.asleep.items()
) )
for task in tasks: for task in tasks:
try: try: