8

我使用 Alembic 作为迁移工具，并在已更新的数据库上启动以下伪脚本（Alembic 没有修订条目，数据库架构是最新的）。

revision = '1067fd2d11c8'
down_revision = None

from alembic import op
import sqlalchemy as sa


def upgrade():
    op.add_column('box', sa.Column('has_data', sa.Boolean, server_default='0'))


def downgrade():
    pass

它只在 PostgreSQL 后面给我以下错误（对 MySQL 来说一切都很好）：

INFO  [alembic.migration] Context impl PostgresqlImpl.
INFO  [alembic.migration] Will assume transactional DDL.
INFO  [root] (ProgrammingError) ERREUR:  la colonne « has_data » de la relation « box » existe déjà

最后一行表示该列has_data已经存在。

我想检查该列之前是否存在op.add_column。

4

2 回答 2

6

最简单的答案是不要尝试这样做。相反，让您的 Alembic 迁移代表数据库的完整布局。然后，您进行的任何迁移都将基于对现有数据库的更改。

如果您已经有一个数据库，要开始迁移，请暂时指向一个空数据库并运行alembic revision --autogenerate -m "base". 然后，指向实际的数据库并运行alembic stamp head说数据库的当前状态由最新的迁移表示，而不是实际运行它。

如果您出于某种原因不想这样做，您可以选择不使用--autogenerate，而是生成空的修订版，然后用您想要的操作填写这些修订版。Alembic 不会阻止您这样做，只是不太方便。

于 2014-06-06T13:23:23.077 回答
4

不幸的是，我的情况是，我们有多个具有不同架构的版本，都需要迁移到单个代码库。目前还没有任何迁移，也没有任何数据库中标记的版本。所以第一次迁移会有这些条件检查。第一次迁移后，一切都将处于已知状态，我可以避免这种黑客攻击。

所以我在我的迁移中添加了这个（信用属于http://www.derstappen-it.de/tech-blog/sqlalchemie-alembic-check-if-table-has-column）：

from alembic import op
from sqlalchemy import engine_from_config
from sqlalchemy.engine import reflection

def _table_has_column(table, column):
    config = op.get_context().config
    engine = engine_from_config(
        config.get_section(config.config_ini_section), prefix='sqlalchemy.')
    insp = reflection.Inspector.from_engine(engine)
    has_column = False
    for col in insp.get_columns(table):
        if column not in col['name']:
            continue
        has_column = True
    return has_column

我的升级功能具有以下检查（请注意，我设置了一个添加with op.batch_alter_table行的批处理标志，这可能不在大多数设置中：

def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    with op.batch_alter_table('mytable', schema=None) as batch_op:
        if not _table_has_column('mytable', 'mycol'):
            batch_op.add_column(sa.Column('mycol', sa.Integer(), nullable=True))
        if not _table_has_column('mytable', 'mycol2'):
            batch_op.add_column(sa.Column('mycol2', sa.Integer(), nullable=True))
于 2018-10-18T00:22:05.910 回答