Make public exports of aiotools.taskgroup static-checker friendly (#72)

Author: achimnol

.github/workflows/timeline-check.yml

@@ -8,11 +8,11 @@ jobs:
     if: |
       !contains(github.event.pull_request.labels.*.name, 'skip:changelog')
     steps:
-    - uses: actions/checkout@v2
+    - uses: actions/checkout@v4
       with:
         fetch-depth: 0
     - name: Set up Python
-      uses: actions/setup-python@v3
+      uses: actions/setup-python@v5
       with:
         python-version: "3.10"
         cache: "pip"

changes/72.fix

@@ -0,0 +1 @@
+Fix static type checker's recognition of `aiotools.taskgroup` public exports, such as `aiotools.PersistentTaskGroup`

src/aiotools/taskgroup/__init__.py

@@ -2,34 +2,29 @@
 
 from .types import MultiError, TaskGroupError
 
+__all__ = [
+    "MultiError",
+    "TaskGroup",
+    "TaskGroupError",
+    "PersistentTaskGroup",
+    "as_completed_safe",
+    "persistent",
+    "persistent_compat",
+    "current_taskgroup",
+    "current_ptaskgroup",
+    "has_contextvars",
+]
+
 if hasattr(asyncio, "TaskGroup"):
     from . import persistent
     from .base import *  # noqa
     from .persistent import *  # noqa
 
-    __all__ = [
-        "MultiError",
-        "TaskGroup",
-        "TaskGroupError",
-        "current_taskgroup",
-        *persistent.__all__,
-    ]
+    has_contextvars = True
 else:
     from . import persistent_compat
     from .base_compat import *  # type: ignore  # noqa
     from .base_compat import has_contextvars
     from .persistent_compat import *  # type: ignore  # noqa
 
-    __all__ = [  # type: ignore  # noqa
-        "MultiError",
-        "TaskGroup",
-        "TaskGroupError",
-        *persistent_compat.__all__,
-    ]
-    if has_contextvars:
-        __all__.append("current_taskgroup")
-
-
 from .utils import as_completed_safe  # noqa
-
-__all__.append("as_completed_safe")