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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ez-a-sync
-Version: 0.30.0
+Version: 0.30.1
 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.1}/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.1}/ez_a_sync.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ez-a-sync
-Version: 0.30.0
+Version: 0.30.1
 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