Coverage for slidge/db/alembic/versions/431bec3fe41c_0_6_0_migrations.py: 96%

28 statements  

« prev     ^ index     » next       coverage.py v7.17.0, created at 2026-11-10 12:26 +0000

1"""0.6.0 migrations 

2 

3Revision ID: 431bec3fe41c 

4Revises: 9c5a127d4600 

5Create Date: 2026-10-13 06:24:32.184312 

6 

7""" 

8 

9from collections.abc import Sequence 

10 

11import sqlalchemy as sa 

12from alembic import op 

13 

14# revision identifiers, used by Alembic. 

15revision: str = "431bec3fe41c" 

16down_revision: str | None = "9c5a127d4600" 

17branch_labels: str | Sequence[str] | None = None 

18depends_on: str | Sequence[str] | None = None 

19 

20 

21def upgrade() -> None: 

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

23 batch_op.add_column( 

24 sa.Column( 

25 "more_metadata_fetched", 

26 sa.Boolean(), 

27 server_default="false", 

28 nullable=False, 

29 ) 

30 ) 

31 batch_op.add_column(sa.Column("bio", sa.String(), nullable=True)) 

32 batch_op.add_column( 

33 sa.Column("online", sa.Boolean(), server_default="true", nullable=True) 

34 ) 

35 batch_op.drop_column("vcard_fetched") 

36 batch_op.drop_column("ptype") 

37 batch_op.drop_column("cached_presence") 

38 batch_op.drop_column("note") 

39 batch_op.drop_column("updated") 

40 

41 with op.batch_alter_table("participant", schema=None) as batch_op: 

42 batch_op.add_column( 

43 sa.Column("online", sa.Boolean(), server_default="true", nullable=True) 

44 ) 

45 batch_op.add_column(sa.Column("last_seen", sa.DateTime(), nullable=True)) 

46 batch_op.add_column(sa.Column("pstatus", sa.String(), nullable=True)) 

47 batch_op.add_column( 

48 sa.Column( 

49 "pshow", 

50 sa.Enum("away", "chat", "dnd", "xa", native_enum=False), 

51 nullable=True, 

52 ) 

53 ) 

54 

55 with op.batch_alter_table("room", schema=None) as batch_op: 

56 batch_op.add_column( 

57 sa.Column( 

58 "notify_settings", 

59 sa.Enum("never", "always", "on-mention", native_enum=False), 

60 nullable=True, 

61 ) 

62 ) 

63 batch_op.add_column(sa.Column("pinned", sa.Boolean(), nullable=True)) 

64 batch_op.drop_column("updated") 

65 

66 

67def downgrade() -> None: 

68 raise RuntimeError("downgrades are not supported")