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

1"""add_variables 

2 

3Revision ID: 310dda75f561 

4Revises: 3bf47e3ce2dd 

5Create Date: 2023-04-05 13:45:20.470247 

6 

7""" 

8 

9import sqlalchemy as sa 

10from alembic import op 

11 

12import prefect 

13 

14# revision identifiers, used by Alembic. 

15revision = "310dda75f561" 

16down_revision = "43c94d4c7aa3" 

17branch_labels = None 

18depends_on = None 

19 

20 

21def upgrade(): 

22 op.create_table( 

23 "variable", 

24 sa.Column( 

25 "id", 

26 prefect.server.utilities.database.UUID(), 

27 server_default=sa.text("(GEN_RANDOM_UUID())"), 

28 nullable=False, 

29 ), 

30 sa.Column( 

31 "created", 

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

33 server_default=sa.text("CURRENT_TIMESTAMP"), 

34 nullable=False, 

35 ), 

36 sa.Column( 

37 "updated", 

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

39 server_default=sa.text("CURRENT_TIMESTAMP"), 

40 nullable=False, 

41 ), 

42 sa.Column("name", sa.String(), nullable=False), 

43 sa.Column("value", sa.String(), nullable=False), 

44 sa.Column( 

45 "tags", 

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

47 server_default="[]", 

48 nullable=False, 

49 ), 

50 sa.PrimaryKeyConstraint("id", name=op.f("pk_variable")), 

51 sa.UniqueConstraint("name", name=op.f("uq_variable__name")), 

52 ) 

53 op.create_index(op.f("ix_variable__updated"), "variable", ["updated"], unique=False) 

54 

55 

56def downgrade(): 

57 op.drop_index(op.f("ix_variable__updated"), table_name="variable") 

58 op.drop_table("variable")