config.py

import settings
import yaml

_config: dict = None


def get_config() -> dict:
    """This is to keep logic of accessing config in one place"""
    global _config
    with open(settings.CONSERVER_CONFIG_FILE) as file:
        _config = yaml.safe_load(file) or {}
    return _config


def get_worker_count() -> int:
    """Get the number of worker processes to spawn.
    
    Returns:
        int: Number of workers (minimum 1)
    """
    return max(1, settings.CONSERVER_WORKERS)


def is_parallel_storage_enabled() -> bool:
    """Check if parallel storage writes are enabled.
    
    Returns:
        bool: True if parallel storage is enabled
    """
    return settings.CONSERVER_PARALLEL_STORAGE


def get_start_method() -> str | None:
    """Get the multiprocessing start method.
    
    Returns:
        str | None: "fork", "spawn", "forkserver", or None for platform default
    """
    method = settings.CONSERVER_START_METHOD
    if method and method not in ("fork", "spawn", "forkserver"):
        raise ValueError(
            f"Invalid CONSERVER_START_METHOD: {method}. "
            "Must be 'fork', 'spawn', 'forkserver', or empty for default."
        )
    return method


class Configuration:
    @classmethod
    def get_config(cls) -> dict:
        return get_config()

    @classmethod
    def get_storages(cls) -> dict:
        config = cls.get_config()
        return config.get("storages", {})

    @classmethod
    def get_followers(cls) -> dict:
        config = cls.get_config()
        return config.get("followers", {})

    @classmethod
    def get_imports(cls) -> dict:
        config = cls.get_config()
        return config.get("imports", {})

    @classmethod
    def get_ingress_auth(cls) -> dict:
        """Get ingress-specific API key configuration.

        Returns:
            dict: Dictionary mapping ingress list names to their API keys.
                  Values can be either a single string (one API key) or
                  a list of strings (multiple API keys for the same ingress list).
        """
        config = cls.get_config()
        return config.get("ingress_auth", {})