config.py

# coding=utf-8

import os
import tempfile

basedir = os.path.abspath(os.path.dirname(__file__))
tempdir = tempfile.gettempdir()


class Config:
    SECRET_KEY = os.environ.get('SECRET_KEY') or 'it is a secret'
    SQLALCHEMY_TRACK_MODIFICATIONS = False

    @staticmethod
    def init_app(app):
        pass


class DevConfig(Config):
    DEBUG = True
    TESTING = False
    SQLALCHEMY_DATABASE_URI = os.environ.get('DEV_DATABASE_URI') \
        or 'sqlite:///' + os.path.join(basedir, 'data-dev.sqlite')


class TestConfig(Config):
    DEBUG = False
    TESTING = True
    SERVER_NAME = 'test.flaskr'
    SQLALCHEMY_DATABASE_URI = os.environ.get('TEST_DATABASE_URI') \
        or 'sqlite:///' + os.path.join(tempdir, 'data-test.sqlite')


class ProdConfig(Config):
    DEBUG = False
    TESTING = False
    SQLALCHEMY_DATABASE_URI = os.environ.get('PROD_DATABASE_URI') \
        or 'sqlite:///' + os.path.join(basedir, 'data-prod.sqlite')


config = {
    'development': DevConfig,
    'testing': TestConfig,
    'production': ProdConfig,
    'default': DevConfig,
}