From 8a754e0858d922e955e71b253c139e071ecec432 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Sun, 28 Apr 2024 18:04:21 +0200 Subject: Adding upstream version 2.14.3. Signed-off-by: Daniel Baumann --- test/lib/ansible_test/_internal/cache.py | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100644 test/lib/ansible_test/_internal/cache.py (limited to 'test/lib/ansible_test/_internal/cache.py') diff --git a/test/lib/ansible_test/_internal/cache.py b/test/lib/ansible_test/_internal/cache.py new file mode 100644 index 0000000..3afe422 --- /dev/null +++ b/test/lib/ansible_test/_internal/cache.py @@ -0,0 +1,31 @@ +"""Cache for commonly shared data that is intended to be immutable.""" +from __future__ import annotations + +import collections.abc as c +import typing as t + +from .config import ( + CommonConfig, +) + +TValue = t.TypeVar('TValue') + + +class CommonCache: + """Common cache.""" + def __init__(self, args: CommonConfig) -> None: + self.args = args + + def get(self, key: str, factory: c.Callable[[], TValue]) -> TValue: + """Return the value from the cache identified by the given key, using the specified factory method if it is not found.""" + if key not in self.args.cache: + self.args.cache[key] = factory() + + return self.args.cache[key] + + def get_with_args(self, key: str, factory: c.Callable[[CommonConfig], TValue]) -> TValue: + """Return the value from the cache identified by the given key, using the specified factory method (which accepts args) if it is not found.""" + if key not in self.args.cache: + self.args.cache[key] = factory(self.args) + + return self.args.cache[key] -- cgit v1.2.3