<p>你使用环境变量的本能是正确的。但是,使用错误的数据库运行单元测试有一定的危险。此外,您可能不想对每个请求和您想使用的任何位置都使用<code>connect_db</code>。您可以使用您显式设置的配置目录和环境变量。这是迄今为止我想到的最好的。</p>
<pre><code>run.py
shell.py
config/__init__.py
config/test.py
config/postgres.py
...
main/__init__.py
main/someapp/__init__.py
main/someapp/models.py
...
main/tests/__init__.py
main/tests/testutils.py
</code></pre>
<p>因此,配置文件可能是:</p>
<pre><code># config/test.py
SQLALCHEMY_DATABASE_URI = "sqlite://"
</code></pre>
<p>以及</p>
<pre><code># config/postgres.py
SQLALCHEMY_DATABASE_URI = 'postgresql://user:pw@localhost/somedb'
</code></pre>
<p>因此,我可以在基本测试用例中显式地设置db:</p>
<pre><code>import os
from flask.ext.testing import TestCase
os.environ["DIAG_CONFIG_MODULE"] = "config.test"
from main import app, db
class SQLAlchemyTest(TestCase):
def create_app(self):
return app
def setUp(self):
db.create_all()
def tearDown(self):
db.session.remove()
db.drop_all()
</code></pre>
<p>然后,对我来说,<code>main/__init__.py</code>:</p>
<pre><code>import os
from flask import Flask, render_template, g
from flask.ext.sqlalchemy import SQLAlchemy
# by default, let's use a DB we don't care about
# but, we can override if we want
config_obj = os.environ.get("DIAG_CONFIG_MODULE", "config.test")
app = Flask(__name__)
app.config.from_object(config_obj)
db = SQLAlchemy(app)
@app.before_request
def before_request():
g.db = db
g.app = app
# ...
@app.route('/', methods=['GET'])
def get():
return render_template('home.html')
# ...
from main.someapp.api import mod as someappmod
app.register_blueprint(someappmod)
</code></pre>
<p>然后,在其他文件中,我知道要运行什么配置,可能:</p>
<pre><code># run.py
import os
os.environ["DIAG_CONFIG_MODULE"] = "config.postgres"
from main import app
app.run(debug=True)
</code></pre>
<p>以及</p>
<pre><code># shell.py
import os
os.environ["DIAG_CONFIG_MODULE"] = "config.postgres"
from main import app, db
from main.symdiag.models import *
from main.auth.models import *
print sorted(k for k in locals().keys() if not k.startswith("_"))
import IPython
IPython.embed()
</code></pre>
<p>也许。。迄今为止最好的:P</p>