config.py

# coding=utf-8

import os
from datetime import timedelta

basedir = os.path.abspath(os.path.dirname(__file__))


class Config:
    SECRET_KEY = os.environ.get('SECRET_KEY') or 'wibbly wobbly timey wimey'

    SQLALCHEMY_COMMIT_ON_TEARDOWN = True
    SQLALCHEMY_TRACK_MODIFICATIONS = True

    POLARIS_ADMIN = os.environ.get('POLARIS_ADMIN') or 'earrow.liu@gmail.com'
    POLARIS_RECORDS_PER_PAGE = 20
    POLARIS_TASKS_PER_PAGE = 20
    POLARIS_PROJECTS_PER_PAGE = 20
    POLARIS_SERVERS_PER_PAGE = 10

    @classmethod
    def init_app(cls, app):
        pass


class DevelopmentConfig(Config):
    DEBUG = False

    SQLALCHEMY_DATABASE_URI = 'sqlite:///' + os.path.join(basedir, 'data-dev.sqlite')

    JENKINS_HOST = ''
    JENKINS_USERNAME = ''
    JENKINS_PASSWORD = ''

    EMAIL_HOST = ''
    EMAIL_SENDER = ''
    EMAIL_SENDER_PASSWORD = ''

    CELERY_BROKER_URL = 'redis://localhost:6379'
    CELERY_RESULT_BACKEND = 'redis://localhost:6379'
    CELERY_REDIRECT_STDOUTS_LEVEL = 'info'
    CELERY_TASK_SERIALIZER = 'pickle'
    CELERY_ACCEPT_CONTENT = ['pickle']
    CELERYBEAT_SCHEDULE = {
                              'check_state': {
                                  'task': 'app.celery_tasks.check_state',
                                  'schedule': timedelta(seconds=300)
                              }
                          }

    @classmethod
    def init_app(cls, app):
        Config.init_app(app)

        import logging
        from flask.logging import default_handler

        app.logger.setLevel(logging.DEBUG)
        stream_handler = logging.StreamHandler()
        formatter = logging.Formatter('[%(asctime)s][%(levelname)s][%(funcName)s()] %(message)s')
        stream_handler.setFormatter(formatter)

        app.logger.removeHandler(default_handler)
        app.logger.addHandler(stream_handler)


class TestingConfig(Config):
    TESTING = True


class ProductionConfig(Config):
    pass


config = {
    'development': DevelopmentConfig,
    'testing': TestingConfig,
    'production': ProductionConfig,
    'default': DevelopmentConfig
}