Source code for poiesis.constants

"""Constants for the Poiesis.

Contains constants used throughout the Poiesis application.
Much more general than the constants in the core or api modules.
"""

import os
from dataclasses import dataclass
from functools import lru_cache
from typing import Literal, cast




@dataclass(frozen=True)
class PoesisConstants:
    """Constants for the Poiesis.

    Attributes:
        ENVIRONMENT: The environment in which the application is running.
        DATABASE: The database in which the application is running.
    """

    ENVIRONMENT: Literal["dev", "prod"] = cast(
        Literal["dev", "prod"], os.environ.get("POIESIS_ENV", "dev")
    )
    LOG_LEVEL: Literal["DEBUG", "INFO", "WARNING", "ERROR"] = cast(
        Literal["DEBUG", "INFO", "WARNING", "ERROR"], os.getenv("LOG_LEVEL", "INFO")
    )



    @dataclass(frozen=True)
    class Database:
        """Constants for the database.

        Attributes:
            MONGODB: The MongoDB database.
        """



        @dataclass(frozen=True)
        class MongoDB:
            """Constants for the MongoDB database.

            Attributes:
                TASK_COLLECTION: The name of the tasks collection.
                SERVICE_COLLECTION: The name of the services collection.
                DATABASE: The name of the database to use.
                MAX_POOL_SIZE: The maximum number of connections to the database.
                HOST: Host for the mango instance.
                PORT: Port for the mongo instance.
            """

            TASK_COLLECTION: str = "tasks"
            SERVICE_COLLECTION: str = "services"
            DATABASE: str = os.environ.get("MONGODB_DATABASE", "poiesis")
            MAX_POOL_SIZE: int = int(os.environ.get("MONGODB_MAX_POOL_SIZE", "10"))







@lru_cache
def get_poiesis_constants() -> PoesisConstants:
    """Get the Poiesis constants.

    Returns:
        PoesisConstants: The Poiesis constants.
    """
    return PoesisConstants()