Alembic migration for GeoAlchemy2 raises NameError: name 'geoalchemy2' is not defined

Alembic migration for GeoAlchemy2 raises NameError: name 'geoalchemy2' is not defined

我决定使用 Flask、postgresql 和 leaflet 编写一个小型 webapp。我想使用 PostGIS extender for postgresql 来存储坐标(纬度和经度)。我的 Flask 应用程序使用 Flask-SQLAlchemy、蓝图,尤其是 Flask-Migrate 用于数据库迁移过程。

这是我的数据库模型的摘录:

from . import db
from geoalchemy2 import Geometry


class Station(db.Model):
    __tablename__ = 'stations'

    id = db.Column(db.Integer, primary_key=True, unique=True)
    name = db.Column(db.String(255))
    position = db.Column(Geometry('Point', srid=4326))

这里是我的应用程序的摘录/init.py

import os
from flask import Flask
from flask_sqlalchemy import SQLAlchemy
from flask_migrate import Migrate
from config import config

db = SQLAlchemy()
migrate = Migrate()

def create_app(config_name=None, main=True):

if config_name is None:
    config_name = os.environ.get('FLASK_CONFIG', 'development')

app = Flask(__name__)
app.config.from_object(config[config_name])

db.init_app(app)
migrate.init_app(app, db)

from .home import home as home_blueprint
app.register_blueprint(home_blueprint)

from .admin import admin as admin_blueprint
app.register_blueprint(admin_blueprint, url_prefix='/admin')

return app

在尝试使用特定的扩展器之前,我没有遇到任何问题来调整我的模型。从那时起,迁移工作正常,但升级不起作用(python manage.py db upgrade)。

这里是我从网络服务器获取的日志:

sa.Column('position', geoalchemy2.types.Geometry(geometry_type='POINT', srid=4326), nullable=True),
`enter code here`NameError: name 'geoalchemy2' is not defined

我有点迷茫,在这个特定主题上找不到太多帮助。知道可能导致此问题的原因吗?

Alembic 不会尝试确定和呈现迁移脚本中自定义类型的所有导入。编辑生成的脚本以包含 from geoalchemy2.types import Geometry 并将列 def 更改为仅使用 Geometry.

您应该始终在 运行 之前查看自动生成的脚本。剧本里还有评论这么说。

另一种不需要手动编辑修订版的方法是将 import geoalchemy2 添加到 alembic/script.py.mako 中,然后 alembic 每次都添加模块。

"""${message}
  
Revision ID: ${up_revision}
Revises: ${down_revision | comma,n}
Create Date: ${create_date}

"""
from alembic import op
import sqlalchemy as sa
import geoalchemy2 # <--- right here
${imports if imports else ""}

# revision identifiers, used by Alembic.
revision = ${repr(up_revision)}
down_revision = ${repr(down_revision)}
branch_labels = ${repr(branch_labels)}
depends_on = ${repr(depends_on)}


def upgrade():
    ${upgrades if upgrades else "pass"}


def downgrade():
    ${downgrades if downgrades else "pass"}