dycw-utilities 0.149.7__tar.gz → 0.149.9__tar.gz

{dycw_utilities-0.149.7 → dycw_utilities-0.149.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dycw-utilities
-Version: 0.149.7
+Version: 0.149.9
 Author-email: Derek Wan <d.wan@icloud.com>
 License-File: LICENSE
 Requires-Python: >=3.12

{dycw_utilities-0.149.7 → dycw_utilities-0.149.9}/pyproject.toml

@@ -103,7 +103,7 @@ dependencies = [
 name = "dycw-utilities"
 readme = "README.md"
 requires-python = ">= 3.12"
-version = "0.149.7"
+version = "0.149.9"
 
 [project.entry-points.pytest11]
 pytest-randomly = "utilities.pytest_plugins.pytest_randomly"
@@ -136,7 +136,7 @@ test = [
 # bump-my-version
 [tool.bumpversion]
 allow_dirty = true
-current_version = "0.149.7"
+current_version = "0.149.9"
 
 [[tool.bumpversion.files]]
 filename = "src/utilities/__init__.py"

dycw_utilities-0.149.9/src/tests/test_contextvars.py

@@ -0,0 +1,37 @@
+from __future__ import annotations
+
+from contextvars import ContextVar
+
+from utilities.contextvars import (
+    _GLOBAL_BREAKPOINT,
+    global_breakpoint,
+    set_global_breakpoint,
+    yield_set_context,
+)
+from utilities.text import unique_str
+
+
+class TestGlobalBreakpoint:
+    def test_disabled(self) -> None:
+        global_breakpoint()
+
+    def test_set(self) -> None:
+        try:
+            set_global_breakpoint()
+        finally:
+            _ = _GLOBAL_BREAKPOINT.set(False)
+
+
+class TestYieldSetContext:
+    def test_disabled(self) -> None:
+        context = ContextVar(unique_str(), default=False)
+        assert not context.get()
+        with yield_set_context(context):
+            assert context.get()
+        assert not context.get()
+
+    def test_set(self) -> None:
+        try:
+            set_global_breakpoint()
+        finally:
+            _ = _GLOBAL_BREAKPOINT.set(False)

{dycw_utilities-0.149.7 → dycw_utilities-0.149.9}/src/utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.149.7"
+__version__ = "0.149.9"

{dycw_utilities-0.149.7 → dycw_utilities-0.149.9}/src/utilities/contextvars.py

@@ -1,6 +1,12 @@
 from __future__ import annotations
 
+from contextlib import contextmanager
 from contextvars import ContextVar
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from collections.abc import Iterator
+
 
 ##
 
@@ -19,4 +25,17 @@ def set_global_breakpoint() -> None:
     _ = _GLOBAL_BREAKPOINT.set(True)
 
 
-__all__ = ["global_breakpoint", "set_global_breakpoint"]
+##
+
+
+@contextmanager
+def yield_set_context(var: ContextVar[bool], /) -> Iterator[None]:
+    """Yield a context var as being set."""
+    token = var.set(True)
+    try:
+        yield
+    finally:
+        _ = var.reset(token)
+
+
+__all__ = ["global_breakpoint", "set_global_breakpoint", "yield_set_context"]

{dycw_utilities-0.149.7 → dycw_utilities-0.149.9}/src/utilities/pottery.py

@@ -2,6 +2,7 @@ from __future__ import annotations
 
 from contextlib import nullcontext, suppress
 from dataclasses import dataclass
+from functools import wraps
 from sys import maxsize
 from typing import TYPE_CHECKING, override
 
@@ -88,13 +89,23 @@ class CoroutineLooper:
     async def __call__[**P](
         self, func: Callable[P, Coro[None]], *args: P.args, **kwargs: P.kwargs
     ) -> bool:
-        def make_coro() -> Coro[None]:
-            return func(*args, **kwargs)
-
         return await loop_until_succeed(
-            make_coro, logger=self.logger, errors=ExtendUnlockedLock, sleep=self.sleep
+            lambda: self._make_coro(func, *args, **kwargs),
+            logger=self.logger,
+            errors=ExtendUnlockedLock,
+            sleep=self.sleep,
         )
 
+    def _make_coro[**P](
+        self, func: Callable[P, Coro[None]], /, *args: P.args, **kwargs: P.kwargs
+    ) -> Coro[None]:
+        @wraps(func)
+        async def wrapped() -> None:
+            await extend_lock(lock=self.lock)
+            return await func(*args, **kwargs)
+
+        return wrapped()
+
 
 ##
 

dycw_utilities-0.149.7/src/tests/test_contextvars.py

@@ -1,18 +0,0 @@
-from __future__ import annotations
-
-from utilities.contextvars import (
-    _GLOBAL_BREAKPOINT,
-    global_breakpoint,
-    set_global_breakpoint,
-)
-
-
-class TestGlobalBreakpoint:
-    def test_disabled(self) -> None:
-        global_breakpoint()
-
-    def test_set(self) -> None:
-        try:
-            set_global_breakpoint()
-        finally:
-            _ = _GLOBAL_BREAKPOINT.set(False)