import sqlalchemy as sa

from dmp.adapters.db import ReportingRepository
from dmp.shopify.models import ShopifyCollectionArtist


class CollectionArtistRepository(ReportingRepository[ShopifyCollectionArtist]):
    def delete_by_association_id(self, association_id: str) -> None:
        query = sa.delete(ShopifyCollectionArtist).where(
            ShopifyCollectionArtist.association_id == association_id
        )
        self.db.session.execute(query)
        return None