Coverage for /usr/local/lib/python3.12/site-packages/prefect/server/database/_migrations/versions/postgresql/2022_10_19_165110_8ea825da948d_track_retries_restarts.py: 90%

1"""Add retry and restart metadata 

2 

3Revision ID: 8ea825da948d 

4Revises: ad4b1b4d1e9d 

5Create Date: 2022-10-19 16:51:10.239643 

6 

7""" 

8 

9import sqlalchemy as sa 

10from alembic import op 

11 

12# revision identifiers, used by Alembic. 

13revision = "8ea825da948d" 

14down_revision = "3ced59d8806b" 

15branch_labels = None 

16depends_on = None 

17 

18 

19def upgrade(): 

20 # ### commands auto generated by Alembic - please adjust! ### 

21 op.add_column( 

22 "task_run", 

23 sa.Column( 

24 "flow_run_run_count", sa.Integer(), server_default="0", nullable=False 

25 ), 

26 ) 

27 # ### end Alembic commands ### 

28 

29 

30def downgrade(): 

31 # ### commands auto generated by Alembic - please adjust! ### 

32 op.drop_column("task_run", "flow_run_run_count") 

33 # ### end Alembic commands ###