7

我正在为烧瓶+sqlalchemy 项目使用alembic 迁移，并且在我尝试查询alembic 中的模型之前，一切都按预期工作。

from models import StoredFile

def upgrade():
    ### commands auto generated by Alembic - please adjust! ###
    op.add_column('stored_file', sa.Column('mimetype', sa.Unicode(length=32))
    for sf in StoredFile.query.all():
        sf.mimetype = guess_type(sf.title)

上面的代码在添加列后卡住了，永远不会出来。我想这StoredFile.query是尝试使用与 alembic 使用的数据库连接不同的数据库连接。（但是为什么？我错过了什么env.py吗？）

我可以通过使用解决它，op.get_bind().execute(...)但问题是如何直接在 alembic 中使用模型？

4

2 回答 2

6

models你不应该在你的 alembic 迁移中使用类。如果需要使用模型类，则应在每个迁移文件中重新定义它们，以使迁移自包含。原因是可以在一个命令中部署多个迁移，并且在编写迁移到实际在生产中执行之前，模型类可能已根据“稍后”迁移进行了更改。

例如，请参阅Operations.execute文档中的此示例：

from sqlalchemy.sql import table, column
from sqlalchemy import String
from alembic import op

account = table('account',
    column('name', String)
)
op.execute(
    account.update(). \
        where(account.c.name==op.inline_literal('account 1')). \
        values({'name':op.inline_literal('account 2')})
        )

提示：您不需要包含完整的模型类，只需包含迁移所需的部分。

于 2017-08-21T20:31:06.437 回答
4

我有同样的问题。当您使用时，StoredFile.query您使用的会话与 Alembic 使用的会话不同。它尝试查询数据库，但表被锁定，因为您正在更改它。所以升级只是坐在那里永远等待，因为你有两个会话在等待对方。根据@SowingSadness 的回复，这对我有用：

from models import StoredFile

def upgrade():
    ### commands auto generated by Alembic - please adjust! ###
    op.add_column('stored_file', sa.Column('mimetype', sa.Unicode(length=32))

    connection = op.get_bind()
    SessionMaker = sessionmaker(bind=connection.engine)
    session = SessionMaker(bind=connection)
    for sf in session.query(StoredFile):
        sf.mimetype = guess_type(sf.title)
    session.flush()
    op.other_operations()
于 2015-12-04T20:52:19.860 回答