GeoAlchemy2 的 Alembic 迁移引发 NameError：未定义名称“geoalchemy2”

Question

我决定使用 Flask、postgresql 和传单编写一个小型 web 应用程序。我想使用Postgresql 的 PostGIS扩展程序存储坐标（纬度和经度）。我的烧瓶应用程序使用 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

我有点迷茫，在这个特定主题上我没有找到太多帮助。关于可能导致此问题的任何想法？

Answer 1

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

在运行自动生成的脚本之前，您应该始终查看它们。剧本中甚至有这样的评论。