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

27 statements  

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

1"""Remove flow runners 

2 

3Revision ID: 2fe8ef6a6514 

4Revises: 628a873f0d1a 

5Create Date: 2022-07-20 11:34:51.903172 

6 

7""" 

8 

9import sqlalchemy as sa 1a

10from alembic import op 1a

11from sqlalchemy.dialects import sqlite 1a

12 

13# revision identifiers, used by Alembic. 

14revision = "2fe8ef6a6514" 1a

15down_revision = "628a873f0d1a" 1a

16branch_labels = None 1a

17depends_on = None 1a

18 

19 

20def upgrade(): 1a

21 op.execute("PRAGMA foreign_keys=OFF") 1a

22 

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

24 batch_op.drop_column("flow_runner_type") 1a

25 batch_op.drop_column("flow_runner_config") 1a

26 

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

28 batch_op.drop_index("ix_flow_run__flow_runner_type") 1a

29 batch_op.drop_column("flow_runner_type") 1a

30 batch_op.drop_column("empirical_config") 1a

31 batch_op.drop_column("flow_runner_config") 1a

32 

33 op.execute("PRAGMA foreign_keys=ON") 1a

34 

35 # ### end Alembic commands ### 

36 

37 

38def downgrade(): 1a

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

40 batch_op.add_column( 

41 sa.Column("flow_runner_config", sqlite.JSON(), nullable=True) 

42 ) 

43 batch_op.add_column( 

44 sa.Column( 

45 "empirical_config", 

46 sqlite.JSON(), 

47 server_default=sa.text("'{}'"), 

48 nullable=False, 

49 ) 

50 ) 

51 batch_op.add_column(sa.Column("flow_runner_type", sa.VARCHAR(), nullable=True)) 

52 batch_op.create_index( 

53 "ix_flow_run__flow_runner_type", ["flow_runner_type"], unique=False 

54 ) 

55 

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

57 batch_op.add_column( 

58 sa.Column("flow_runner_config", sqlite.JSON(), nullable=True) 

59 ) 

60 batch_op.add_column(sa.Column("flow_runner_type", sa.VARCHAR(), nullable=True))