from collections.abc import Sequence

import sqlalchemy as sa

from email_campaigns.adapters.db import Repository
from email_campaigns.emails.models import EmailDomain


class EmailDomainRepository(Repository[EmailDomain]):
    def find_by_brand(self, brand: str) -> Sequence[EmailDomain]:
        stmt = sa.select(EmailDomain).where(EmailDomain.brand == brand)
        return self.db.session.execute(stmt).scalars().all()