from polyfactory.factories.pydantic_factory import ModelFactory

from app.adapters.ows_dmp import (
    AudienceShare,
    GoogleAudienceShare,
    MetaAudienceShare,
    TikTokAudience,
    TikTokAudienceShare,
)
from app.adapters.tiktok.types import AudienceFile
from app.schemas import S3Record


class S3RecordFactory(ModelFactory[S3Record]):
    __model__ = S3Record


class AudienceShareFactory(ModelFactory[AudienceShare]):
    __model__ = AudienceShare


class MetaAudienceShareFactory(ModelFactory[MetaAudienceShare]):
    __model__ = MetaAudienceShare


class TikTokAudienceShareFactory(ModelFactory[TikTokAudienceShare]):
    __model__ = TikTokAudienceShare


class TikTokAudienceFactory(ModelFactory[TikTokAudience]):
    __model__ = TikTokAudience


class GoogleAudienceShareFactory(ModelFactory[GoogleAudienceShare]):
    __model__ = GoogleAudienceShare


class TikTokAudienceFileFactory(ModelFactory[AudienceFile]):
    __model__ = AudienceFile