ansible/test/lib/ansible_test/_internal/cache.py

"""Cache for commonly shared data that is intended to be immutable."""
from __future__ import annotations

import typing as t

from .config import (
    CommonConfig,
)

TValue = t.TypeVar('TValue')


class CommonCache:
    """Common cache."""
    def __init__(self, args):  # type: (CommonConfig) -> None
        self.args = args

    def get(self, key, factory):  # type: (str, t.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, factory):  # type: (str, t.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]