ez-a-sync 0.22.14__py3-none-any.whl → 0.22.15__py3-none-any.whl

a_sync/primitives/locks/semaphore.py

@@ -1,3 +1,8 @@
+"""
+This module provides various semaphore implementations, including a debug-enabled semaphore,
+a dummy semaphore that does nothing, and a threadsafe semaphore for use in multi-threaded applications.
+"""
+
 import asyncio
 import functools
 import logging
@@ -10,17 +15,19 @@ from a_sync.primitives._debug import _DebugDaemonMixin
 
 logger = logging.getLogger(__name__)
 
+
 class Semaphore(asyncio.Semaphore, _DebugDaemonMixin):
     """
     A semaphore with additional debugging capabilities.
-    
-    This semaphore includes debug logging.
-    
-    Also, it can be used to decorate coroutine functions so you can rewrite this pattern:
+
+    This semaphore includes debug logging and can be used to decorate coroutine functions.
+    It allows rewriting the pattern of acquiring a semaphore within a coroutine using a decorator.
+
+    So you can write this pattern:
 
     ```
     semaphore = Semaphore(5)
-    
+
     async def limited():
         async with semaphore:
             return 1
@@ -37,91 +44,67 @@ class Semaphore(asyncio.Semaphore, _DebugDaemonMixin):
         return 1
     ```
     """
+
     if sys.version_info >= (3, 10):
         __slots__ = "name", "_value", "_waiters", "_decorated"
     else:
         __slots__ = "name", "_value", "_waiters", "_loop", "_decorated"
-        
+
     def __init__(self, value: int, name=None, **kwargs) -> None:
         """
         Initialize the semaphore with a given value and optional name for debugging.
-        
+
         Args:
             value: The initial value for the semaphore.
             name (optional): An optional name used only to provide useful context in debug logs.
         """
         super().__init__(value, **kwargs)
-        self.name = name or self.__origin__ if hasattr(self, '__origin__') else None
+        self.name = name or self.__origin__ if hasattr(self, "__origin__") else None
         self._decorated: Set[str] = set()
-    
-    # Dank new functionality
 
     def __call__(self, fn: CoroFn[P, T]) -> CoroFn[P, T]:
         """
-        Convenient decorator method to wrap coroutine functions with the semaphore so you can rewrite this pattern:
+        Decorator method to wrap coroutine functions with the semaphore.
 
-        ```
-        semaphore = Semaphore(5)
-        
-        async def limited():
-            async with semaphore:
-                return 1
-
-        ```
+        This allows rewriting the pattern of acquiring a semaphore within a coroutine using a decorator.
 
-        like this:
-
-        ```
-        semaphore = Semaphore(5)
+        Example:
+            semaphore = Semaphore(5)
 
-        @semaphore
-        async def limited():
-            return 1
-        ```
+            @semaphore
+            async def limited():
+                return 1
         """
         return self.decorate(fn)  # type: ignore [arg-type, return-value]
-    
+
     def __repr__(self) -> str:
         representation = f"<{self.__class__.__name__} name={self.name} value={self._value} waiters={len(self)}>"
         if self._decorated:
             representation = f"{representation[:-1]} decorates={self._decorated}"
         return representation
-    
+
     def __len__(self) -> int:
         return len(self._waiters) if self._waiters else 0
-    
+
     def decorate(self, fn: CoroFn[P, T]) -> CoroFn[P, T]:
         """
         Wrap a coroutine function to ensure it runs with the semaphore.
-        
-        Example:
-            Now you can rewrite this pattern:
-
-            ```
-            semaphore = Semaphore(5)
-            
-            async def limited():
-                async with semaphore:
-                    return 1
 
-            ```
-
-            like this:
-
-            ```
+        Example:
             semaphore = Semaphore(5)
 
             @semaphore
             async def limited():
                 return 1
-            ```
         """
         if not asyncio.iscoroutinefunction(fn):
             raise TypeError(f"{fn} must be a coroutine function")
+
         @functools.wraps(fn)
         async def semaphore_wrapper(*args: P.args, **kwargs: P.kwargs) -> T:
             async with self:
                 return await fn(*args, **kwargs)
