Question

使用Flask的应用程序工厂模式时，我无法在数据库（PostgresSQL）中创建表。

我看过与Stackoverflow和Flask-SQLAlchemy源代码不同的示例。我的理解是，使用工厂应用程序模式I need to set up the so-called context before I can try creating the tables.，但是，当我创建上下文时，Flask应用程序的配置字典会重置，并且不会向前传播配置。

这是我的model.py

import datetime
from flask_sqlalchemy import SQLAlchemy

db = SQLAlchemy()


class MyModel(db.Model):

    id = db.Column(db.Integer, primary_key=True)
    some = db.Column(db.DateTime, nullable=False)
    random = db.Column(db.String, nullable=False)
    model = db.Column(db.String, nullable=False)
    fields = db.Column(db.Integer, nullable=False)

    def __init__(self, some: datetime.datetime, random: str, model: str,
                 fields: int) -> None:
        self.some = some
        self.random = random
        self.model = model
        self.fields = fields

    def __repr__(self):
        return f"""<MyModel(some={self.some}, random={self.random},
        model={self.model}, fields={self.fields})>"""

这是应用程序的__init__.py文件

import os

from flask import Flask
from flask_migrate import Migrate
from myapp.models import db

migrate = Migrate()


def create_app(config):

    app = Flask(__name__)
    app.config.from_object(config)
    db.init_app(app)
    migrate.init_app(app, db)
    from .models import MyModel
    with app.app_context():
        db.create_all()

    @app.route('/hello')
    def hello():
        return('Hello World!')

    return app

我还有一个main.py文件：

from myapp import create_app
from config import Config

app = create_app(Config)

if __name__ == "__main__":
    app.run(host='127.0.0.1', port=8080, debug=True)

根文件夹包含main.py和MyModule应用程序文件夹。此外，我在config.py文件中设置了Postgres实例和所需的配置常量：

import os
from dotenv import load_dotenv

basedir = os.path.abspath(__file__)
load_dotenv(os.path.join(basedir, '.env'))


class Config(object):
    DEBUG = False
    TESTING = False
    CSRF_ENABLED = True
    SECRET_KEY = os.environ.get('SECRET_KEY')
    SQLALCHEMY_DATABASE_URI = os.environ.get('SQLALCHEMY_DATABASE_URI')
    SQLALCHEMY_TRACK_MODIFICATIONS = False

我正在从.env文件中读取变量。

运行main.py时，出现以下错误：

(venv) $:project-name username$ python main.py
Traceback (most recent call last):
  File "main.py", line 4, in <module>
    app = create_app(Config)
  File "/Users/username/project-name/myapp/__init__.py", line 18, in create_app
    db.create_all()
  File "/Users/username/venv/lib/python3.6/site-packages/flask_sqlalchemy/__init__.py", line 1033, in create_all
    self._execute_for_all_tables(app, bind, 'create_all')
  File "/Users/username/venv/lib/python3.6/site-packages/flask_sqlalchemy/__init__.py", line 1025, in _execute_for_all_tables
    op(bind=self.get_engine(app, bind), **extra)
  File "/Users/username/venv/lib/python3.6/site-packages/flask_sqlalchemy/__init__.py", line 956, in get_engine
    return connector.get_engine()
  File "/Users/username/venv/lib/python3.6/site-packages/flask_sqlalchemy/__init__.py", line 560, in get_engine
    options = self.get_options(sa_url, echo)
  File "/Users/username/venv/lib/python3.6/site-packages/flask_sqlalchemy/__init__.py", line 575, in get_options
    self._sa.apply_driver_hacks(self._app, sa_url, options)
  File "/Users/username/venv/lib/python3.6/site-packages/flask_sqlalchemy/__init__.py", line 877, in apply_driver_hacks
    if sa_url.drivername.startswith('mysql'):
AttributeError: 'NoneType' object has no attribute 'drivername'

现在奇怪的是，当我在create_app函数中打印print(app.config)时，配置就位了，就像我希望的那样。例如SQLALCHEMY_DATABASE_URI=postgresql://testuser:testpassword@localhost:5432/testdb'。但是，当我在app.app_context（）循环内打印相同的信息时，SQLALCHEMY_DATABASE_URI=None（作为示例，其他键值对也被重置）。

我在这里想念什么？

Answer 1

您错误地加载了.env文件：

basedir = os.path.abspath(__file__)
load_dotenv(os.path.join(basedir, '.env'))

basedir是模块文件本身，而不是目录。因此，对于名为config.py的文件，basedir设置为/..absolutepath../config.py，而不是/..absolutepath../config.py。结果，您要求dotenv()加载文件/..absolutepath../config.py/.env，该文件将不存在。

您错过了一个os.path.dirname()通话：

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

您可以使用dotenv.find_dotenv()来完全避免此问题：

load_dotenv(find_dotenv())

使用其调用模块的__file__属性。

其他备注：

您正在使用Flask-Migrate管理模式，因此请勿调用db.create_all() 。而是使用Flask CLI和Flask Migrate命令（flask db init，flask db migrate和flask db upgrade创建迁移目录和初始迁移脚本，然后升级连接的数据库以使用最新的架构版本。
如果要通过环境变量配置数据库，请不要使用变量config对象。如果您仅从当前项目中导入config以获得适用于所有开发的默认配置，则使用Flask应用程序工厂会更容易。您随时可以根据需要通过app.config.from_obj() or app.config.from_envvar()使用可选参数和环境变量来加载更多配置。

带flask-sqlalchemy的create_app模式

1 个答案: