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

18 statements  

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

1"""Adds block schema version 

2 

3Revision ID: e757138e954a 

4Revises: 575634b7acd4 

5Create Date: 2022-08-18 10:25:27.189680 

6 

7""" 

8 

9import sqlalchemy as sa 1a

10from alembic import op 1a

11 

12# revision identifiers, used by Alembic. 

13revision = "e757138e954a" 1a

14down_revision = "575634b7acd4" 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_schema", schema=None) as batch_op: 1a

23 batch_op.add_column( 1a

24 sa.Column( 

25 "version", sa.String(), server_default="non-versioned", nullable=False 

26 ) 

27 ) 

28 batch_op.drop_index("uq_block_schema__checksum") 1a

29 batch_op.create_index( 1a

30 "uq_block_schema__checksum_version", ["checksum", "version"], unique=True 

31 ) 

32 

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

34 

35 

36def downgrade(): 1a

37 with op.batch_alter_table("block_schema", schema=None) as batch_op: 

38 batch_op.drop_index("uq_block_schema__checksum_version") 

39 batch_op.create_index("uq_block_schema__checksum", ["checksum"], unique=False) 

40 batch_op.drop_column("version")