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

1"""Add last_polled 

2 

3Revision ID: 3ced59d8806b 

4Revises: 6d548701edef 

5Create Date: 2022-10-14 10:14:23.979848 

6 

7""" 

8 

9import sqlalchemy as sa 

10from alembic import op 

11 

12import prefect 

13 

14# revision identifiers, used by Alembic. 

15revision = "3ced59d8806b" 

16down_revision = "6d548701edef" 

17branch_labels = None 

18depends_on = None 

19 

20 

21def upgrade(): 

22 with op.batch_alter_table("work_queue", schema=None) as batch_op: 

23 batch_op.add_column( 

24 sa.Column( 

25 "last_polled", 

26 prefect.server.utilities.database.Timestamp(timezone=True), 

27 nullable=True, 

28 ) 

29 ) 

30 

31 

32def downgrade(): 

33 with op.batch_alter_table("work_queue", schema=None) as batch_op: 

34 batch_op.drop_column("last_polled")