lagom.injection_context

Complexity

Total Complexity

7

Size/Duplication

Total Lines	39
Duplicated Lines	0 %

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	TemporaryInjectionContext.__exit__()	0	2	1
A	TemporaryInjectionContext.__enter__()	0	2	1
A	TemporaryInjectionContext.rebind()	0	2	1
A	TemporaryInjectionContext.__init__()	0	15	4

from typing import Optional, Callable

from .interfaces import ReadableContainer, ExtendableContainer


class TemporaryInjectionContext:
    """
    Context Manager wrapper around a container which applies the supplied function on
    __enter__
    """

    _base_container: ExtendableContainer
    _update_function: Optional[Callable[[ReadableContainer], ReadableContainer]] = None

    def __init__(
        self,
        container: ExtendableContainer,
        update_function: Optional[
            Callable[[ReadableContainer], ReadableContainer]
        ] = None,
    ):
        self._base_container = container
        self._update_function = update_function
        if self._update_function:
            self._build_temporary_container = lambda: self._update_function(
                self._base_container
            )
        else:
            self._build_temporary_container = lambda: self._base_container.clone()

    def __enter__(self) -> ReadableContainer:
        return self._build_temporary_container()

    def __exit__(self, exc_type, exc_val, exc_tb):
        pass

    def rebind(self, new_container):
        return TemporaryInjectionContext(new_container, self._update_function)