53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
"""fc2d-vii-c: image_record.artist_id + backfill + drop artist tags
|
|
|
|
Revision ID: 0008
|
|
Revises: 0007
|
|
Create Date: 2026-05-18
|
|
|
|
Internal forward-correctness migration (the big legacy-import migration
|
|
stays deferred). downgrade() does NOT recreate deleted artist tags;
|
|
downgrade is dev-only and the data is reconstructable by re-import.
|
|
"""
|
|
from typing import Sequence, Union
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
from backend.app.utils.artist_backfill import (
|
|
BACKFILL_PRIMARY_SQL,
|
|
BACKFILL_PROVENANCE_SQL,
|
|
BACKFILL_TAG_SQL,
|
|
DELETE_ARTIST_TAGS_SQL,
|
|
)
|
|
|
|
revision: str = "0008"
|
|
down_revision: Union[str, None] = "0007"
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
op.add_column(
|
|
"image_record",
|
|
sa.Column("artist_id", sa.Integer(), nullable=True),
|
|
)
|
|
op.create_foreign_key(
|
|
"fk_image_record_artist_id", "image_record", "artist",
|
|
["artist_id"], ["id"], ondelete="SET NULL",
|
|
)
|
|
op.create_index(
|
|
"ix_image_record_artist_id", "image_record", ["artist_id"],
|
|
)
|
|
op.execute(BACKFILL_PRIMARY_SQL)
|
|
op.execute(BACKFILL_PROVENANCE_SQL)
|
|
op.execute(BACKFILL_TAG_SQL)
|
|
op.execute(DELETE_ARTIST_TAGS_SQL)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_index("ix_image_record_artist_id", table_name="image_record")
|
|
op.drop_constraint(
|
|
"fk_image_record_artist_id", "image_record", type_="foreignkey"
|
|
)
|
|
op.drop_column("image_record", "artist_id")
|