Coverage for slidge/db/alembic/versions/259a1b71eadf_add_origin_group_id_table.py: 80%

1"""Add origin group ID table 

2 

3Revision ID: 259a1b71eadf 

4Revises: cef02a8b1451 

5Create Date: 2025-11-26 06:50:04.509013 

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 = "259a1b71eadf" 

16down_revision: str | None = "cef02a8b1451" 

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

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

19 

20 

21def upgrade() -> None: 

22 op.create_table( 

23 "group_msg_origin", 

24 sa.Column("foreign_key", sa.Integer(), nullable=False), 

25 sa.Column("id", sa.Integer(), nullable=False), 

26 sa.Column("legacy_id", sa.String(), nullable=False), 

27 sa.Column("xmpp_id", sa.String(), nullable=False), 

28 sa.ForeignKeyConstraint( 

29 ["foreign_key"], 

30 ["room.id"], 

31 name=op.f("fk_group_msg_origin_foreign_key_room"), 

32 ), 

33 sa.PrimaryKeyConstraint("id", name=op.f("pk_group_msg_origin")), 

34 ) 

35 with op.batch_alter_table("group_msg_origin", schema=None) as batch_op: 

36 batch_op.create_index( 

37 "ix_group_msg_origin_legacy_id", ["legacy_id", "foreign_key"], unique=False 

38 ) 

39 

40 # ### end Alembic commands ### 

41 

42 

43def downgrade() -> None: 

44 with op.batch_alter_table("group_msg_origin", schema=None) as batch_op: 

45 batch_op.drop_index("ix_group_msg_origin_legacy_id") 

46 

47 op.drop_table("group_msg_origin") 

48 # ### end Alembic commands ###