Coverage for slidge/db/alembic/versions/805fdc0bf984_add_unique_constraint_for_occupant_id.py: 83%

1"""Add unique constraint for occupant ID 

2 

3Revision ID: 805fdc0bf984 

4Revises: b7a06a86416a 

5Create Date: 2025-12-17 19:55:11.058263 

6 

7""" 

8 

9from collections.abc import Sequence 

10 

11from alembic import op 

12 

13# revision identifiers, used by Alembic. 

14revision: str = "805fdc0bf984" 

15down_revision: str | None = "b7a06a86416a" 

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

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

18 

19 

20def upgrade() -> None: 

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

22 batch_op.create_unique_constraint( 

23 "uq_participant_room_id_occupant_id", ["room_id", "occupant_id"] 

24 ) 

25 

26 

27def downgrade() -> None: 

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

29 batch_op.drop_constraint( 

30 batch_op.f("uq_participant_room_id_occupant_id"), type_="unique" 

31 )