summaryrefslogtreecommitdiffstats
path: root/test/lib/ansible_test/_internal/cache.py
blob: 85fdbb1f1bfb070cd9135ebc345d1f10a776a952 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
"""Cache for commonly shared data that is intended to be immutable."""
from __future__ import (absolute_import, division, print_function)
__metaclass__ = type


class CommonCache:
    """Common cache."""
    def __init__(self, args):
        """
        :param args: CommonConfig
        """
        self.args = args

    def get(self, key, factory):
        """
        :param key: str
        :param factory: () -> any
        :rtype: any
        """
        if key not in self.args.cache:
            self.args.cache[key] = factory()

        return self.args.cache[key]

    def get_with_args(self, key, factory):
        """
        :param key: str
        :param factory: (CommonConfig) -> any
        :rtype: any
        """

        if key not in self.args.cache:
            self.args.cache[key] = factory(self.args)

        return self.args.cache[key]