661ff8ef4425_rename_transforms_to_flows.py 1.0 KB

123456789101112131415161718192021222324252627282930313233343536
  1. """rename_transforms_to_flows
  2. Revision ID: 661ff8ef4425
  3. Revises: 9a3b9a199aa8
  4. Create Date: 2025-08-15 16:16:12.792775
  5. """
  6. from typing import Sequence, Union
  7. from alembic import op
  8. import sqlalchemy as sa
  9. # revision identifiers, used by Alembic.
  10. revision: str = '661ff8ef4425'
  11. down_revision: Union[str, None] = '9a3b9a199aa8'
  12. branch_labels: Union[str, Sequence[str], None] = None
  13. depends_on: Union[str, Sequence[str], None] = None
  14. def upgrade() -> None:
  15. """Upgrade schema."""
  16. # Rename the table from 'transforms' to 'flows'
  17. op.rename_table('transforms', 'flows')
  18. # Rename the column from 'transform_schema' to 'flow_schema'
  19. op.alter_column('flows', 'transform_schema', new_column_name='flow_schema')
  20. def downgrade() -> None:
  21. """Downgrade schema."""
  22. # Rename the column back from 'flow_schema' to 'transform_schema'
  23. op.alter_column('flows', 'flow_schema', new_column_name='transform_schema')
  24. # Rename the table back from 'flows' to 'transforms'
  25. op.rename_table('flows', 'transforms')