Coverage for /usr/local/lib/python3.12/site-packages/prefect/server/database/_migrations/versions/postgresql/2024_09_11_090317_555ed31b284d_add_concurrency_options.py: 92%

1"""add concurrency options 

2 

3Revision ID: 555ed31b284d 

4Revises: 97429116795e 

5Create Date: 2024-09-11 09:03:17.744587 

6 

7""" 

8 

9import sqlalchemy as sa 

10from alembic import op 

11 

12import prefect 

13from prefect.server.schemas.core import ConcurrencyOptions 

14 

15# revision identifiers, used by Alembic. 

16revision = "555ed31b284d" 

17down_revision = "97429116795e" 

18branch_labels = None 

19depends_on = None 

20 

21 

22def upgrade(): 

23 op.add_column( 

24 "deployment", 

25 sa.Column( 

26 "concurrency_options", 

27 prefect.server.utilities.database.Pydantic(ConcurrencyOptions), 

28 nullable=True, 

29 ), 

30 ) 

31 

32 

33def downgrade(): 

34 op.drop_column("deployment", "concurrency_options")