MrChiBlog/config.py at master · mrchi/MrChiBlog · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
# coding=utf-8

import os


class Config:
    REPO_DIR = os.environ.get("REPO_DIR")
    REPO_BRANCH = "master"
    REPO_SSHKEY = os.environ.get("REPO_SSHKEY")
    WEBHOOK_TOKEN = os.environ.get("WEBHOOK_TOKEN")

    HMAC_KEY = os.environ.get("HMAC_KEY")

    SQLALCHEMY_TRACK_MODIFICATIONS = True

    @staticmethod
    def init_app(app):
        pass


class DevelopmentConfig(Config):
    DEBUG = True
    SQLALCHEMY_DATABASE_URI = os.environ.get("DEV_DATABASE_URI")
    REDIS_URL = os.environ.get("DEV_REDIS_URI") or "redis://127.0.0.1:6379/0"
    WHOOSH_BASE = os.environ.get("DEV_WHOOSH_BASE") or "dev_whoosh_idx"
    DINGTALK_ROBOT_TOKEN = os.environ.get("DEV_DINGTALK_TOKEN")
    RQ_REDIS_URL = os.environ.get("DEV_RQ_REDIS_URI") or "redis://127.0.0.1:6379/1"


class TestingConfig(Config):
    TESTING = True
    SQLALCHEMY_DATABASE_URI = os.environ.get("TEST_DATABASE_URI")
    REDIS_URL = os.environ.get("TEST_REDIS_URI")
    WHOOSH_BASE = os.environ.get("TEST_WHOOSH_BASE") or "test_whoosh_idx"
    DINGTALK_ROBOT_TOKEN = os.environ.get("TEST_DINGTALK_TOKEN")
    RQ_REDIS_URL = os.environ.get("TEST_RQ_REDIS_URI")
    RQ_ASYNC = False


class ProductionConfig(Config):
    SQLALCHEMY_DATABASE_URI = os.environ.get("PROD_DATABASE_URI")
    REDIS_URL = os.environ.get("PROD_REDIS_URI")
    WHOOSH_BASE = os.environ.get("PROD_WHOOSH_BASE") or "prod_whoosh_idx"
    DINGTALK_ROBOT_TOKEN = os.environ.get("PROD_DINGTALK_TOKEN")
    RQ_REDIS_URL = os.environ.get("PROD_RQ_REDIS_URI")


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