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

1"""Add last_polled 

2 

3Revision ID: 3ced59d8806b 

4Revises: fa319f214160 

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

6 

7""" 

8 

9import sqlalchemy as sa 1 ctx1a

10from alembic import op 1 ctx1a

11 

12import prefect 1 ctx1a

13 

14# revision identifiers, used by Alembic. 

15revision = "3ced59d8806b" 1 ctx1a

16down_revision = "fa319f214160" 1 ctx1a

17branch_labels = None 1 ctx1a

18depends_on = None 1 ctx1a

19 

20 

21def upgrade(): 1 ctx1a

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

23 batch_op.add_column( 1 ctx1a

24 sa.Column( 

25 "last_polled", 

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

27 nullable=True, 

28 ) 

29 ) 

30 

31 

32def downgrade(): 1 ctx1a

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

34 batch_op.drop_column("last_polled")