PyCyBase 0.1.8.post4__cp313-cp313-win_amd64.whl

cbase/env.pxd

@@ -0,0 +1,8 @@
+
+cdef class EnvConfigContext:
+    cdef dict overrides
+    cdef dict originals
+
+    cdef void c_activate(self)
+
+    cdef void c_deactivate(self)

cbase/env.pyi

@@ -0,0 +1,48 @@
+from collections.abc import Callable
+from typing import Any
+
+
+class EnvConfigContext:
+    """Context manager for temporary environment configuration changes."""
+
+    def __init__(self, **kwargs) -> None:
+        """
+        Initialize the context with configuration changes.
+
+        Args:
+            **kwargs: Configuration key-value pairs to set temporarily
+        """
+        ...
+
+    def __enter__(self) -> EnvConfigContext:
+        """Enter the context, applying configuration changes."""
+        ...
+
+    def __exit__(self, exc_type: type[BaseException] | None, exc_value: BaseException | None, traceback: Any) -> None:
+        """Exit the context, reverting configuration changes."""
+        ...
+
+    def __call__(self, func: Callable[..., Any]) -> Callable[..., Any]:
+        """Decorator to apply the context to a function."""
+        ...
+
+    def __or__(self, other: EnvConfigContext) -> EnvConfigContext:
+        """
+        Combine two EnvConfigContext instances.
+
+        Args:
+            other: Another EnvConfigContext instance
+
+        Returns:
+            A new EnvConfigContext with combined configurations
+        """
+        ...
+
+    def __invert__(self) -> EnvConfigContext:
+        """
+        Invert the EnvConfigContext.
+
+        Returns:
+            A new EnvConfigContext that reverts the configurations set in the original.
+        """
+        ...

cbase/env.pyx

@@ -0,0 +1,46 @@
+
+cdef class EnvConfigContext:
+    def __cinit__(self):
+        self.overrides = {}
+        self.originals = {}
+
+    def __init__(self, dict overrides=None, **kwargs):
+        if overrides:
+            self.overrides.update(overrides)
+
+        if kwargs:
+            self.overrides.update(kwargs)
+
+    cdef void c_activate(self):
+        pass
+
+    cdef void c_deactivate(self):
+        pass
+
+    def __repr__(self):
+        return f'{self.__class__.__name__}({self.overrides!r})'
+
+    def __enter__(self):
+        self.c_activate()
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.c_deactivate()
+
+    def __or__(self, EnvConfigContext other):
+        if not isinstance(other, EnvConfigContext):
+            return NotImplemented
+        cdef dict merged_overrides = self.overrides | other.overrides
+        return self.__class__(**merged_overrides)
+
+    def __invert__(self):
+        cdef dict inverted_overrides = {k: not v if isinstance(v, bool) else v for k, v in self.overrides.items()}
+        return self.__class__(**inverted_overrides)
+
+    def __call__(self, func):
+        def wrapper(*args, **kwargs):
+            self.c_activate()
+            ret = func(*args, **kwargs)
+            self.c_deactivate()
+            return ret
+        return wrapper