python - 使用 flask-sqlalchemy 创建应用程序模式

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'

现在奇怪的是，当我打印 print(app.config)在 create_app 函数内部，配置就位，就像我希望的那样。例如 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，使用 Flask 应用程序工厂会更容易。从当前项目中获取适用于所有开发的默认配置。您可以随时通过 app.config.from_obj() or app.config.from_envvar() 加载更多配置，根据需要使用可选参数和环境变量。