feat: 通过环境变量区别加载不同的配置文件

Author: colorful3

app/api/cms/user.py

@@ -4,10 +4,9 @@
     :copyright: © 2019 by the Lin team.
     :license: MIT, see LICENSE for more details.
 """
-import os
 from operator import and_
 
-from flask import jsonify, current_app
+from flask import jsonify
 from flask_jwt_extended import create_access_token, get_jwt_identity, get_current_user, \
     create_refresh_token, verify_jwt_refresh_token_in_request
 from lin.core import manager, route_meta, Log

app/app.py

@@ -62,10 +62,16 @@ def log_response(resp):
         return resp
 
 
-def create_app(register_all=True):
+def create_app(register_all=True, environment='production'):
     app = Flask(__name__, static_folder='./assets')
-    app.config.from_object('app.config.setting')
-    app.config.from_object('app.config.secure')
+    app.config['ENV'] = environment
+    env = app.config.get('ENV')
+    if env == 'production':
+        app.config.from_object('app.config.setting.ProductionConfig')
+        app.config.from_object('app.config.secure.ProductionSecure')
+    elif env == 'development':
+        app.config.from_object('app.config.setting.DevelopmentConfig')
+        app.config.from_object('app.config.secure.DevelopmentSecure')
     app.config.from_object('app.config.log')
     if register_all:
         register_blueprints(app)

app/config/secure.py

@@ -4,8 +4,26 @@
 """
 
 # 安全性配置
-SQLALCHEMY_DATABASE_URI = 'mysql+cymysql://root:123456@localhost:3306/lin-cms'
+from app.config.setting import BaseConfig
 
-SQLALCHEMY_ECHO = False
 
-SECRET_KEY = '\x88W\xf09\x91\x07\x98\x89\x87\x96\xa0A\xc68\xf9\xecJJU\x17\xc5V\xbe\x8b\xef\xd7\xd8\xd3\xe6\x95*4'
+class DevelopmentSecure(BaseConfig):
+    """
+    开发环境安全性配置
+    """
+    SQLALCHEMY_DATABASE_URI = 'mysql+cymysql://root:123456@localhost:3306/lin-cms'
+
+    SQLALCHEMY_ECHO = False
+
+    SECRET_KEY = '\x88W\xf09\x91\x07\x98\x89\x87\x96\xa0A\xc68\xf9\xecJJU\x17\xc5V\xbe\x8b\xef\xd7\xd8\xd3\xe6\x95*4'
+
+
+class ProductionSecure(BaseConfig):
+    """
+    生产环境安全性配置
+    """
+    SQLALCHEMY_DATABASE_URI = 'mysql+cymysql://root:123456@localhost:3306/lin-cms'
+
+    SQLALCHEMY_ECHO = False
+
+    SECRET_KEY = '\x88W\xf09\x91\x07\x98\x89\x87\x96\xa0A\xc68\xf9\xecJJU\x17\xc5V\xbe\x8b\xef\xd7\xd8\xd3\xe6\x95*4'

app/config/setting.py

@@ -5,19 +5,54 @@
 
 from datetime import timedelta
 
-# 分页配置
-COUNT_DEFAULT = 10
-PAGE_DEFAULT = 0
-
-# 令牌配置
-JWT_ACCESS_TOKEN_EXPIRES = timedelta(hours=1)
-
-# 屏蔽 sql alchemy 的 FSADeprecationWarning
-SQLALCHEMY_TRACK_MODIFICATIONS = False
-
-# 插件模块暂时没有开启，以下配置可忽略
-# plugin config写在字典里面
-PLUGIN_PATH = {
-    'poem': {'path': 'app.plugins.poem', 'enable': True, 'version': '0.0.1', 'limit': 20},
-    'oss': {'path': 'app.plugins.oss', 'enable': True, 'version': '0.0.1', 'access_key_id': 'not complete', 'access_key_secret': 'not complete', 'endpoint': 'http://oss-cn-shenzhen.aliyuncs.com', 'bucket_name': 'not complete', 'upload_folder': 'app', 'allowed_extensions': ['jpg', 'gif', 'png', 'bmp']}
-}
+
+class BaseConfig(object):
+    """
+    基础配置
+    """
+    # 分页配置
+    COUNT_DEFAULT = 10
+    PAGE_DEFAULT = 0
+
+    # 屏蔽 sql alchemy 的 FSADeprecationWarning
+    SQLALCHEMY_TRACK_MODIFICATIONS = False
+
+
+class DevelopmentConfig(BaseConfig):
+    """
+    开发环境普通配置
+    """
+    DEBUG = True
+
+    # 令牌配置
+    JWT_ACCESS_TOKEN_EXPIRES = timedelta(hours=1)
+
+    # 插件模块暂时没有开启，以下配置可忽略
+    # plugin config写在字典里面
+    PLUGIN_PATH = {
+        'poem': {'path': 'app.plugins.poem', 'enable': True, 'version': '0.0.1', 'limit': 20},
+        'oss': {'path': 'app.plugins.oss', 'enable': True, 'version': '0.0.1', 'access_key_id': 'not complete',
+                'access_key_secret': 'not complete', 'endpoint': 'http://oss-cn-shenzhen.aliyuncs.com',
+                'bucket_name': 'not complete', 'upload_folder': 'app',
+                'allowed_extensions': ['jpg', 'gif', 'png', 'bmp']}
+    }
+
+
+class ProductionConfig(BaseConfig):
+    """
+    生产环境普通配置
+    """
+    DEBUG = False
+
+    # 令牌配置
+    JWT_ACCESS_TOKEN_EXPIRES = timedelta(hours=1)
+
+    # 插件模块暂时没有开启，以下配置可忽略
+    # plugin config写在字典里面
+    PLUGIN_PATH = {
+        'poem': {'path': 'app.plugins.poem', 'enable': True, 'version': '0.0.1', 'limit': 20},
+        'oss': {'path': 'app.plugins.oss', 'enable': True, 'version': '0.0.1', 'access_key_id': 'not complete',
+                'access_key_secret': 'not complete', 'endpoint': 'http://oss-cn-shenzhen.aliyuncs.com',
+                'bucket_name': 'not complete', 'upload_folder': 'app',
+                'allowed_extensions': ['jpg', 'gif', 'png', 'bmp']}
+    }

starter.py

@@ -5,7 +5,7 @@
 
 from app.app import create_app
 
-app = create_app()
+app = create_app(environment='development')
 
 
 @app.route('/', methods=['GET'], strict_slashes=False)