ez-a-sync 0.30.0__tar.gz → 0.30.2__tar.gz

{ez_a_sync-0.30.0 → ez_a_sync-0.30.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ez-a-sync
-Version: 0.30.0
+Version: 0.30.2
 Summary: A library that makes it easy to define objects that can be used for both sync and async use cases.
 Home-page: https://github.com/BobTheBuidler/a-sync
 Author: BobTheBuidler

{ez_a_sync-0.30.0 → ez_a_sync-0.30.2}/a_sync/debugging.py

@@ -1,6 +1,6 @@
 from asyncio import create_task, sleep
 from inspect import isasyncgenfunction
-from logging import DEBUG, getLogger
+from logging import DEBUG, Logger, getLogger
 from functools import wraps
 from time import time
 from typing import AsyncIterator, Awaitable, Callable, NoReturn, TypeVar, overload
@@ -17,46 +17,52 @@ __T = TypeVar("__T")
 __B = TypeVar("__B", bound=ASyncGenericBase)
 
 
+_FIVE_MINUTES = 300
+
 logger = getLogger("a_sync.debugging")
-__logger_is_enabled_for = logger.isEnabledFor
-__logger_log = logger._log
 
 
 @overload
 def stuck_coro_debugger(
     fn: Callable[Concatenate[__B, __P], AsyncIterator[__T]],
+    logger: Logger = logger,
+    interval: int = _FIVE_MINUTES,
 ) -> ASyncGeneratorFunction[__P, __T]: ...
 
 
 @overload
 def stuck_coro_debugger(
     fn: Callable[Concatenate[__B, __P], Awaitable[__T]],
+    logger: Logger = logger,
+    interval: int = _FIVE_MINUTES,
 ) -> ASyncBoundMethod[__B, __P, __T]: ...
 
 
 @overload
 def stuck_coro_debugger(
-    fn: Callable[Concatenate[__B, __P], __T],
+    fn: Callable[Concatenate[__B, __P], __T], logger: Logger = logger, interval: int = _FIVE_MINUTES
 ) -> ASyncBoundMethod[__B, __P, __T]: ...
 
 
 @overload
 def stuck_coro_debugger(
-    fn: Callable[__P, AsyncIterator[__T]],
+    fn: Callable[__P, AsyncIterator[__T]], logger: Logger = logger, interval: int = _FIVE_MINUTES
 ) -> Callable[__P, AsyncIterator[__T]]: ...
 
 
 @overload
-def stuck_coro_debugger(fn: Callable[__P, Awaitable[__T]]) -> Callable[__P, Awaitable[__T]]: ...
+def stuck_coro_debugger(
+    fn: Callable[__P, Awaitable[__T]], logger: Logger = logger, interval: int = _FIVE_MINUTES
+) -> Callable[__P, Awaitable[__T]]: ...
+
 
+def stuck_coro_debugger(fn, logger=logger, interval=_FIVE_MINUTES):
+    __logger_is_enabled_for = logger.isEnabledFor
 
-def stuck_coro_debugger(fn):
     if isasyncgenfunction(fn):
 
         @wraps(fn)
-        async def stuck_async_gen_wrap(
-            *args: __P.args, **kwargs: __P.kwargs
-        ) -> AsyncIterator[__T]:
+        async def stuck_async_gen_wrap(*args: __P.args, **kwargs: __P.kwargs) -> AsyncIterator[__T]:
             aiterator = fn(*args, **kwargs)
 
             if not __logger_is_enabled_for(DEBUG):
@@ -65,7 +71,7 @@ def stuck_coro_debugger(fn):
                 return
 
             task = create_task(
-                coro=_stuck_debug_task(fn, args, kwargs),
+                coro=_stuck_debug_task(logger, interval, fn, args, kwargs),
                 name="_stuck_debug_task",
             )
             try:
@@ -83,7 +89,7 @@ def stuck_coro_debugger(fn):
                 return await fn(*args, **kwargs)
 
             task = create_task(
-                coro=_stuck_debug_task(fn, args, kwargs),
+                coro=_stuck_debug_task(logger, interval, fn, args, kwargs),
                 name="_stuck_debug_task",
             )
             try:
@@ -96,18 +102,19 @@ def stuck_coro_debugger(fn):
 
 
 async def _stuck_debug_task(
-    fn: Callable[__P, __T], *args: __P.args, **kwargs: __P.kwargs
+    logger: Logger, interval: int, fn: Callable[__P, __T], *args: __P.args, **kwargs: __P.kwargs
 ) -> NoReturn:
     # sleep early so fast-running coros can exit early
-    await sleep(300)
+    await sleep(interval)
 
-    start = time() - 300
+    start = time() - interval
     module = fn.__module__
     name = fn.__name__
     formatted_args = tuple(map(str, args))
     formatted_kwargs = dict(zip(kwargs.keys(), map(str, kwargs.values())))
+    log = logger._log
     while True:
-        __logger_log(
+        log(
             DEBUG,
             "%s.%s still executing after %sm with args %s kwargs %s",
             (
@@ -118,7 +125,7 @@ async def _stuck_debug_task(
                 formatted_kwargs,
             ),
         )
-        await sleep(300)
+        await sleep(interval)
 
 
-__all__ = ["stuck_coro_debugger"]
+__all__ = ["stuck_coro_debugger"]

{ez_a_sync-0.30.0 → ez_a_sync-0.30.2}/a_sync/executor.py

@@ -155,7 +155,8 @@ class _AsyncExecutorMixin(concurrent.futures.Executor, _DebugDaemonMixin):
                 self.__super_submit(fn, *args, **kwargs)
             return None
 
-        fut = self._create_future()
+        loop = self._get_loop()
+        fut = loop.create_future()
         if self.sync_mode:
             try:
                 _set_fut_result(fut, fn(*args, **kwargs))
@@ -176,7 +177,7 @@ class _AsyncExecutorMixin(concurrent.futures.Executor, _DebugDaemonMixin):
             def _call_copy_future_state(cf_fut: "concurrent.futures.Future"):
                 if _fut_is_cancelled(fut):
                     return
-                self._call_soon_threadsafe(
+                loop.call_soon_threadsafe(
                     _copy_future_state,
                     cf_fut,
                     fut,
@@ -207,10 +208,7 @@ class _AsyncExecutorMixin(concurrent.futures.Executor, _DebugDaemonMixin):
 
     def __init_mixin__(self):
         self.sync_mode = self._max_workers == 0
-        loop = self._get_loop()
-        self._create_future = loop.create_future
         self.__super_submit = super().submit
-        self._call_soon_threadsafe = loop.call_soon_threadsafe
 
     async def _debug_daemon(self, fut: asyncio.Future, fn, *args, **kwargs) -> None:
         """