import datetime
from dataclasses import dataclass

from pydantic import BaseModel

from dmp.app_connections.enums import AppConnectionStatus


@dataclass
class CollectionArtist:
    collection_id: int
    global_participant_id: str


class ShopifyCollectionWithArtist(BaseModel):
    id: int
    title: str
    products_count: int
    global_participant_id: str | None = None


class ShopifyStoreArtist(BaseModel):
    global_participant_id: str
    collections_count: int
    products_count: int


class ShopifyStore(BaseModel):
    id: str
    name: str | None
    shop_domain: str
    custom_domain: str | None
    status: AppConnectionStatus
    vendor_id: int
    subaccount_id: int
    global_participant_id: str | None
    collections_count: int | None
    products_count: int | None
    artists_count: int | None
    synced_at: datetime.datetime | None