dycw-utilities 0.166.30__py3-none-any.whl → 0.185.8__py3-none-any.whl

utilities/contextlib.py

@@ -1,6 +1,5 @@
 from __future__ import annotations
 
-import re
 from asyncio import create_task, get_event_loop
 from contextlib import (
     _AsyncGeneratorContextManager,
@@ -90,12 +89,13 @@ def enhanced_context_manager[**P, T_co](
             with gcm as value:
                 yield value
         finally:
-            _ = signal(SIGABRT, sigabrt0) if sigabrt else None
-            _ = signal(SIGFPE, sigfpe0) if sigfpe else None
-            _ = signal(SIGILL, sigill0) if sigill else None
-            _ = signal(SIGINT, sigint0) if sigint else None
-            _ = signal(SIGSEGV, sigsegv0) if sigsegv else None
-            _ = signal(SIGTERM, sigterm0) if sigterm else None
+            with _suppress_signal_error():
+                _ = signal(SIGABRT, sigabrt0) if sigabrt else None
+                _ = signal(SIGFPE, sigfpe0) if sigfpe else None
+                _ = signal(SIGILL, sigill0) if sigill else None
+                _ = signal(SIGINT, sigint0) if sigint else None
+                _ = signal(SIGSEGV, sigsegv0) if sigsegv else None
+                _ = signal(SIGTERM, sigterm0) if sigterm else None
 
     return wrapped
 
@@ -173,12 +173,13 @@ def enhanced_async_context_manager[**P, T_co](
             async with agcm as value:
                 yield value
         finally:
-            _ = signal(SIGABRT, sigabrt0) if sigabrt else None
-            _ = signal(SIGFPE, sigfpe0) if sigfpe else None
-            _ = signal(SIGILL, sigill0) if sigill else None
-            _ = signal(SIGINT, sigint0) if sigint else None
-            _ = signal(SIGSEGV, sigsegv0) if sigsegv else None
-            _ = signal(SIGTERM, sigterm0) if sigterm else None
+            with _suppress_signal_error():
+                _ = signal(SIGABRT, sigabrt0) if sigabrt else None
+                _ = signal(SIGFPE, sigfpe0) if sigfpe else None
+                _ = signal(SIGILL, sigill0) if sigill else None
+                _ = signal(SIGINT, sigint0) if sigint else None
+                _ = signal(SIGSEGV, sigsegv0) if sigsegv else None
+                _ = signal(SIGTERM, sigterm0) if sigterm else None
 
     return wrapped
 
@@ -191,7 +192,8 @@ def _swap_handler(
 ) -> _HANDLER:
     orig_handler = getsignal(signum)
     new_handler = _make_handler(signum, obj)
-    _ = signal(signum, new_handler)
+    with _suppress_signal_error():
+        _ = signal(signum, new_handler)
     return orig_handler
 
 
@@ -220,24 +222,15 @@ def _make_handler(
     return new_handler
 
 
-##
-
-
-_SUPER_OBJECT_HAS_NO_ATTRIBUTE = re.compile(r"'super' object has no attribute '\w+'")
-
-
 @contextmanager
-def suppress_super_object_attribute_error() -> Iterator[None]:
-    """Suppress the super() attribute error, for mix-ins."""
+def _suppress_signal_error() -> Iterator[None]:
     try:
         yield
-    except AttributeError as error:
-        if not _SUPER_OBJECT_HAS_NO_ATTRIBUTE.search(error.args[0]):
-            raise
+    except ValueError as error:
+        (msg,) = error.args
+        if msg == "signal only works in main thread of the main interpreter":
+            return
+        raise  # pragma: no cover
 
 
-__all__ = [
-    "enhanced_async_context_manager",
-    "enhanced_context_manager",
-    "suppress_super_object_attribute_error",
-]
+__all__ = ["enhanced_async_context_manager", "enhanced_context_manager"]

utilities/contextvars.py

@@ -1,15 +1,6 @@
 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
-
-
-##
-
 
 _GLOBAL_BREAKPOINT = ContextVar("GLOBAL_BREAKPOINT", default=False)
 
@@ -25,17 +16,4 @@ def set_global_breakpoint() -> None:
     _ = _GLOBAL_BREAKPOINT.set(True)
 
 
-##
-
-
-@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"]
+__all__ = ["global_breakpoint", "set_global_breakpoint"]