config.py

import logging.config
import os

from utils.log import Log
from utils.redis import RedisClient

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


class Config:
    SERVICE_NAME = "kong_tools"
    SECRET_KEY = os.environ.get('SECRET_KEY') or 'v*Ri4eJYAUAjxkQ%jCJ$0ty1@cJa6o%BH@dEPDr%fJFSmi9tT12O36hHLRZQXGrh'
    SQLALCHEMY_COMMIT_ON_TEARDOWN = True
    FLASKY_MAIL_SUBJECT_PREFIX = '[Flasky]'
    FLASKY_MAIL_SENDER = 'Flasky Admin <flasky@example.com>'
    FLASKY_ADMIN = os.environ.get('FLASKY_ADMIN')
    UPLOAD_DIR = os.path.join(basedir, 'uploads')
    if not os.path.exists(UPLOAD_DIR):
        os.makedirs(UPLOAD_DIR)

    # 构建日志相关信息
    LOG_DIR = os.path.join(basedir, 'logs')
    if not os.path.isdir(LOG_DIR):
        os.mkdir(LOG_DIR)

    log = Log(LOG_DIR, SERVICE_NAME)
    logging.config.dictConfig(log.log_config_dict)

    # 缓存相关数据
    REDIS_HOST = os.environ.get("REDIS_HOST", "127.0.0.1")
    REDIS_PORT = os.environ.get("REDIS_PORT", 6379)
    REDIS_PASS = os.environ.get("REDIS_PASS", "")
    REDIS_DB = os.environ.get("REDIS_DB", 0)

    # 高德地图，根据IP地址转换到城市的KEY
    AMAP_KEY = os.environ.get("AMAP_KEY", "")

    # ElasticSearch需要的日志
    ELASTICSEARCH_URL = os.environ.get("ELASTICSEARCH_URL", "127.0.0.1:9200")
    ELASTICSEARCH_INDEX = os.environ.get("ELASTICSEARCH_INDEX", "idx_kong_req_log")

    # 存储用户TOKEN信息KEY
    USER_KEY = os.environ.get("USER_KEY", "user_key:%s")

    @classmethod
    def init_app(cls, app):
        app.redis = RedisClient.get_client(cls.REDIS_HOST, cls.REDIS_PORT,
                                           cls.REDIS_DB, cls.REDIS_PASS)


class DevelopmentConfig(Config):
    DEBUG = True


class TestingConfig(Config):
    TESTING = True


class ProductionConfig(Config):
    SQLALCHEMY_TRACK_MODIFICATIONS = False


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