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

17 statements  

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

1"""Expand deployment schema for improved ux 

2 

3Revision ID: 296e2665785f 

4Revises: 53c19b31aa09 

5Create Date: 2022-08-06 13:00:09.419944 

6 

7""" 

8 

9import sqlalchemy as sa 1a

10from alembic import op 1a

11 

12import prefect 1a

13 

14# revision identifiers, used by Alembic. 

15revision = "296e2665785f" 1a

16down_revision = "53c19b31aa09" 1a

17branch_labels = None 1a

18depends_on = None 1a

19 

20 

21def upgrade(): 1a

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

23 batch_op.add_column( 1a

24 sa.Column( 

25 "infra_overrides", 

26 prefect.server.utilities.database.JSON(astext_type=sa.Text()), 

27 server_default="{}", 

28 nullable=False, 

29 ) 

30 ) 

31 batch_op.add_column( 1a

32 sa.Column( 

33 "path", 

34 sa.String(), 

35 nullable=True, 

36 ) 

37 ) 

38 batch_op.add_column( 1a

39 sa.Column( 

40 "entrypoint", 

41 sa.String(), 

42 nullable=True, 

43 ) 

44 ) 

45 

46 

47def downgrade(): 1a

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

49 batch_op.drop_column("entrypoint") 

50 batch_op.drop_column("path") 

51 batch_op.drop_column("infra_overrides")