env.py

import os
import sys
from typing import Union

import dotenv

dotenv.load_dotenv()


def get_env_var(name: str, nullable_variable: bool = False) -> Union[str, None]:
    """
    get_env_var.

    Parameters
    ----------
    name : str
        The name of the environment variable to access.
    nullable_variable : bool
        Boolean for allowing null variables
    Returns
    -------
    value : str | None
        The value of the environment variable.
        If nullable_variable is True, the value will
        be None if the environment variable is empty.
    """
    try:
        value = os.getenv(name, None)
        if value is not None or nullable_variable:
            return value
        else:
            print(f"Value of {name} is empty. Exiting")
            sys.exit(1)
    except Exception as e:
        print(f"Could not get value of {name} due to {e}. Exiting")
        sys.exit(1)


# Discord


def get_discord_token() -> str:
    """get_discord_token."""
    return get_env_var("DISCORD_TOKEN")


def get_guild_id() -> int:
    """get_guild_id."""
    return int(get_env_var("GUILD_ID"))


def get_force_guild_id() -> Union[int, None]:
    guild_id = get_env_var("FORCE_GUILD_ID", True)
    return int(guild_id) if guild_id else None


def get_bot_config_path() -> Union[str, None]:
    return get_env_var("BOT_CONFIG_PATH", True)


# Twitch


def get_twitch_client() -> str:
    """get_twitch_client."""
    return get_env_var("TWITCH_CLIENT")


def get_twitch_secret() -> str:
    """get_twitch_secret."""
    return get_env_var("TWITCH_SECRET")


def get_twitch_user() -> str:
    """get_twitch_user."""
    return get_env_var("TWITCH_USER")


# YouTube


def get_yt_api_key() -> str:
    """get_yt_api_key."""
    return get_env_var("YT_API_KEY")


def get_yt_channel_id() -> str:
    """get_yt_channel_id."""
    return get_env_var("YT_CHANNEL_ID")


# Logging channel
def get_logging_channel_name() -> str:
    """get_logging_channel_name."""
    return get_env_var("LOGGING_CHANNEL_NAME")