from dataclasses import dataclass, field
from datetime import date
from typing import Optional, List, Set


@dataclass
class PerformanceMetricsDataHealthSchema:
    startDate: date
    endDate: date
    labelIds: Optional[List[int]]


@dataclass
class DataHealthResponseUnit:
    startDate: date
    status: Optional[str]
    endDate: Optional[date] = None
    adAccounts: Optional[Set[str]] = field(default_factory=set)


@dataclass
class PerformanceMetricsDataHealthResponse:
    linkfire: Optional[List[DataHealthResponseUnit]]
    facebook: Optional[List[DataHealthResponseUnit]]
    google: Optional[List[DataHealthResponseUnit]]

    def __init__(self):
        self.linkfire = None
        self.facebook = None
        self.google = None


@dataclass
class ReportingDataHeathQuery:
    startDate: date
    endDate: date


@dataclass
class ReportingDataHealthResponse:
    amazon: List[DataHealthResponseUnit]
    apple: List[DataHealthResponseUnit]
    spotify: List[DataHealthResponseUnit]