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

16 statements  

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

1"""Removing default storage block document. 

2 

3Revision ID: 56be24fdb383 

4Revises: 638cbcc2a158 

5Create Date: 2022-07-14 11:31:38.843301 

6 

7""" 

8 

9import sqlalchemy as sa 1a

10from alembic import op 1a

11 

12# revision identifiers, used by Alembic. 

13revision = "56be24fdb383" 1a

14down_revision = "638cbcc2a158" 1a

15branch_labels = None 1a

16depends_on = None 1a

17 

18 

19def upgrade(): 1a

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

21 

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

23 batch_op.drop_index("ix_block_document__is_default_storage_block_document") 1a

24 batch_op.drop_column("is_default_storage_block_document") 1a

25 

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

27 

28 

29def downgrade(): 1a

30 with op.batch_alter_table("block_document", schema=None) as batch_op: 

31 batch_op.add_column( 

32 sa.Column( 

33 "is_default_storage_block_document", 

34 sa.BOOLEAN(), 

35 server_default=sa.text("'0'"), 

36 nullable=True, 

37 ) 

38 ) 

39 batch_op.create_index( 

40 "ix_block_document__is_default_storage_block_document", 

41 ["is_default_storage_block_document"], 

42 unique=False, 

43 )