修复数据库表名引用

This commit is contained in:
jxxghp
2025-08-19 12:54:47 +08:00
parent 968cfd8654
commit dc50a68b01

View File

@@ -95,14 +95,14 @@ def convert_to_identity(connection, table_name):
# 获取当前序列的最大值
result = connection.execute(sa.text(f"""
SELECT COALESCE(MAX(id), 0) + 1 as next_value
FROM {table_name}
FROM "{table_name}"
"""))
next_value = result.fetchone()[0]
# 直接修改列属性添加Identity保持其他约束不变
# 这种方式不会删除主键约束和索引
connection.execute(sa.text(f"""
ALTER TABLE {table_name}
ALTER TABLE "{table_name}"
ALTER COLUMN id ADD GENERATED BY DEFAULT AS IDENTITY (START WITH {next_value})
"""))