File: cache.py

package info (click to toggle)
ansible-core 2.14.18-0%2Bdeb12u2
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 29,072 kB
  • sloc: python: 172,173; cs: 4,367; sh: 3,898; makefile: 41; xml: 34
file content (32 lines) | stat: -rw-r--r-- 1,050 bytes parent folder | download
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
"""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]