Coverage for /usr/local/lib/python3.12/site-packages/prefect/server/database/_migrations/versions/sqlite/2022_04_19_181604_7f5f335cace3_add_flow_run_state_name.py: 70%

20 statements  

« prev     ^ index     » next       coverage.py v7.10.6, created at 2025-12-05 11:21 +0000

1"""Add flow_run.state_name 

2 

3Revision ID: 7f5f335cace3 

4Revises: 71a57ec351d1 

5Create Date: 2022-04-19 18:16:04.893465 

6 

7""" 

8 

9import sqlalchemy as sa 1a

10from alembic import op 1a

11 

12# revision identifiers, used by Alembic. 

13revision = "7f5f335cace3" 1a

14down_revision = "71a57ec351d1" 1a

15branch_labels = None 1a

16depends_on = None 1a

17 

18 

19def upgrade(): 1a

20 with op.batch_alter_table("flow_run", schema=None) as batch_op: 1a

21 batch_op.add_column(sa.Column("state_name", sa.String(), nullable=True)) 1a

22 batch_op.create_index("ix_flow_run__state_name", ["state_name"], unique=False) 1a

23 

24 with op.batch_alter_table("task_run", schema=None) as batch_op: 1a

25 batch_op.add_column(sa.Column("state_name", sa.String(), nullable=True)) 1a

26 batch_op.create_index("ix_task_run__state_name", ["state_name"], unique=False) 1a

27 

28 

29def downgrade(): 1a

30 with op.batch_alter_table("task_run", schema=None) as batch_op: 

31 batch_op.drop_index("ix_task_run__state_name") 

32 batch_op.drop_column("state_name") 

33 

34 with op.batch_alter_table("flow_run", schema=None) as batch_op: 

35 batch_op.drop_index("ix_flow_run__state_name") 

36 batch_op.drop_column("state_name")