python - 如何使用 sqlalchemy-migrate 将列类型从字符变化更改为整数

Question

我正在使用 sqlalchemy-migrate 更改 Postgre SQL 数据库中表中某一列的类型。我使用的升级脚本是:

# -*- cofing: utf-8 -*-

from sqlalchemy import MetaData, Table, Column, String, Integer
from migrate import changeset


metadata = MetaData()


def upgrade(migrate_engine):
    # ALTER TABLE courses ALTER COLUMN number SET DATA TYPE character varying;
    metadata.bind = migrate_engine
    courses = Table('courses', metadata, Column("number", Integer), extend_existing=True)
    courses.c.number.alter(type=String)


def downgrade(migrate_engine):
    # ALTER TABLE courses ALTER COLUMN number SET DATA TYPE integer;
    metadata.bind = migrate_engine
    courses = Table('courses', metadata, Column("number", String), extend_existing=True)
    courses.c.number.alter(type=Integer, cast='numeric')

升级部分似乎有效，但降级总是失败并出现以下错误:

sqlalchemy.exc.ProgrammingError: (ProgrammingError) column "number" cannot be cast to type integer
'\nALTER TABLE courses ALTER COLUMN number TYPE INTEGER' {}

现在，如果我使用纯 SQL，我可以使用 ALTER TABLE courses ALTER COLUMN number TYPE INTEGER USING number::numeric 将列类型从 character varying 改回到 integer，但我不知道如何使用 sqlalchemy-migrate 实现它。

有没有办法强制 sqlalchemy 在 ALTER 子句中包含 USING number::numeric？还是有其他方法可以避免我在上面发布的错误？

感谢您的帮助。

Answer 1

看起来 sqlalchemy.migrate 不支持在 postgresql 从 String 更改为 Integer 列类型的情况下呈现有效查询。

在您的情况下，我会将其实现为直接查询执行并继续。

def downgrade(migrate_engine):
    # ALTER TABLE courses ALTER COLUMN number SET DATA TYPE integer;
    migrate_engine.execute('ALTER TABLE courses ALTER COLUMN number TYPE INTEGER USING number::numeric')

顺便说一句，从字符串迁移到整数可能会因不同的原因而失败——当列值包含一些无法转换为数字的值时。因此，我会在应用程序逻辑中添加一些额外的验证，以确保以后可以进行降级迁移。