Coverage for slidge/db/alembic/versions/80e3206c67f4_space_avatars.py: 93%

1"""space avatars 

2 

3Revision ID: 80e3206c67f4 

4Revises: 4681a639c030 

5Create Date: 2026-08-28 10:54:53.889599 

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 = "80e3206c67f4" 

16down_revision: str | None = "4681a639c030" 

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("space", schema=None) as batch_op: 

23 batch_op.add_column(sa.Column("avatar_id", sa.Integer(), nullable=True)) 

24 batch_op.add_column(sa.Column("banner_id", sa.Integer(), nullable=True)) 

25 batch_op.create_foreign_key( 

26 batch_op.f("fk_space_banner_id_avatar"), "avatar", ["banner_id"], ["id"] 

27 ) 

28 batch_op.create_foreign_key( 

29 batch_op.f("fk_space_avatar_id_avatar"), "avatar", ["avatar_id"], ["id"] 

30 ) 

31 

32 

33def downgrade() -> None: 

34 raise RuntimeError("downgrades are not supported")