+
         self._decorated.add(f"{fn.__module__}.{fn.__name__}")
         return semaphore_wrapper
 
@@ -129,79 +112,97 @@ class Semaphore(asyncio.Semaphore, _DebugDaemonMixin):
         if self._value <= 0:
             self._ensure_debug_daemon()
         return await super().acquire()
-    
-    # Everything below just adds some debug logs
+
     async def _debug_daemon(self) -> None:
         """
         Daemon coroutine (runs in a background task) which will emit a debug log every minute while the semaphore has waiters.
         """
         while self._waiters:
             await asyncio.sleep(60)
-            self.logger.debug(f"{self} has {len(self)} waiters for any of: {self._decorated}")
-          
-        
+            self.logger.debug(
+                f"{self} has {len(self)} waiters for any of: {self._decorated}"
+            )
+
+
 class DummySemaphore(asyncio.Semaphore):
     """
     A dummy semaphore that implements the standard :class:`asyncio.Semaphore` API but does nothing.
+
+    This class is useful for scenarios where a semaphore interface is required but no actual synchronization is needed.
     """
 
     __slots__ = "name", "_value"
-    
+
     def __init__(self, name: Optional[str] = None):
+        """
+        Initialize the dummy semaphore with an optional name.
+
+        Args:
+            name (optional): An optional name for the dummy semaphore.
+        """
         self.name = name
         self._value = 0
-    
+
     def __repr__(self) -> str:
         return f"<{self.__class__.__name__} name={self.name}>"
-    
+
     async def acquire(self) -> Literal[True]:
         return True
-    
-    def release(self) -> None:
-        ...
-    
-    async def __aenter__(self):
-        ...
-    
-    async def __aexit__(self, *args):
-        ...
-        
+
+    def release(self) -> None: ...
+
+    async def __aenter__(self): ...
+
+    async def __aexit__(self, *args): ...
+
 
 class ThreadsafeSemaphore(Semaphore):
     """
-    While its a bit weird to run multiple event loops, sometimes either you or a lib you're using must do so. 
-    When in use in threaded applications, this semaphore will not work as intended but at least your program will function.
-    You may need to reduce the semaphore value for multi-threaded applications.
-    
-    # TL;DR it's a janky fix for an edge case problem and will otherwise function as a normal a_sync.Semaphore (which is just an asyncio.Semaphore with extra bells and whistles).
+    A semaphore that works in a multi-threaded environment.
+
+    This semaphore ensures that the program functions correctly even when used with multiple event loops.
+    It provides a workaround for edge cases involving multiple threads and event loops.
     """
+
     __slots__ = "semaphores", "dummy"
-    
+
     def __init__(self, value: Optional[int], name: Optional[str] = None) -> None:
+        """
+        Initialize the threadsafe semaphore with a given value and optional name.
+
+        Args:
+            value: The initial value for the semaphore, should be an integer.
+            name (optional): An optional name for the semaphore.
+        """
         assert isinstance(value, int), f"{value} should be an integer."
         super().__init__(value, name=name)
         self.semaphores: DefaultDict[Thread, Semaphore] = defaultdict(lambda: Semaphore(value, name=self.name))  # type: ignore [arg-type]
         self.dummy = DummySemaphore(name=name)
-    
+
     def __len__(self) -> int:
         return sum(len(sem._waiters) for sem in self.semaphores.values())
-    
+
     @functools.cached_property
     def use_dummy(self) -> bool:
+        """
+        Determine whether to use a dummy semaphore.
+
+        Returns:
+            True if the semaphore value is None, indicating the use of a dummy semaphore.
+        """
         return self._value is None
-    
+
     @property
     def semaphore(self) -> Semaphore:
         """
         Returns the appropriate semaphore for the current thread.
-        
+
         NOTE: We can't cache this property because we need to check the current thread every time we access it.
         """
         return self.dummy if self.use_dummy else self.semaphores[current_thread()]
-    
+
     async def __aenter__(self):
         await self.semaphore.acquire()
-    
+
     async def __aexit__(self, *args):
         self.semaphore.release()
-