asyncio-extensions 0.0.1__py3-none-any.whl

asyncio_extensions/__init__.py

@@ -0,0 +1,12 @@
+"""Utility functions and extensions for AsyncIO."""
+
+from .taskgroups import TaskGroup, TerminateTaskGroup, force_terminate_task_group
+from .utils import checkpoint, sleep_forever
+
+__all__ = [
+    "TaskGroup",
+    "TerminateTaskGroup",
+    "checkpoint",
+    "force_terminate_task_group",
+    "sleep_forever",
+]

asyncio_extensions/taskgroups.py

@@ -0,0 +1,60 @@
+"""Extensions to interact with TaskGroups."""
+
+import asyncio
+from collections.abc import Coroutine
+from contextlib import AbstractAsyncContextManager
+from contextvars import Context
+from types import TracebackType
+from typing import Any, Never, Self, TypeVar
+
+T = TypeVar("T")
+
+
+class TerminateTaskGroup(BaseException):
+    """Signal to terminate a task group.
+
+    Technically not an exception, so inherits from `BaseException`.
+    """
+
+
+async def force_terminate_task_group() -> Never:
+    """Force termination of a task group."""
+    raise TerminateTaskGroup
+
+
+class TaskGroup(AbstractAsyncContextManager["TaskGroup"]):
+    """A version of asyncio.TaskGroup with a cancel method."""
+
+    def __init__(self) -> None:  # noqa: D107
+        self.tasks = asyncio.TaskGroup()
+
+    async def __aenter__(self) -> Self:  # noqa: D105
+        await self.tasks.__aenter__()
+        return self
+
+    async def __aexit__(  # noqa: D105
+        self,
+        exc_type: type[BaseException] | None,
+        exc_value: BaseException | None,
+        traceback: TracebackType | None,
+    ) -> bool | None:
+        try:  # noqa: SIM105
+            await self.tasks.__aexit__(exc_type, exc_value, traceback)
+        except* TerminateTaskGroup:
+            pass
+
+        return None
+
+    def create_task(
+        self,
+        coro: Coroutine[Any, Any, T],
+        *,
+        name: str | None = None,
+        context: Context | None = None,
+    ) -> asyncio.Task[T]:
+        """Create a new task in this group and return it."""
+        return self.tasks.create_task(coro, name=name, context=context)
+
+    def cancel(self) -> None:
+        """Cancel any remaining task in the TaskGroup."""
+        self.create_task(force_terminate_task_group())

asyncio_extensions/utils.py

@@ -0,0 +1,27 @@
+"""Utility functions."""
+
+from collections.abc import Generator
+from typing import Any, Never
+
+
+class YieldToEventLoop:
+    """Helper class to give control back to the event loop."""
+
+    def __await__(self) -> Generator[None, Any, None]:
+        """Yield control to the event loop."""
+        yield
+
+
+async def checkpoint() -> None:
+    """
+    Give control back to the eventloop.
+
+    This has the same effect of and asyncio.sleep(0), but it is more semantic
+    """
+    return await YieldToEventLoop()
+
+
+async def sleep_forever() -> Never:
+    """Sleeps forever."""
+    while True:
+        await YieldToEventLoop()

asyncio_extensions-0.0.1.dist-info/METADATA

@@ -0,0 +1,47 @@
+Metadata-Version: 2.4
+Name: asyncio-extensions
+Version: 0.0.1
+Summary: Some (maybe) useful extensions for asyncio
+Project-URL: Documentation, https://github.com/hartungstenio/asyncio-extensions#readme
+Project-URL: Issues, https://github.com/hartungstenio/asyncio-extensions/issues
+Project-URL: Source, https://github.com/hartungstenio/asyncio-extensions
+Author-email: Christian Hartung <6785871+hartungstenio@users.noreply.github.com>
+License-Expression: MIT
+License-File: LICENSE.txt
+Keywords: asyncio,tasks
+Classifier: Development Status :: 4 - Beta
+Classifier: Framework :: AsyncIO
+Classifier: Intended Audience :: Developers
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Classifier: Programming Language :: Python :: Implementation :: CPython
+Classifier: Programming Language :: Python :: Implementation :: PyPy
+Classifier: Topic :: Software Development :: Libraries
+Classifier: Typing :: Typed
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+
+# asyncio-extensions
+
+[![PyPI - Version](https://img.shields.io/pypi/v/asyncio-extensions.svg)](https://pypi.org/project/asyncio-extensions)
+[![PyPI - Python Version](https://img.shields.io/pypi/pyversions/asyncio-extensions.svg)](https://pypi.org/project/asyncio-extensions)
+
+-----
+
+## Table of Contents
+
+- [Installation](#installation)
+- [License](#license)
+
+## Installation
+
+```console
+pip install asyncio-extensions
+```
+
+## License
+
+`asyncio-extensions` is distributed under the terms of the [MIT](https://spdx.org/licenses/MIT.html) license.

asyncio_extensions-0.0.1.dist-info/RECORD

@@ -0,0 +1,8 @@
+asyncio_extensions/__init__.py,sha256=TpsoAdtWockMsW1VRpoDL1hLb2i3T3zh8iTmM4ZZ28Y,311
+asyncio_extensions/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+asyncio_extensions/taskgroups.py,sha256=GBYrYBgpuHJQBdIaElsRgg2GOVjpR_R8Cstuy6usuxM,1728
+asyncio_extensions/utils.py,sha256=H3q1sqrZvw7owshrab65iXYQdQgArX_h7kFvKsyFjV0,618
+asyncio_extensions-0.0.1.dist-info/METADATA,sha256=G2jOAJzF4AtcixXTtCsI6CW-UjN3POBql-qwmKRUG5c,1712
+asyncio_extensions-0.0.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+asyncio_extensions-0.0.1.dist-info/licenses/LICENSE.txt,sha256=9MhvrEwVOMUhmjtwqlza75RnD32B8zRvqNygMxjI2YI,1131
+asyncio_extensions-0.0.1.dist-info/RECORD,,

asyncio_extensions-0.0.1.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.28.0
+Root-Is-Purelib: true
+Tag: py3-none-any

asyncio_extensions-0.0.1.dist-info/licenses/LICENSE.txt

@@ -0,0 +1,9 @@
+MIT License
+
+Copyright (c) 2025-present Christian Hartung <6785871+hartungstenio@users.noreply.github.com>
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.