Coverage for /usr/local/lib/python3.12/site-packages/prefect/server/database/_migrations/versions/sqlite/2024_05_01_103824_20fbd53b3cef_add_catchup_fields_to_deploymentschedule.py: 75%

16 statements  

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

1"""add catchup fields to DeploymentSchedule 

2 

3Revision ID: 20fbd53b3cef 

4Revises: a8e62d4c72cf 

5Create Date: 2024-05-01 10:38:24.336255 

6 

7""" 

8 

9import sqlalchemy as sa 1a

10from alembic import op 1a

11 

12# revision identifiers, used by Alembic. 

13revision = "20fbd53b3cef" 1a

14down_revision = "a8e62d4c72cf" 1a

15branch_labels = None 1a

16depends_on = None 1a

17 

18 

19def upgrade(): 1a

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

21 batch_op.add_column(sa.Column("max_active_runs", sa.Integer(), nullable=True)) 1a

22 batch_op.add_column( 1a

23 sa.Column("max_scheduled_runs", sa.Integer(), nullable=True) 

24 ) 

25 batch_op.add_column( 1a

26 sa.Column("catchup", sa.Boolean(), server_default="0", nullable=False) 

27 ) 

28 

29 

30def downgrade(): 1a

31 with op.batch_alter_table("deployment_schedule", schema=None) as batch_op: 

32 batch_op.drop_column("catchup") 

33 batch_op.drop_column("max_scheduled_runs") 

34 batch_op.drop_column("max_active_runs")