Refactor to get_event_loop_and_check_if_loop_is_running()

Author: sveinugu

src/omnipy/compute/mixins/auto_async.py

@@ -1,9 +1,9 @@
 import asyncio
-from asyncio import AbstractEventLoop, get_event_loop
 from inspect import iscoroutinefunction
 from typing import Callable, cast, Coroutine
 
 from omnipy.api.protocols.private.compute.job import IsJobBase, IsPlainFuncArgJobBase
+from omnipy.util.helpers import get_event_loop_and_check_if_loop_is_running
 
 
 class AutoAsyncJobBaseMixin:
@@ -27,15 +27,9 @@ def _call_job(self, *args: object, **kwargs: object) -> object:
         job_func_is_coroutine: bool = iscoroutinefunction(job_func)
         super_call_job_func = cast(Callable[..., Coroutine], super_as_job_base._call_job)
 
-        loop_is_running: bool
-        try:
-            loop_is_running = get_event_loop().is_running()
-        except RuntimeError:
-            loop_is_running = False
-
+        loop, loop_is_running = get_event_loop_and_check_if_loop_is_running()
         if self.auto_async and job_func_is_coroutine and not self_as_job_base.in_flow_context:
-            if loop_is_running:
-                loop: AbstractEventLoop = get_event_loop()
+            if loop and loop_is_running:
                 res: object = loop.create_task(super_call_job_func(*args, **kwargs))
             else:
                 res = asyncio.run(super_call_job_func(*args, **kwargs))

src/omnipy/util/helpers.py

@@ -1,3 +1,4 @@
+import asyncio
 from collections import defaultdict, UserDict
 from collections.abc import Hashable, Iterable
 from copy import copy, deepcopy
@@ -796,3 +797,16 @@ def called_from_omnipy_tests() -> bool:
                 and 'omnipy/tests' in module.__file__:
             return True
     return False
+
+
+def get_event_loop_and_check_if_loop_is_running() -> tuple[asyncio.AbstractEventLoop | None, bool]:
+    loop_is_running: bool
+    loop: asyncio.AbstractEventLoop | None = None
+
+    try:
+        loop = asyncio.get_event_loop()
+        loop_is_running = loop.is_running()
+    except RuntimeError:
+        loop_is_running = False
+
+    return loop, loop_is_running