Question

我正在使用flask配置，我这样配置我的应用程序：

app = Flask(__name__)
app.config.from_object('yourapplication.Config')

我的Config对象是：

class Config(object):
    DEBUG = True
    TESTING = False
    DB_USER = os.getenv("DB_USER", "db_user")
    DB_PASSWORD = os.getenv("DB_PASSWORD", "db_password")
    DB_HOST = os.getenv("DB_HOST", "localhost")  # 127.0.0.1"
    DB_PORT = os.getenv("DB_PORT", "5555")
    DB_SCHEMA = "my_schema"
    DB_DATABASE_NAME = "my_database"
    SQLALCHEMY_DATABASE_URI = "postgresql://{}:{}@{}:{}/{}".format(
        DB_USER, DB_PASSWORD, DB_HOST, DB_PORT, DB_DATABASE_NAME)
    SQLALCHEMY_TRACK_MODIFICATIONS = False

我像这样运行我的应用程序（使用flask-script）：

my_flask_app cmd1 arg1 arg2

我想做以下事情：

DEBUG=true DB_PORT=1234 my_flask_app cmd1 arg1 arg2

并覆盖2个配置参数。我想知道是否已经有一种方法可以从命令行覆盖配置参数，或者我是否应该去编写一些代码？

编辑：为了澄清，我有几个预设配置继承自'Config（object）'，如下所示：

class Dev(Config):
    pass

class Ua(Config):
    pass

class Prod(Config):
    DEBUG = False

我的应用默认为'Dev'配置。很容易想象一个场景，你想要运行'UA'配置但是覆盖端口，如下所示：

FLASK_ENV=UA DB_PORT=1234 my_flask_app cmd1 arg1 arg2

'FLASK_ENV'将在内部选择'UA'配置对象，但我希望DB_PORT应用于顶部。只是想知道是否有扩展或有人对此有任何想法。

Answer 1

有几种覆盖方法。

1）使用config.from_envvar + config.cfg。例如：

# test.py
import json

from flask import Flask, current_app


class Config(object):
    DEBUG = True
    TESTING = False
    DB_PORT = 1234


app = Flask(__name__)
app.config.from_object('test.Config')
app.config.from_envvar('YOURAPPLICATION_SETTINGS', silent=True)


@app.route('/')
def index():
    return json.dumps({
        'DEBUG': current_app.config['DEBUG'],
        'DB_PORT': current_app.config['DB_PORT'],
    })


if __name__ == '__main__':
    app.run()

config.cfg：

DB_PORT = 4321
DEBUG = False

运行我们的应用：python test.py并打开http://127.0.0.1:5000/。您将看到config的默认值：

{"DEBUG": true, "DB_PORT": 1234}

现在将config.cfg的路径添加到YOURAPPLICATION_SETTINGS变量中，然后重新启动应用。

export YOURAPPLICATION_SETTINGS={full_path_to}/config.cfg
python test.py
# open http://127.0.0.1:5000/ 
# you will see {"DEBUG": false, "DB_PORT": 4321}

2）与第一种方式类似，但没有.cfg。 test.py：

import json
import os

from flask import Flask, current_app


class Config(object):
    DEBUG = True
    TESTING = False
    DB_PORT = 1234


class ProductionConfig(Config):
    DEBUG = False
    DB_PORT = 4321


class TestingConfig(Config):
    TESTING = True


app = Flask(__name__)
app.config.from_object(os.environ.get('CONFIG_CLASS', 'test.Config'))
# route + run ...

如何检查：

python test.py # open http://127.0.0.1:5000/ - default config
# ProductionConfig config
export CONFIG_CLASS=test.ProductionConfig
python test.py # open http://127.0.0.1:5000/ - production config

3）在某些情况下，只有env变量就足够了。

但请注意，您可能遇到类型问题

import json
import os

from flask import Flask, current_app


class Config(object):
    DEBUG = True
    TESTING = False
    DB_PORT = 1234


app = Flask(__name__)
app.config.from_object('test.Config')
app.config.update({
    'DB_PORT': os.environ.get('DB_PORT', Config.DB_PORT)
    # you should to set type of variable here
    # DB_PORT will be string
    # 'DB_PORT': int(os.environ.get('DB_PORT', Config.DB_PORT))
})
# route + run ...

让我们检查一下：

export DB_PORT=4321
python test.py

打开http://127.0.0.1:5000/您会看到DB_PORT已更改但不是整数。因此，对于某些设置和特定情况，这是一个很好的解决方案。我认为1和2种方式更好。

希望这有帮助。

Answer 2

目前似乎没有实现此功能的标准方法或添加此有用行为的库。

如何在Flask配置中覆盖配置参数？

2 个答案: