"""SQLAlchemy model for the artist_yt_delivery_settings table."""

from sqlalchemy import delete
from sqlalchemy.dialects.mysql import INTEGER
from sqlalchemy.orm import Mapped, Session, mapped_column

from contributor.connectors.mysql import BaseModel


class ArtistYtDeliverySettings(BaseModel):
    __tablename__ = "artist_yt_delivery_settings"

    id: Mapped[int] = mapped_column(
        INTEGER(unsigned=True),
        primary_key=True,
        autoincrement=True,
    )
    artist_id: Mapped[int] = mapped_column(INTEGER(unsigned=True), nullable=False)


def remove_artist_info_references(session: Session, *, artist_info_id: int) -> None:
    """
    Remove all references to the given artist_info_id.
    """
    session.execute(
        delete(ArtistYtDeliverySettings).where(
            ArtistYtDeliverySettings.artist_id == artist_info_id
        )
    )