config.py

import os
from dotenv import load_dotenv

load_dotenv()


class Config:
    SQLALCHEMY_TRACK_MODIFICATIONS = False
    PROPAGATE_EXCEPTIONS = False
    AUTH = bool(os.environ.get('AUTH', 'true').lower() == 'true')

    SECRET_KEY = os.environ.get('SECRET_KEY')
    GMAIL_USER = os.environ.get('SMTP_EMAIL')
    GMAIL_PASS = os.environ.get('SMTP_PASS')


class LocalConfig(Config):
    DEBUG = True
    TESTING = False
    SQLALCHEMY_DATABASE_URI = os.environ.get('DB_URI')


class TestingConfig(Config):
    DEBUG = True
    TESTING = True
    PROPAGATE_EXCEPTIONS = True
    AUTH = True
    SQLALCHEMY_DATABASE_URI = os.environ.get('TEST_DB_URI')


class TestingDBConfig(Config):
    DEBUG = True
    TESTING = False
    SQLALCHEMY_DATABASE_URI = os.environ.get('TEST_DB_URI')


class LocalTestingConfig(Config):
    DEBUG = True
    TESTING = True
    PROPAGATE_EXCEPTIONS = True
    AUTH = True
    SQLALCHEMY_DATABASE_URI = 'sqlite:///:memory:'


class ProductionConfig(Config):
    DEBUG = False
    TESTING = False
    AUTH = True
    SQLALCHEMY_DATABASE_URI = os.environ.get('DB_URI')


config_by_name = {
    'local': LocalConfig,
    'debug': LocalConfig,
    'test': TestingConfig,
    'testdb': TestingDBConfig,
    'testlocal': LocalTestingConfig,
    'production': ProductionConfig
}

environment = os.environ.get('ENVIRONMENT')
config = config_by_name[environment]