WIP better import drops

twitch/schemas.py

@@ -0,0 +1,131 @@
+from __future__ import annotations
+
+from typing import Literal
+
+from pydantic import BaseModel
+from pydantic import Field
+
+
+class Organization(BaseModel):
+    """Schema for Twitch Organization objects."""
+
+    twitch_id: str = Field(alias="id")
+    name: str
+    type_name: Literal["Organization"] = Field(alias="__typename")
+
+    model_config = {
+        "extra": "forbid",
+        "validate_assignment": True,
+        "strict": True,
+        "populate_by_name": True,
+    }
+
+
+class Game(BaseModel):
+    """Schema for Twitch Game objects."""
+
+    twitch_id: str = Field(alias="id")
+    display_name: str = Field(alias="displayName")
+    box_art_url: str = Field(alias="boxArtURL")
+    type_name: Literal["Game"] = Field(alias="__typename")
+
+    model_config = {
+        "extra": "forbid",
+        "validate_assignment": True,
+        "strict": True,
+        "populate_by_name": True,
+    }
+
+
+class DropCampaignSelfEdge(BaseModel):
+    """Schema for the 'self' edge on DropCampaign objects."""
+
+    is_account_connected: bool = Field(alias="isAccountConnected")
+    type_name: Literal["DropCampaignSelfEdge"] = Field(alias="__typename")
+
+    model_config = {
+        "extra": "forbid",
+        "validate_assignment": True,
+        "strict": True,
+        "populate_by_name": True,
+    }
+
+
+class DropCampaign(BaseModel):
+    """Schema for Twitch DropCampaign objects."""
+
+    twitch_id: str = Field(alias="id")
+    name: str
+    owner: Organization
+    game: Game
+    status: Literal["ACTIVE", "EXPIRED"]
+    start_at: str = Field(alias="startAt")
+    end_at: str = Field(alias="endAt")
+    details_url: str = Field(alias="detailsURL")
+    account_link_url: str = Field(alias="accountLinkURL")
+    self: DropCampaignSelfEdge
+    type_name: Literal["DropCampaign"] = Field(alias="__typename")
+
+    model_config = {
+        "extra": "forbid",
+        "validate_assignment": True,
+        "strict": True,
+        "populate_by_name": True,
+    }
+
+
+class CurrentUser(BaseModel):
+    """Schema for Twitch User objects."""
+
+    twitch_id: str = Field(alias="id")
+    login: str
+    drop_campaigns: list[DropCampaign] = Field(alias="dropCampaigns")
+    type_name: Literal["User"] = Field(alias="__typename")
+
+    model_config = {
+        "extra": "forbid",
+        "validate_assignment": True,
+        "strict": True,
+        "populate_by_name": True,
+    }
+
+
+class Data(BaseModel):
+    """Schema for the data field in Twitch API responses."""
+
+    current_user: CurrentUser = Field(alias="currentUser")
+
+    model_config = {
+        "extra": "forbid",
+        "validate_assignment": True,
+        "strict": True,
+        "populate_by_name": True,
+    }
+
+
+class Extensions(BaseModel):
+    """Schema for the extensions field in Twitch API responses."""
+
+    duration_milliseconds: int = Field(alias="durationMilliseconds")
+    operation_name: Literal["ViewerDropsDashboard"] = Field(alias="operationName")
+    request_id: str = Field(alias="requestID")
+
+    model_config = {
+        "extra": "forbid",
+        "validate_assignment": True,
+        "strict": True,
+        "populate_by_name": True,
+    }
+
+
+class ViewerDropsDashboardPayload(BaseModel):
+    """Schema for the ViewerDropsDashboard response."""
+
+    data: Data
+    extensions: Extensions
+
+    model_config = {
+        "extra": "forbid",
+        "validate_assignment": True,
+        "strict": True,
+    }