6

我正在尝试创建一个使用 UUID 作为主键的用户模型：

from src.db import db # SQLAlchemy instance

import sqlalchemy_utils

import uuid


class User(db.Model):
    __tablename__ = 'user'

    id = db.Column(sqlalchemy_utils.UUIDType(binary=True), primary_key=True, nullable=False)

但是当我生成我收到的迁移时：

File "/home/pc/Downloads/project/auth/venv/lib/python3.6/site-packages/alembic/runtime/environment.py", line 836, in run_migrations
    self.get_context().run_migrations(**kw)
  File "/home/pc/Downloads/project/auth/venv/lib/python3.6/site-packages/alembic/runtime/migration.py", line 330, in run_migrations
step.migration_fn(**kw)
  File "/home/pc/Downloads/project/auth/migrations/versions/efae4166f832_.py", line 22, in upgrade
    sa.Column('id', sqlalchemy_utils.types.uuid.UUIDType(length=16), nullable=False),
NameError: name 'sqlalchemy_utils' is not defined`

我曾尝试像这样明确地通知我正在使用的模块并使用SQLAlchemy的“内部”实现

Obs：如果我手动导入并删除自动生成sqlalchemy_utils的/migrations/version/efae4166f832_.py长度，sa.Column('id', sqlalchemy_utils.types.uuid.UUIDType(length=16), nullable=False)它可以正常工作

我使用generate.py脚本生成迁移：

from src import create_app

from src.db import db

from flask_migrate import Migrate

# Models

from src.user.models.user import User

app = create_app()

migrate = Migrate(app, db)`

在此处输入图像描述

Obs：MySQL引擎

我希望当我生成迁移时，它会生成一个用户模型，该模型使用从 SQLAlchemy Utils 实现的 UUID 作为主键

4

4 回答 4

10

您只需添加：

import sqlalchemy_utils

到迁移文件夹中的 script.py.mako

于 2019-01-30T12:35:04.003 回答
2

谢谢，马可，但我已经修好了。我已将导入import sqlalchemy_utils放入env.py和script.py.mako中，我还放入了以下函数：

def render_item(type_, obj, autogen_context):
    """Apply custom rendering for selected items"""

    if type_ == "type" and isinstance(obj, sqlalchemy_utils.types.uuid.UUIDType):
        # Add import for this type
        autogen_context.imports.add("import sqlalchemy_utils")

        autogen_context.imports.add("import uuid")

        return "sqlalchemy_utils.types.uuid.UUIDType(), default=uuid.uuid4"

    # Default rendering for other objects
    return False

在env.py内部，以及我render_item=render_item在函数中设置的同一个文件run_migrations_online：

context.configure(
    ...,
    render_item=render_item,
    ...
)

我研究了自动执行此操作，但找不到任何可以帮助我的东西。

操作的顺序很重要：

  1. export FLASK_APP=manage.py

  2. flask db init

  3. 做上面的教程

  4. flask db migrate

  5. flask db upgrade

于 2019-02-01T01:46:50.153 回答
0

将该行添加import sqlalchemy_utils到新创建的migrations/versions/{hash}_my_comment.py文件中。但是，这只会解决迁移的特定步骤的问题。如果您希望对引用的列进行大量更改sqlalchemy_utils，您可能应该像 Walter 的建议那样做一些更强大的事情。尽管如此，看起来您可能需要添加代码来正确处理您最终使用的每种列类型。

注意：尽管在多个地方看到了将导入行添加到script.py.mako文件中的建议，但这对我不起作用。

于 2019-02-25T18:17:51.610 回答
0

添加文件将自动import sqlalchemy_utils在script.py.mako生成的所有迁移文件上导入此行并解决问题。

from alembic import op
import sqlalchemy as sa
import sqlalchemy_utils
${imports if imports else ""}
于 2021-12-29T06:18:08.567 回答