feat: allow customizing the async task creator of the Autorun instance by overriding its _create_task method

Author: sassanh

.github/workflows/integration_delivery.yml

@@ -93,14 +93,10 @@ jobs:
       - uses: actions/checkout@v4
         name: Checkout
 
-      - name: Setup Python
-        uses: actions/setup-python@v5
-        with:
-          python-version: ${{ env.PYTHON_VERSION }}
-
       - name: Install uv
         uses: astral-sh/setup-uv@v3
         with:
+          python-version: ${{ env.PYTHON_VERSION }}
           enable-cache: true
 
       - name: Create virtualenv
@@ -159,9 +155,9 @@ jobs:
         run: |
           git fetch --prune --unshallow
           git fetch --depth=1 origin +refs/tags/*:refs/tags/*
-          echo "VERSION=$(uvx hatch version)" >> "$GITHUB_OUTPUT"
+          echo "VERSION=$(uvx hatch version)" >>"$GITHUB_OUTPUT"
           echo "VERSION=$(uvx hatch version)"
-          echo "NAME=$(uvx hatch project metadata | jq -r .name)" >> "$GITHUB_OUTPUT"
+          echo "NAME=$(uvx hatch project metadata | jq -r .name)" >>"$GITHUB_OUTPUT"
           echo "NAME=$(uvx hatch project metadata | jq -r .name)"
 
       - name: Extract Version from CHANGELOG.md
@@ -275,12 +271,12 @@ jobs:
       - name: Extract Changelog
         id: changelog
         run: |
-          perl -0777 -ne 'while (/## Version ${{ needs.build.outputs.version }}\n(\s*\n)*(.*?)(\s*\n)*## Version \d+\.\d+\.\d+\n/sg) {print "$2\n"}' CHANGELOG.md > CURRENT_CHANGELOG.md
+          perl -0777 -ne 'while (/## Version ${{ needs.build.outputs.version }}\n(\s*\n)*(.*?)(\s*\n)*## Version \d+\.\d+\.\d+\n/sg) {print "$2\n"}' CHANGELOG.md >CURRENT_CHANGELOG.md
           {
             echo "CONTENT<<EOF"
             cat CURRENT_CHANGELOG.md
             echo "EOF"
-          } >> "$GITHUB_OUTPUT"
+          } >>"$GITHUB_OUTPUT"
 
       - name: Release
         uses: softprops/action-gh-release@v2

CHANGELOG.md

@@ -7,6 +7,7 @@
 - feat: support `with_state` to be applied to methods of classes, not just functions
 - feat: support `view` to be applied to methods of classes, not just functions, it works for `autorun` too, but only when it is being called directly like a view
 - refactor: rename `_id` field of combine reducer state to `combine_reducers_id`
+- feat: allow customizing the async task creator of the `Autorun` instance by overriding its `_create_task` method
 
 ## Version 0.23.0
 

redux/autorun.py

@@ -143,19 +143,22 @@ def __init__(  # noqa: C901, PLR0912
             async def default_value_wrapper() -> ReturnType | None:
                 return options.default_value
 
-            create_task = self._store._create_task  # noqa: SLF001
             default_value = default_value_wrapper()
 
-            if create_task:
-                create_task(default_value)
+            self._create_task(default_value)
             self._latest_value: ReturnType = default_value
         else:
             self._latest_value: ReturnType = options.default_value
         self._subscriptions: set[
             Callable[[ReturnType], Any] | weakref.ref[Callable[[ReturnType], Any]]
         ] = set()
 
-        if self.check(store._state) and self._options.initial_call:  # noqa: SLF001
+        if (
+            store.with_state(lambda state: state, ignore_uninitialized_store=True)(
+                self.check,
+            )()
+            and self._options.initial_call
+        ):
             self._should_be_called = False
             self.call()
 
@@ -164,6 +167,11 @@ async def default_value_wrapper() -> ReturnType | None:
         else:
             self._unsubscribe = None
 
+    def _create_task(self: Autorun, coro: Coroutine[None, None, Any]) -> None:
+        """Create a task for the coroutine."""
+        if self._store.store_options.task_creator:
+            self._store.store_options.task_creator(coro)
+
     def react(
         self: Autorun,
         state: State,
@@ -267,10 +275,12 @@ def call(
                 *args,
                 **kwargs,
             )
-            create_task = self._store._create_task  # noqa: SLF001
             previous_value = self._latest_value
-            if iscoroutine(value) and create_task:
-                if self._options.auto_await is False:
+            if iscoroutine(value):
+                if (
+                    self._options.auto_await
+                    is False  # only explicit `False` disables auto-await, not `None`
+                ):
                     if (
                         self._latest_value is not None
                         and isinstance(self._latest_value, AwaitableWrapper)
@@ -280,7 +290,7 @@ def call(
                     self._latest_value = cast('ReturnType', AwaitableWrapper(value))
                 else:
                     self._latest_value = cast('ReturnType', None)
-                    create_task(value)
+                    self._create_task(value)
             else:
                 self._latest_value = value
             if self._latest_value is not previous_value:
@@ -300,8 +310,9 @@ def __call__(
         **kwargs: Args.kwargs,
     ) -> ReturnType:
         """Call the wrapped function with the current state of the store."""
-        state = self._store._state  # noqa: SLF001
-        self.check(state)
+        self._store.with_state(lambda state: state, ignore_uninitialized_store=True)(
+            self.check,
+        )()
         if self._should_be_called or args or kwargs or not self._options.memoization:
             self._should_be_called = False
             self.call(*args, **kwargs)
@@ -386,7 +397,7 @@ def __get__(
             ReturnType,
         ],
         obj: object | None,
-        type_: type | None = None,
+        _: type | None = None,
     ) -> Autorun[
         State,
         Action,

redux/main.py

@@ -73,7 +73,6 @@ def __init__(
         """Create a new store."""
         self.store_options = options or StoreOptions()
         self.reducer = reducer
-        self._create_task = self.store_options.task_creator
 
         self._action_middlewares = list(self.store_options.action_middlewares)
         self._event_middlewares = list(self.store_options.event_middlewares)
@@ -97,7 +96,7 @@ def __init__(
         self._workers = [
             self.store_options.side_effect_runner_class(
                 task_queue=self._event_handlers_queue,
-                create_task=self._create_task,
+                create_task=self.store_options.task_creator,
             )
             for _ in range(self.store_options.side_effect_threads)
         ]
@@ -122,12 +121,17 @@ def _call_listeners(self: Store[State, Action, Event], state: State) -> None:
         for listener_ in self._listeners.copy():
             if isinstance(listener_, weakref.ref):
                 listener = listener_()
-                assert listener is not None  # noqa: S101
+                if listener is None:
+                    msg = (
+                        'Listener has been garbage collected. '
+                        'Consider using `keep_ref=True` if it suits your use case.'
+                    )
+                    raise RuntimeError(msg)
             else:
                 listener = listener_
             result = listener(state)
-            if asyncio.iscoroutine(result) and self._create_task:
-                self._create_task(result)
+            if asyncio.iscoroutine(result) and self.store_options.task_creator:
+                self.store_options.task_creator(result)
 
     def _run_actions(self: Store[State, Action, Event]) -> None:
         while len(self._actions) > 0:
@@ -385,6 +389,8 @@ def view_decorator(
     def with_state(
         self: Store[State, Action, Event],
         selector: Callable[[State], SelectorOutput],
+        *,
+        ignore_uninitialized_store: bool = False,
     ) -> WithStateDecorator[SelectorOutput]:
         """Wrap a function, passing the result of the selector as its first argument.
 
@@ -425,6 +431,8 @@ def with_state_decorator(
         ):
             def wrapper(*args: Args.args, **kwargs: Args.kwargs) -> ReturnType:
                 if self._state is None:
+                    if ignore_uninitialized_store:
+                        return cast('ReturnType', None)
                     msg = 'Store has not been initialized yet.'
                     raise RuntimeError(msg)
                 return call_func(func, [selector(self._state)], *args, **kwargs)