lbry-sdk/lbry/wallet/tasks.py

34 lines
796 B
Python
Raw Permalink Normal View History

2018-12-07 02:03:22 +01:00
from asyncio import Event, get_event_loop
class TaskGroup:
def __init__(self, loop=None):
self._loop = loop or get_event_loop()
self._tasks = set()
self.done = Event()
self.started = Event()
2018-12-07 02:03:22 +01:00
2020-03-24 04:54:55 +01:00
def __len__(self):
return len(self._tasks)
2018-12-07 02:03:22 +01:00
def add(self, coro):
task = self._loop.create_task(coro)
self._tasks.add(task)
self.started.set()
2018-12-07 02:03:22 +01:00
self.done.clear()
task.add_done_callback(self._remove)
return task
def _remove(self, task):
self._tasks.remove(task)
if len(self._tasks) < 1:
self.done.set()
self.started.clear()
2018-12-07 02:03:22 +01:00
def cancel(self):
for task in self._tasks:
task.cancel()
self.done.set()
self.started.clear()