[DOP-25645] change migrations

Author: TiGrib

data_rentgen/db/migrations/versions/2024-06-27_0b9aac68402b_create_operation.py

@@ -24,7 +24,6 @@ def upgrade() -> None:
         sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
         sa.Column("id", sa.UUID(), nullable=False),
         sa.Column("run_id", sa.UUID(), nullable=False),
-        sa.Column("sql_query_id", sa.BigInteger, nullable=True),
         sa.Column("status", sa.SmallInteger(), nullable=False),
         sa.Column("name", sa.String(), nullable=False),
         sa.Column("type", sa.String(length=32), nullable=False),
@@ -37,10 +36,8 @@ def upgrade() -> None:
         postgresql_partition_by="RANGE (created_at)",
     )
     op.create_index(op.f("ix__operation__run_id"), "operation", ["run_id"], unique=False)
-    op.create_index(op.f("ix__operation__sql_query_id"), "operation", ["sql_query_id"], unique=False)
 
 
 def downgrade() -> None:
     op.drop_index(op.f("ix__operation__run_id"), table_name="operation")
-    op.drop_index(op.f("ix__operation__sql_query_id"), table_name="operation")
     op.drop_table("operation")

data_rentgen/db/migrations/versions/2025-04-30_8267d8769ea0_create_sql_query.py

@@ -28,7 +28,14 @@ def upgrade() -> None:
     )
     op.create_index(op.f("ix__sql_query__fingerprint"), "sql_query", ["fingerprint"], unique=True)
 
+    op.add_column(
+        "operation",
+        sa.Column("sql_query_id", sa.BigInteger, nullable=True),
+    )
+    op.create_index(op.f("ix__operation__sql_query_id"), "operation", ["sql_query_id"], unique=False)
+
 
 def downgrade() -> None:
+    op.drop_index(op.f("ix__operation__sql_query_id"), table_name="operation")
     op.drop_index(op.f("ix__sql_query__fingerprint"), table_name="sql_query")
     op.drop_table("sql_query")