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

1"""Add job_variables column to flow-runs 

2 

3Revision ID: 342220764f0b 

4Revises: 265eb1a2da4c 

5Create Date: 2024-03-05 11:52:58.330563 

6 

7""" 

8 

9import sqlalchemy as sa 1 ctx1a

10from alembic import op 1 ctx1a

11from sqlalchemy import Text 1 ctx1a

12 

13import prefect 1 ctx1a

14 

15# revision identifiers, used by Alembic. 

16revision = "342220764f0b" 1 ctx1a

17down_revision = "265eb1a2da4c" 1 ctx1a

18branch_labels = None 1 ctx1a

19depends_on = None 1 ctx1a

20 

21 

22def upgrade(): 1 ctx1a

23 op.add_column( 1 ctx1a

24 "flow_run", 

25 sa.Column( 

26 "job_variables", 

27 prefect.server.utilities.database.JSON(astext_type=Text()), 

28 server_default="{}", 

29 nullable=True, 

30 ), 

31 ) 

32 

33 

34def downgrade(): 1 ctx1a

35 op.drop_column("flow_run", "job_variables")