40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
# Copyright (c) 2023 - 2025, AG2ai, Inc., AG2ai open-source projects maintainers and core contributors
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
#
|
|
# Portions derived from https://github.com/https://github.com/Lancetnik/FastDepends are under the MIT License.
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
from contextlib import contextmanager
|
|
from typing import Any, Callable, Dict, Iterator
|
|
|
|
|
|
class Provider:
|
|
dependency_overrides: Dict[Callable[..., Any], Callable[..., Any]]
|
|
|
|
def __init__(self) -> None:
|
|
self.dependency_overrides = {}
|
|
|
|
def clear(self) -> None:
|
|
self.dependency_overrides = {}
|
|
|
|
def override(
|
|
self,
|
|
original: Callable[..., Any],
|
|
override: Callable[..., Any],
|
|
) -> None:
|
|
self.dependency_overrides[original] = override
|
|
|
|
@contextmanager
|
|
def scope(
|
|
self,
|
|
original: Callable[..., Any],
|
|
override: Callable[..., Any],
|
|
) -> Iterator[None]:
|
|
self.dependency_overrides[original] = override
|
|
yield
|
|
self.dependency_overrides.pop(original, None)
|
|
|
|
|
|
dependency_provider = Provider()
|