offline_handlers.py

import json
from pathlib import Path
from typing import Protocol

from flagsmith.mappers import map_environment_document_to_context
from flagsmith.types import SDKEvaluationContext


class OfflineHandler(Protocol):
    def get_evaluation_context(self) -> SDKEvaluationContext: ...


class EvaluationContextLocalFileHandler:
    """
    Handler to load evaluation context from a local JSON file.
    The JSON file should contain the full evaluation context as per Flagsmith Engine's specification.

    JSON schema:
    https://raw.githubusercontent.com/Flagsmith/flagsmith/main/sdk/evaluation-context.json
    """

    def __init__(self, file_path: str) -> None:
        self.evaluation_context: SDKEvaluationContext = json.loads(
            Path(file_path).read_text(),
        )

    def get_evaluation_context(self) -> SDKEvaluationContext:
        return self.evaluation_context


class EnvironmentDocumentLocalFileHandler:
    """
    Handler to load evaluation context from a local JSON file containing the environment document.
    The JSON file should contain the environment document as returned by the Flagsmith API.

    API documentation:
    https://api.flagsmith.com/api/v1/docs/#/api/api_v1_environment-document_list
    """

    def __init__(self, file_path: str) -> None:
        self.evaluation_context: SDKEvaluationContext = (
            map_environment_document_to_context(
                json.loads(
                    Path(file_path).read_text(),
                ),
            )
        )

    def get_evaluation_context(self) -> SDKEvaluationContext:
        return self.evaluation_context


# For backward compatibility, use the old class name for
# the local file handler implementation dependant on the environment document.
LocalFileHandler = EnvironmentDocumentLocalFileHandler