fix: ensure proper quoting in migrations

This commit is contained in:
Matthias
2026-04-19 13:04:23 +02:00
parent c1795c79f3
commit 55361f0a9e
+6 -4
View File
@@ -35,10 +35,12 @@ def get_last_sequence_ids(engine, sequence_name: str, table_back_name: str) -> i
if engine.name == "postgresql": if engine.name == "postgresql":
with engine.begin() as connection: with engine.begin() as connection:
last_id = connection.execute(text(f"select nextval('{sequence_name}')")).fetchone()[0] last_id = connection.execute(
text(f"""select nextval('"{sequence_name}"')""")
).fetchone()[0]
with engine.begin() as connection: with engine.begin() as connection:
connection.execute( connection.execute(
text(f"ALTER SEQUENCE {sequence_name} rename to {table_back_name}_id_seq_bak") text(f'ALTER SEQUENCE "{sequence_name}" rename to "{table_back_name}_id_seq_bak"')
) )
return last_id return last_id
@@ -88,9 +90,9 @@ def drop_index_on_table(engine, inspector, table_bak_name):
# drop indexes on backup table in new session # drop indexes on backup table in new session
for index in inspector.get_indexes(table_bak_name): for index in inspector.get_indexes(table_bak_name):
if engine.name == "mysql": if engine.name == "mysql":
connection.execute(text(f"drop index {index['name']} on {table_bak_name}")) connection.execute(text(f'drop index "{index["name"]}" on {table_bak_name}'))
else: else:
connection.execute(text(f"drop index {index['name']}")) connection.execute(text(f'drop index "{index["name"]}"'))
def migrate_trades_and_orders_table( def migrate_trades_and_orders_table(