我有一个创建表的alembic升级脚本,但是如果它已经存在,我不希望它创建表。
根据alembic doc,我可以将关键字args传递给op.create_tables
,这是sqlalchemy.schema.table可以接受的,所以我使用的是keep_existing
关键字:
op.create_table('foo_model',
sa.Column('foo_id', sa.Integer(), nullable=False),
sa.Column('foo_str', sa.String(length=255), nullable=True),
sa.PrimaryKeyConstraint('foo_id'),
keep_existing= True
)
但是我仍然得到表已经存在的错误。
sqlalchemy.exc.InternalError: (InternalError) (1050, u"Table 'foo_model' already exists") '\nCREATE TABLE foo_model (\n\tfoo_id INTEGER NOT NULL AUTO_INCREMENT, \n\tfoo_str VARCHAR(255), \n\tPRIMARY KEY (foo_id)\n)\n\n' ()
您可以获取现有表的列表,如下所示:
conn = op.get_bind()
inspector = Inspector.from_engine(conn)
tables = inspector.get_table_names()
然后检查表是否已存在
if table_name not in tables:
op.create_table()
正如其他地方(Check if a table column exists in the database using SQLAlchemy and Alembic)说的那样,alembic应该反映数据库的完整状态,这意味着它会自动知道表是否存在。
确保定义升级和降级,以便在升级创建表时降级将其删除。
如果你这样做,你可以“降级”到以前的版本并再次升级,它将工作。在修订上使用autogenerate来创建初始状态。
这可能是因为该表已经存在。只需使用psql
并运行drop table foo_model;
即可从数据库中删除表