Source code for behave_kit.data.cache

"""Per-scope cache for loaded test data.

Avoids re-reading the same file multiple times within a scenario (or
feature, if requested). Keys are ``(path, scope)`` pairs.
"""

from __future__ import annotations

from pathlib import Path

from behave_kit._core.types import Scope


def _normalize_path(path: str | Path) -> str:
    """Return a forward-slash, normalized key regardless of input type."""
    return Path(path).as_posix()




class DataCache:
    """Cache of loaded data, keyed by ``(path, scope)``."""

    def __init__(self) -> None:
        self._entries: dict[tuple[str, Scope], object] = {}



    def get(self, path: str | Path, scope: Scope = Scope.SCENARIO) -> object | None:
        """Return the cached data for ``path`` at ``scope``, or ``None``.

        Args:
            path: File path or identifier used as the cache key.
            scope: Scope whose cached value should be returned.
        """
        return self._entries.get((_normalize_path(path), scope))




    def set(self, path: str | Path, scope: Scope, data: object) -> None:
        """Store ``data`` for ``path`` under ``scope``.

        Args:
            path: File path or identifier used as the cache key.
            scope: Scope at which the data should be cached.
            data: Value to cache.
        """
        self._entries[(_normalize_path(path), scope)] = data




    def invalidate(self, scope: Scope) -> None:
        """Remove every entry cached at ``scope``."""
        for key in [key for key in self._entries if key[1] == scope]:
            del self._entries[key]