from app.adapters.db import Repository
from app.models import BatchRecipient


class BatchRecipientRepository(Repository[BatchRecipient]):
    def create_bulk(self, *, campaign_id: str, fansifter_only: bool) -> int:
        query = self.db.query_from_template(
            "batch_recipient/create_bulk.sql",
            context={
                "campaign_id": campaign_id,
                "fansifter_only": fansifter_only,
            },
        )
        result = self.db.session.connection().execute(query)
        return result.rowcount