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

25 statements  

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

1"""Add work queue name to runs 

2 

3Revision ID: 575634b7acd4 

4Revises: 296e2665785f 

5Create Date: 2022-08-07 13:41:38.128173 

6 

7""" 

8 

9import sqlalchemy as sa 1a

10from alembic import op 1a

11 

12# revision identifiers, used by Alembic. 

13revision = "575634b7acd4" 1a

14down_revision = "296e2665785f" 1a

15branch_labels = None 1a

16depends_on = None 1a

17 

18 

19def upgrade(): 1a

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

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

22 batch_op.create_index( 1a

23 batch_op.f("ix_deployment__work_queue_name"), 

24 ["work_queue_name"], 

25 unique=False, 

26 ) 

27 

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

29 batch_op.add_column(sa.Column("work_queue_name", sa.String(), nullable=True)) 1a

30 batch_op.create_index( 1a

31 batch_op.f("ix_flow_run__work_queue_name"), 

32 ["work_queue_name"], 

33 unique=False, 

34 ) 

35 

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

37 batch_op.alter_column("filter", nullable=True, server_default=None) 1a

38 

39 

40def downgrade(): 1a

41 op.execute( 

42 """ 

43 UPDATE work_queue 

44 SET filter = '{}' 

45 WHERE filter IS NULL; 

46 """ 

47 ) 

48 

49 with op.batch_alter_table("work_queue", schema=None) as batch_op: 

50 batch_op.alter_column("filter", nullable=False, server_default=sa.text("'{}'")) 

51 

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

53 batch_op.drop_index(batch_op.f("ix_flow_run__work_queue_name")) 

54 batch_op.drop_column("work_queue_name") 

55 

56 with op.batch_alter_table("deployment", schema=None) as batch_op: 

57 batch_op.drop_index(batch_op.f("ix_deployment__work_queue_name")) 

58 batch_op.drop_column("work_queue_name")