| # |
| # Copyright BitBake Contributors |
| # |
| # SPDX-License-Identifier: GPL-2.0-only |
| # |
| import asyncio |
| import logging |
| |
| logger = logging.getLogger("asyncio.TaskGroup") |
| |
| |
| class TaskGroup(object): |
| def __init__(self): |
| self._tasks = [] |
| |
| def create_task(self, coro, **kwargs): |
| self._tasks.append(asyncio.create_task(coro, **kwargs)) |
| |
| async def __aenter__(self): |
| return self |
| |
| async def __aexit__(self, exc_type, exc, tb): |
| try: |
| if exc is None: |
| while self._tasks: |
| done, pending = await asyncio.wait( |
| self._tasks, return_when=asyncio.FIRST_COMPLETED |
| ) |
| self._tasks = pending |
| |
| for t in done: |
| try: |
| await t |
| except asyncio.CancelledError: |
| pass |
| |
| finally: |
| for t in self._tasks: |
| t.cancel() |
| try: |
| await t |
| except: |
| # Ignore exceptions |
| pass |
| |
| return False |
| |
| @classmethod |
| async def run(cls, *coros): |
| async with cls() as group: |
| for c in coros: |
| group.create_task(c) |
