Coverage for /usr/local/lib/python3.12/site-packages/prefect/server/database/_migrations/versions/sqlite/2023_04_05_120713_340f457b315f_add_column_to_deployments_for_pull_steps.py: 76%

17 statements  

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

1"""Add column to deployments for pull steps 

2 

3Revision ID: 340f457b315f 

4Revises: 3e1eb8281d5e 

5Create Date: 2023-04-05 12:07:13.733969 

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 = "340f457b315f" 1a

15down_revision = "3e1eb8281d5e" 1a

16branch_labels = None 1a

17depends_on = None 1a

18 

19 

20def upgrade(): 1a

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

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

23 batch_op.add_column(sa.Column("pull_steps", sqlite.JSON(), nullable=True)) 1a

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

25 

26 

27def downgrade(): 1a

28 op.execute("PRAGMA foreign_keys=OFF") 

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

30 batch_op.drop_column("pull_steps") 

31 op.execute("PRAGMA foreign_keys=ON")