setup.py

import os
import logging
import asyncpg

import config


def logging_setup():
    # Discord
    discord_logger = logging.getLogger("discord")
    discord_logger.setLevel(logging.DEBUG)

    discord_file_handler = logging.FileHandler(
        filename="discord.log",
        encoding="utf-8",
        mode="w"
    )

    discord_console_handler = logging.StreamHandler()
    discord_console_handler.setLevel(logging.WARNING)

    discord_formatter = logging.Formatter(
        "%(asctime)s - %(levelname)s - %(name)s: %(message)s"
    )
    discord_file_handler.setFormatter(discord_formatter)
    discord_console_handler.setFormatter(discord_formatter)

    discord_logger.addHandler(discord_file_handler)
    discord_logger.addHandler(discord_console_handler)

    # Reddit (praw module)
    reddit_file_handler = logging.FileHandler(
        filename="reddit.log",
        encoding="utf-8",
        mode="w"
    )

    reddit_console_handler = logging.StreamHandler()
    reddit_console_handler.setLevel(logging.WARNING)

    praw_logger = logging.getLogger("praw")
    praw_logger.setLevel(logging.DEBUG)

    praw_logger.addHandler(reddit_file_handler)
    praw_logger.addHandler(reddit_console_handler)

    prawcore_logger = logging.getLogger("prawcore")
    prawcore_logger.setLevel(logging.DEBUG)

    prawcore_logger.addHandler(reddit_file_handler)
    prawcore_logger.addHandler(reddit_console_handler)


def check_environment():
    global on_heroku

    if "DYNO" in os.environ:
        on_heroku = True
    else:
        on_heroku = False


def handle_constants():
    if not on_heroku:
        choice = input(
            "Have any constants changed or NOT been set? (y/n) - "
        ).lower().strip(" ")

        if choice in ("y", "yes"):
            token_input = input(
                "Input Discord bot token: "
            ).strip(" ")
            client_id_input = input(
                "Input Reddit client ID: "
            ).strip(" ")
            client_secret_input = input(
                "Input Reddit client secret: "
            ).strip(" ")
            username_input = input(
                "Input Reddit username: "
            ).strip(" ")
            password_input = input(
                "Input Reddit password: "
            ).strip(" ")
            user_agent_input = input(
                "Input Reddit user agent: "
            ).strip(" ")

            # creates a file called ".env" and stores the constants
            with open(".env", "w") as file:
                file.write(
                    f"D_TOKEN={token_input}\n"
                    f"R_CLIENT_ID={client_id_input}\n"
                    f"R_CLIENT_SECRET={client_secret_input}\n"
                    f"R_USERNAME={username_input}\n"
                    f"R_PASSWORD={password_input}\n"
                    f"R_USER_AGENT=\"{user_agent_input}\"\n"
                )

            del token_input
            del client_id_input
            del client_secret_input
            del username_input
            del password_input
            del user_agent_input
        elif choice in ("n", "no"):
            pass
        else:
            print("Invalid input, try again.")
            handle_constants()

        from dotenv import load_dotenv

        load_dotenv()

    config.D_TOKEN = os.environ["D_TOKEN"]
    config.R_CLIENT_ID = os.environ["R_CLIENT_ID"]
    config.R_CLIENT_SECRET = os.environ["R_CLIENT_SECRET"]
    config.R_USERNAME = os.environ["R_USERNAME"]
    config.R_PASSWORD = os.environ["R_PASSWORD"]
    config.R_USER_AGENT = os.environ["R_USER_AGENT"]
    if on_heroku:
        config.DATABASE_URL = os.environ["DATABASE_URL"]


async def database_setup():
    if on_heroku:
        config.connection = await asyncpg.connect(
            config.DATABASE_URL,
            ssl="require"
        )
    else:
        # manually input arguments
        config.connection = await asyncpg.connect(
            user=None,
            password=None,
            host=None,
            port=None,
            database=None
        )


logging_setup()

on_heroku = None
check_environment()

handle_constants()