From 52ba52f4ca81e042ebb09aaa5b95841ef3a07e17 Mon Sep 17 00:00:00 2001 From: nocturn9x Date: Fri, 17 Jul 2020 09:39:22 +0000 Subject: [PATCH] Backup --- giambio/_managers.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/giambio/_managers.py b/giambio/_managers.py index d6e42fe..2cc63ce 100644 --- a/giambio/_managers.py +++ b/giambio/_managers.py @@ -10,6 +10,9 @@ class TaskManager(object): """Object constructor""" self.scheduler = scheduler + self.tasks = self.scheduler.tasks + self.asleep = self.scheduler.paused + self.event_waiting = self.scheduler.event_waiting self.values = {} async def __aenter__(self): @@ -22,9 +25,9 @@ class TaskManager(object): errors = [] for task in itertools.chain( - self.scheduler.tasks.copy(), - self.scheduler.paused.items(), - self.scheduler.event_waiting.values(), + self.tasks.copy(), + self.asleep.items(), + self.event_waiting.values(), ): await task.cancel() try: @@ -44,7 +47,7 @@ class TaskManager(object): while True: tasks = itertools.chain( - self.scheduler.tasks.copy(), self.scheduler.paused.items() + self.tasks.copy(), self.asleep.items() ) for task in tasks: try: