internet_ml/internet_ml/utils/config.py

from typing import Any, List, Tuple

# General
CONF_DEBUG: bool = True
# Google
GOOGLE_API_KEY: str = ""
GOOGLE_SEARCH_ENGINE_ID: str = ""
# NLP
NLP_CONF_MODE: str = "default"


class FullConfig:
    def __init__(self: Any) -> None:
        self.CONF_DEBUG: bool = True
        self.GOOGLE_API_KEY: str = ""
        self.GOOGLE_SEARCH_ENGINE_ID: str = ""
        self.NLP_CONF_MODE: str = "default"

    def general_config(self: Any, CONF_DEBUG: bool) -> None:
        self.CONF_DEBUG = CONF_DEBUG

    def google_config(
        self: Any, GOOGLE_API_KEY: str, GOOGLE_SEARCH_ENGINE_ID: str
    ) -> None:
        self.GOOGLE_API_KEY = GOOGLE_API_KEY
        self.GOOGLE_SEARCH_ENGINE_ID = GOOGLE_SEARCH_ENGINE_ID

    def NLP_config(self: Any, NLP_CONF_MODE: str = "default") -> None:
        if (
            NLP_CONF_MODE == "accuracy"
            or NLP_CONF_MODE == "speed"
            or NLP_CONF_MODE == "default"
        ):
            self.NLP_CONF_MODE = NLP_CONF_MODE


config = FullConfig()