haiway 0.11.1__py3-none-any.whl → 0.12.0__py3-none-any.whl

haiway/__init__.py

@@ -48,6 +48,7 @@ from haiway.utils import (
     async_always,
     async_noop,
     freeze,
+    getenv_base64,
     getenv_bool,
     getenv_float,
     getenv_int,
@@ -96,6 +97,7 @@ __all__ = [
     "ctx",
     "freeze",
     "frozenlist",
+    "getenv_base64",
     "getenv_bool",
     "getenv_float",
     "getenv_int",

haiway/context/access.py

@@ -23,7 +23,7 @@ from haiway.context.metrics import MetricsContext, MetricsHandler
 from haiway.context.state import StateContext
 from haiway.context.tasks import TaskGroupContext
 from haiway.state import State
-from haiway.utils import freeze, mimic_function
+from haiway.utils import mimic_function
 
 __all__ = [
     "ctx",
@@ -32,6 +32,16 @@ __all__ = [
 
 @final
 class ScopeContext:
+    __slots__ = (
+        "_disposables",
+        "_identifier",
+        "_logger_context",
+        "_metrics_context",
+        "_state",
+        "_state_context",
+        "_task_group_context",
+    )
+
     def __init__(
         self,
         label: str,
@@ -40,29 +50,77 @@ class ScopeContext:
         disposables: Disposables | None,
         metrics: MetricsHandler | None,
     ) -> None:
-        self._identifier: ScopeIdentifier = ScopeIdentifier.scope(label)
-        self._logger_context: LoggerContext = LoggerContext(
-            self._identifier,
-            logger=logger,
+        self._identifier: ScopeIdentifier
+        object.__setattr__(
+            self,
+            "_identifier",
+            ScopeIdentifier.scope(label),
+        )
+        self._logger_context: LoggerContext
+        object.__setattr__(
+            self,
+            "_logger_context",
+            LoggerContext(
+                self._identifier,
+                logger=logger,
+            ),
         )
-        self._task_group_context: TaskGroupContext = TaskGroupContext()
+        # postponing task group creation to include only when needed
+        self._task_group_context: TaskGroupContext
         # postponing state creation to include disposables state when prepared
         self._state_context: StateContext
-        self._state: tuple[State, ...] = state
-        self._disposables: Disposables | None = disposables
-        # pre-building metrics context to ensure nested context registering
-        self._metrics_context: MetricsContext = MetricsContext.scope(
-            self._identifier,
-            metrics=metrics,
+        self._state: tuple[State, ...]
+        object.__setattr__(
+            self,
+            "_state",
+            state,
+        )
+        self._disposables: Disposables | None
+        object.__setattr__(
+            self,
+            "_disposables",
+            disposables,
+        )
+        self._metrics_context: MetricsContext
+        object.__setattr__(
+            self,
+            "_metrics_context",
+            # pre-building metrics context to ensure nested context registering
+            MetricsContext.scope(
+                self._identifier,
+                metrics=metrics,
+            ),
         )
 
-        freeze(self)
+    def __setattr__(
+        self,
+        name: str,
+        value: Any,
+    ) -> Any:
+        raise AttributeError(
+            f"Can't modify immutable {self.__class__.__qualname__},"
+            f" attribute - '{name}' cannot be modified"
+        )
+
+    def __delattr__(
+        self,
+        name: str,
+    ) -> None:
+        raise AttributeError(
+            f"Can't modify immutable {self.__class__.__qualname__},"
+            f" attribute - '{name}' cannot be deleted"
+        )
 
     def __enter__(self) -> str:
         assert self._disposables is None, "Can't enter synchronous context with disposables"  # nosec: B101
         self._identifier.__enter__()
         self._logger_context.__enter__()
-        self._state_context = StateContext.updated(self._state)
+        # lazily initialize state
+        object.__setattr__(
+            self,
+            "_state_context",
+            StateContext.updated(self._state),
+        )
         self._state_context.__enter__()
         self._metrics_context.__enter__()
 
@@ -101,15 +159,33 @@ class ScopeContext:
     async def __aenter__(self) -> str:
         self._identifier.__enter__()
         self._logger_context.__enter__()
+        # lazily initialize group when needed
+        object.__setattr__(
+            self,
+            "_task_group_context",
+            TaskGroupContext(),
+        )
         await self._task_group_context.__aenter__()
 
+        # lazily initialize state to include disposables results
         if self._disposables is not None:
-            self._state_context = StateContext.updated(
-                (*self._state, *await self._disposables.__aenter__())
+            object.__setattr__(
+                self,
+                "_state_context",
+                StateContext.updated(
+                    (
+                        *self._state,
+                        *await self._disposables.__aenter__(),
+                    )
+                ),
             )
 
         else:
-            self._state_context = StateContext.updated(self._state)
+            object.__setattr__(
+                self,
+                "_state_context",
+                StateContext.updated(self._state),
+            )
 
         self._state_context.__enter__()
         self._metrics_context.__enter__()
@@ -200,6 +276,8 @@ class ScopeContext:
 
 @final
 class ctx:
+    __slots__ = ()
+
     @staticmethod
     def trace_id() -> str:
         """

haiway/context/disposables.py

@@ -3,10 +3,9 @@ from collections.abc import Iterable
 from contextlib import AbstractAsyncContextManager
 from itertools import chain
 from types import TracebackType
-from typing import final
+from typing import Any, final
 
 from haiway.state import State
-from haiway.utils import freeze
 
 __all__ = [
     "Disposable",
@@ -18,13 +17,37 @@ type Disposable = AbstractAsyncContextManager[Iterable[State] | State | None]
 
 @final
 class Disposables:
+    __slots__ = ("_disposables",)
+
     def __init__(
         self,
         *disposables: Disposable,
     ) -> None:
-        self._disposables: tuple[Disposable, ...] = disposables
+        self._disposables: tuple[Disposable, ...]
+        object.__setattr__(
+            self,
+            "_disposables",
+            disposables,
+        )
+
+    def __setattr__(
+        self,
+        name: str,
+        value: Any,
+    ) -> Any:
+        raise AttributeError(
+            f"Can't modify immutable {self.__class__.__qualname__},"
+            f" attribute - '{name}' cannot be modified"
+        )
 
-        freeze(self)
+    def __delattr__(
+        self,
+        name: str,
+    ) -> None:
+        raise AttributeError(
+            f"Can't modify immutable {self.__class__.__qualname__},"
+            f" attribute - '{name}' cannot be deleted"
+        )
 
     def __bool__(self) -> bool:
         return len(self._disposables) > 0

haiway/context/identifier.py

@@ -48,6 +48,15 @@ class ScopeIdentifier:
             trace_id=current.trace_id,
         )
 
+    __slots__ = (
+        "_token",
+        "label",
+        "parent_id",
+        "scope_id",
+        "trace_id",
+        "unique_name",
+    )
+
     def __init__(
         self,
         trace_id: str,
@@ -55,11 +64,61 @@ class ScopeIdentifier:
         scope_id: str,
         label: str,
     ) -> None:
-        self.trace_id: str = trace_id
-        self.parent_id: str = parent_id
-        self.scope_id: str = scope_id
-        self.label: str = label
-        self.unique_name: str = f"[{trace_id}] [{label}] [{scope_id}]"
+        self.trace_id: str
+        object.__setattr__(
+            self,
+            "trace_id",
+            trace_id,
+        )
+        self.parent_id: str
+        object.__setattr__(
+            self,
+            "parent_id",
+            parent_id,
+        )
+        self.scope_id: str
+        object.__setattr__(
+            self,
+            "scope_id",
+            scope_id,
+        )
+        self.label: str
+        object.__setattr__(
+            self,
+            "label",
+            label,
+        )
+        self.unique_name: str
+        object.__setattr__(
+            self,
+            "unique_name",
+            f"[{trace_id}] [{label}] [{scope_id}]",
+        )
+        self._token: Token[ScopeIdentifier] | None
+        object.__setattr__(
+            self,
+            "_token",
+            None,
+        )
+
+    def __setattr__(
+        self,
+        name: str,
+        value: Any,
+    ) -> Any:
+        raise AttributeError(
+            f"Can't modify immutable {self.__class__.__qualname__},"
+            f" attribute - '{name}' cannot be modified"
+        )
+
+    def __delattr__(
+        self,
+        name: str,
+    ) -> None:
+        raise AttributeError(
+            f"Can't modify immutable {self.__class__.__qualname__},"
+            f" attribute - '{name}' cannot be deleted"
+        )
 
     @property
     def is_root(self) -> bool:
@@ -78,8 +137,12 @@ class ScopeIdentifier:
         return hash(self.scope_id)
 
     def __enter__(self) -> None:
-        assert not hasattr(self, "_token"), "Context reentrance is not allowed"  # nosec: B101
-        self._token: Token[ScopeIdentifier] = ScopeIdentifier._context.set(self)
+        assert self._token is None, "Context reentrance is not allowed"  # nosec: B101
+        object.__setattr__(
+            self,
+            "_token",
+            ScopeIdentifier._context.set(self),
+        )
 
     def __exit__(
         self,
@@ -87,6 +150,10 @@ class ScopeIdentifier:
         exc_val: BaseException | None,
         exc_tb: TracebackType | None,
     ) -> None:
-        assert hasattr(self, "_token"), "Unbalanced context enter/exit"  # nosec: B101
+        assert self._token is not None, "Unbalanced context enter/exit"  # nosec: B101
         ScopeIdentifier._context.reset(self._token)
-        del self._token
+        object.__setattr__(
+            self,
+            "_token",
+            None,
+        )

haiway/context/logging.py

@@ -133,13 +133,61 @@ class LoggerContext:
                 exc_info=exception,
             )
 
+    __slots__ = (
+        "_entered",
+        "_logger",
+        "_prefix",
+        "_token",
+    )
+
     def __init__(
         self,
         scope: ScopeIdentifier,
         logger: Logger | None,
     ) -> None:
-        self._prefix: str = scope.unique_name
-        self._logger: Logger = logger or getLogger(name=scope.label)
+        self._prefix: str
+        object.__setattr__(
+            self,
+            "_prefix",
+            scope.unique_name,
+        )
+        self._logger: Logger
+        object.__setattr__(
+            self,
+            "_logger",
+            logger or getLogger(name=scope.label),
+        )
+        self._token: Token[LoggerContext] | None
+        object.__setattr__(
+            self,
+            "_token",
+            None,
+        )
+        self._entered: float | None
+        object.__setattr__(
+            self,
+            "_entered",
+            None,
+        )
+
+    def __setattr__(
+        self,
+        name: str,
+        value: Any,
+    ) -> Any:
+        raise AttributeError(
+            f"Can't modify immutable {self.__class__.__qualname__},"
+            f" attribute - '{name}' cannot be modified"
+        )
+
+    def __delattr__(
+        self,
+        name: str,
+    ) -> None:
+        raise AttributeError(
+            f"Can't modify immutable {self.__class__.__qualname__},"
+            f" attribute - '{name}' cannot be deleted"
+        )
 
     def log(
         self,
@@ -157,10 +205,18 @@ class LoggerContext:
         )
 
     def __enter__(self) -> None:
-        assert not hasattr(self, "_token"), "Context reentrance is not allowed"  # nosec: B101
-        assert not hasattr(self, "_entered"), "Context reentrance is not allowed"  # nosec: B101
-        self._entered: float = monotonic()
-        self._token: Token[LoggerContext] = LoggerContext._context.set(self)
+        assert self._token is None, "Context reentrance is not allowed"  # nosec: B101
+        assert self._entered is None, "Context reentrance is not allowed"  # nosec: B101
+        object.__setattr__(
+            self,
+            "_token",
+            LoggerContext._context.set(self),
+        )
+        object.__setattr__(
+            self,
+            "_entered",
+            monotonic(),
+        )
         self.log(DEBUG, "Entering context...")
 
     def __exit__(
@@ -169,8 +225,18 @@ class LoggerContext:
         exc_val: BaseException | None,
         exc_tb: TracebackType | None,
     ) -> None:
-        assert hasattr(self, "_token"), "Unbalanced context enter/exit"  # nosec: B101
+        assert (  # nosec: B101
+            self._token is not None and self._entered is not None
+        ), "Unbalanced context enter/exit"
         LoggerContext._context.reset(self._token)
-        del self._token
+        object.__setattr__(
+            self,
+            "_token",
+            None,
+        )
         self.log(DEBUG, f"...exiting context after {monotonic() - self._entered:.2f}s")
-        del self._entered
+        object.__setattr__(
+            self,
+            "_entered",
+            None,
+        )

haiway/context/metrics.py

@@ -1,6 +1,6 @@
 from contextvars import ContextVar, Token
 from types import TracebackType
-from typing import Protocol, Self, final, runtime_checkable
+from typing import Any, Protocol, Self, final, runtime_checkable
 
 from haiway.context.identifier import ScopeIdentifier
 from haiway.context.logging import LoggerContext
@@ -138,17 +138,62 @@ class MetricsContext:
                 exception=exc,
             )
 
+    __slots__ = (
+        "_metrics",
+        "_scope",
+        "_token",
+    )
+
     def __init__(
         self,
         scope: ScopeIdentifier,
         metrics: MetricsHandler | None,
     ) -> None:
-        self._scope: ScopeIdentifier = scope
-        self._metrics: MetricsHandler | None = metrics
+        self._scope: ScopeIdentifier
+        object.__setattr__(
+            self,
+            "_scope",
+            scope,
+        )
+        self._metrics: MetricsHandler | None
+        object.__setattr__(
+            self,
+            "_metrics",
+            metrics,
+        )
+        self._token: Token[MetricsContext] | None
+        object.__setattr__(
+            self,
+            "_token",
+            None,
+        )
+
+    def __setattr__(
+        self,
+        name: str,
+        value: Any,
+    ) -> Any:
+        raise AttributeError(
+            f"Can't modify immutable {self.__class__.__qualname__},"
+            f" attribute - '{name}' cannot be modified"
+        )
+
+    def __delattr__(
+        self,
+        name: str,
+    ) -> None:
+        raise AttributeError(
+            f"Can't modify immutable {self.__class__.__qualname__},"
+            f" attribute - '{name}' cannot be deleted"
+        )
 
     def __enter__(self) -> None:
-        assert not hasattr(self, "_token"), "Context reentrance is not allowed"  # nosec: B101
-        self._token: Token[MetricsContext] = MetricsContext._context.set(self)
+        assert self._token is None, "Context reentrance is not allowed"  # nosec: B101
+        object.__setattr__(
+            self,
+            "_token",
+            MetricsContext._context.set(self),
+        )
         if self._metrics is not None:
             self._metrics.enter_scope(self._scope)
 
@@ -158,8 +203,12 @@ class MetricsContext:
         exc_val: BaseException | None,
         exc_tb: TracebackType | None,
     ) -> None:
-        assert hasattr(self, "_token"), "Unbalanced context enter/exit"  # nosec: B101
+        assert self._token is not None, "Unbalanced context enter/exit"  # nosec: B101
         MetricsContext._context.reset(self._token)
-        del self._token
+        object.__setattr__(
+            self,
+            "_token",
+            None,
+        )
         if self._metrics is not None:
             self._metrics.exit_scope(self._scope)

haiway/context/state.py

@@ -1,11 +1,10 @@
-from collections.abc import Iterable
+from collections.abc import Iterable, MutableMapping
 from contextvars import ContextVar, Token
 from types import TracebackType
-from typing import Self, cast, final
+from typing import Any, Self, cast, final
 
 from haiway.context.types import MissingContext, MissingState
 from haiway.state import State
-from haiway.utils import freeze
 
 __all__ = [
     "ScopeState",
@@ -15,12 +14,37 @@ __all__ = [
 
 @final
 class ScopeState:
+    __slots__ = ("_state",)
+
     def __init__(
         self,
         state: Iterable[State],
     ) -> None:
-        self._state: dict[type[State], State] = {type(element): element for element in state}
-        freeze(self)
+        self._state: MutableMapping[type[State], State]
+        object.__setattr__(
+            self,
+            "_state",
+            {type(element): element for element in state},
+        )
+
+    def __setattr__(
+        self,
+        name: str,
+        value: Any,
+    ) -> Any:
+        raise AttributeError(
+            f"Can't modify immutable {self.__class__.__qualname__},"
+            f" attribute - '{name}' cannot be modified"
+        )
+
+    def __delattr__(
+        self,
+        name: str,
+    ) -> None:
+        raise AttributeError(
+            f"Can't modify immutable {self.__class__.__qualname__},"
+            f" attribute - '{name}' cannot be deleted"
+        )
 
     def state[StateType: State](
         self,
@@ -37,6 +61,7 @@ class ScopeState:
         else:
             try:
                 initialized: StateType = state()
+                # we would need a locking here in multithreaded environment
                 self._state[state] = initialized
                 return initialized
 
@@ -66,6 +91,11 @@ class ScopeState:
 class StateContext:
     _context = ContextVar[ScopeState]("StateContext")
 
+    __slots__ = (
+        "_state",
+        "_token",
+    )
+
     @classmethod
     def current[StateType: State](
         cls,
@@ -96,11 +126,45 @@ class StateContext:
         self,
         state: ScopeState,
     ) -> None:
-        self._state: ScopeState = state
+        self._state: ScopeState
+        object.__setattr__(
+            self,
+            "_state",
+            state,
+        )
+        self._token: Token[ScopeState] | None
+        object.__setattr__(
+            self,
+            "_token",
+            None,
+        )
+
+    def __setattr__(
+        self,
+        name: str,
+        value: Any,
+    ) -> Any:
+        raise AttributeError(
+            f"Can't modify immutable {self.__class__.__qualname__},"
+            f" attribute - '{name}' cannot be modified"
+        )
+
+    def __delattr__(
+        self,
+        name: str,
+    ) -> None:
+        raise AttributeError(
+            f"Can't modify immutable {self.__class__.__qualname__},"
+            f" attribute - '{name}' cannot be deleted"
+        )
 
     def __enter__(self) -> None:
-        assert not hasattr(self, "_token"), "Context reentrance is not allowed"  # nosec: B101
-        self._token: Token[ScopeState] = StateContext._context.set(self._state)
+        assert self._token is None, "Context reentrance is not allowed"  # nosec: B101
+        object.__setattr__(
+            self,
+            "_token",
+            StateContext._context.set(self._state),
+        )
 
     def __exit__(
         self,
@@ -108,6 +172,10 @@ class StateContext:
         exc_val: BaseException | None,
         exc_tb: TracebackType | None,
     ) -> None:
-        assert hasattr(self, "_token"), "Unbalanced context enter/exit"  # nosec: B101
+        assert self._token is not None, "Unbalanced context enter/exit"  # nosec: B101
         StateContext._context.reset(self._token)
-        del self._token
+        object.__setattr__(
+            self,
+            "_token",
+            None,
+        )