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

1"""Adds heartbeat_interval_seconds column to worker table 

2 

3Revision ID: 50f8c182c3ca 

4Revises: 5f623ddbf7fe 

5Create Date: 2023-09-06 08:57:47.166983 

6 

7""" 

8 

9import sqlalchemy as sa 

10from alembic import op 

11 

12# revision identifiers, used by Alembic. 

13revision = "50f8c182c3ca" 

14down_revision = "5f623ddbf7fe" 

15branch_labels = None 

16depends_on = None 

17 

18 

19def upgrade(): 

20 op.add_column( 

21 "worker", sa.Column("heartbeat_interval_seconds", sa.Integer(), nullable=True) 

22 ) 

23 

24 

25def downgrade(): 

26 op.drop_column("worker", "heartbeat_interval_seconds")