Merge pull request #230 from aiokitchen/hotfix/threaded-instance-for-methods

Return Threaded instance for methods

Author: mosquito

aiomisc/__init__.py

@@ -1,7 +1,7 @@
 from . import io, log
 from ._context_vars import StrictContextVar
 from .aggregate import aggregate, aggregate_async
-from .backoff import asyncbackoff, asyncretry, Backoff, BackoffExecution
+from .backoff import Backoff, BackoffExecution, asyncbackoff, asyncretry
 from .circuit_breaker import CircuitBreaker, CircuitBroken, cutout
 from .context import Context, get_context
 from .counters import Statistic, get_statistics

aiomisc/backoff.py

@@ -1,9 +1,8 @@
 import asyncio
-
 from functools import wraps
 from typing import (
-    Any, Callable, Coroutine, Optional, Tuple, Type, TypeVar, Union, Generic,
-    ParamSpec
+    Any, Callable, Coroutine, Generic, Optional, ParamSpec, Tuple, Type,
+    TypeVar, Union,
 )
 
 from .counters import Statistic
@@ -46,7 +45,7 @@ def __init__(
         max_tries: Optional[int] = None,
         giveup: Optional[Callable[[Exception], bool]] = None,
         statistic_name: Optional[str] = None,
-        statistic_class: Type[BackoffStatistic] = BackoffStatistic
+        statistic_class: Type[BackoffStatistic] = BackoffStatistic,
     ):
         if not pause:
             pause = 0
@@ -78,7 +77,7 @@ def __init__(
 
     def prepare(
         self,
-        func: Callable[P, Coroutine[Any, Any, T]]
+        func: Callable[P, Coroutine[Any, Any, T]],
     ) -> "BackoffExecution[P, T]":
         return BackoffExecution(
             function=func,
@@ -95,14 +94,14 @@ async def execute(
         self,
         func: Callable[P, Coroutine[Any, Any, T]],
         *args: P.args,
-        **kwargs: P.kwargs
+        **kwargs: P.kwargs,
     ) -> T:
         execution = self.prepare(func)
         return await execution(*args, **kwargs)
 
     def __call__(
         self,
-        func: Callable[P, Coroutine[Any, Any, T]]
+        func: Callable[P, Coroutine[Any, Any, T]],
     ) -> Callable[P, Coroutine[Any, Any, T]]:
         if not asyncio.iscoroutinefunction(func):
             raise TypeError("Function must be a coroutine function")

aiomisc/compat.py

@@ -3,14 +3,15 @@
 import os
 import socket
 from typing import (
-    Any, Iterator, Optional, final, TypeAlias, ParamSpec,
-    Protocol
+    Any, Iterator, Optional, ParamSpec, Protocol, TypeAlias, final,
 )
 
 from ._context_vars import EVENT_LOOP
 
+
 log = logging.getLogger(__name__)
 
+
 try:
     from time import time_ns
 except ImportError:
@@ -20,6 +21,12 @@ def time_ns() -> int:
         return int(time() * 1000000000)
 
 
+try:
+    from typing import Concatenate
+except ImportError:
+    from typing_extensions import Concatenate
+
+
 class EntrypointProtocol(Protocol):
     @property
     def name(self) -> str:
@@ -98,6 +105,7 @@ def sock_set_reuseport(sock: socket.socket, reuse_port: bool) -> None:
 get_current_loop = EVENT_LOOP.get
 
 __all__ = (
+    "Concatenate",
     "EntrypointProtocol",
     "EventLoopMixin",
     "ParamSpec",

aiomisc/iterator_wrapper.py

@@ -8,8 +8,8 @@
 from time import time
 from types import TracebackType
 from typing import (
-    Any, AsyncIterator, Awaitable, Callable, Deque, Generator, NoReturn,
-    Optional, Type, TypeVar, Union, Generic, ParamSpec,
+    Any, AsyncIterator, Awaitable, Callable, Deque, Generator, Generic,
+    NoReturn, Optional, ParamSpec, Type, TypeVar, Union,
 )
 from weakref import finalize