diff options
Diffstat (limited to 'third_party/python/pip/pip/_internal/cli/command_context.py')
-rw-r--r-- | third_party/python/pip/pip/_internal/cli/command_context.py | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/third_party/python/pip/pip/_internal/cli/command_context.py b/third_party/python/pip/pip/_internal/cli/command_context.py new file mode 100644 index 0000000000..139995ac3f --- /dev/null +++ b/third_party/python/pip/pip/_internal/cli/command_context.py @@ -0,0 +1,27 @@ +from contextlib import ExitStack, contextmanager +from typing import ContextManager, Generator, TypeVar + +_T = TypeVar("_T", covariant=True) + + +class CommandContextMixIn: + def __init__(self) -> None: + super().__init__() + self._in_main_context = False + self._main_context = ExitStack() + + @contextmanager + def main_context(self) -> Generator[None, None, None]: + assert not self._in_main_context + + self._in_main_context = True + try: + with self._main_context: + yield + finally: + self._in_main_context = False + + def enter_context(self, context_provider: ContextManager[_T]) -> _T: + assert self._in_main_context + + return self._main_context.enter_context(context_provider) |