from fansifter_common.testing.factories import SQLAlchemyFactory
from polyfactory import Ignore

from app.models import BatchRecipient, Campaign, CampaignBatch


class CampaignFactory(SQLAlchemyFactory[Campaign]):
    deleted_at = None


class CampaignBatchFactory(SQLAlchemyFactory[CampaignBatch]):
    id = Ignore()
    completed_at = None
    cancelled_at = None


class BatchRecipientFactory(SQLAlchemyFactory[BatchRecipient]):
    id = Ignore()
    last_hourly_quota = None
    last_hour_sent_emails = 0