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

15 statements  

« prev     ^ index     » next       coverage.py v7.10.6, created at 2025-12-05 13:38 +0000

1"""Create csrf_token toble 

2 

3Revision ID: bacc60edce16 

4Revises: 342220764f0b 

5Create Date: 2024-03-13 11:13:16.487004 

6 

7""" 

8 

9import sqlalchemy as sa 1a

10from alembic import op 1a

11 

12import prefect 1a

13 

14# revision identifiers, used by Alembic. 

15revision = "bacc60edce16" 1a

16down_revision = "342220764f0b" 1a

17branch_labels = None 1a

18depends_on = None 1a

19 

20 

21def upgrade(): 1a

22 op.create_table( 1a

23 "csrf_token", 

24 sa.Column("token", sa.String(), nullable=False), 

25 sa.Column("client", sa.String(), nullable=False), 

26 sa.Column( 

27 "expiration", 

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

29 nullable=False, 

30 ), 

31 sa.Column( 

32 "id", 

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

34 server_default=sa.text( 

35 "(\n (\n lower(hex(randomblob(4)))\n || '-'\n || lower(hex(randomblob(2)))\n || '-4'\n || substr(lower(hex(randomblob(2))),2)\n || '-'\n || substr('89ab',abs(random()) % 4 + 1, 1)\n || substr(lower(hex(randomblob(2))),2)\n || '-'\n || lower(hex(randomblob(6)))\n )\n )" 

36 ), 

37 nullable=False, 

38 ), 

39 sa.Column( 

40 "created", 

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

42 server_default=sa.text("(strftime('%Y-%m-%d %H:%M:%f000', 'now'))"), 

43 nullable=False, 

44 ), 

45 sa.Column( 

46 "updated", 

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

48 server_default=sa.text("(strftime('%Y-%m-%d %H:%M:%f000', 'now'))"), 

49 nullable=False, 

50 ), 

51 sa.PrimaryKeyConstraint("id", name=op.f("pk_csrf_token")), 

52 sa.UniqueConstraint("client", name=op.f("uq_csrf_token__client")), 

53 ) 

54 with op.batch_alter_table("csrf_token", schema=None) as batch_op: 1a

55 batch_op.create_index( 1a

56 batch_op.f("ix_csrf_token__updated"), ["updated"], unique=False 

57 ) 

58 

59 

60def downgrade(): 1a

61 with op.batch_alter_table("csrf_token", schema=None) as batch_op: 

62 batch_op.drop_index(batch_op.f("ix_csrf_token__updated")) 

63 

64 op.drop_table("csrf_token")