from typing import List
from pydantic import BaseModel

from src.schema.comics.comic import ComicResponse
from src.schema.comics.issue_details import IssueDetailsResponse
from src.schema.comics.worktype import WorktypeResponse


class ArtistWorktypeComicResponse(BaseModel):
    worktype: WorktypeResponse
    comics: List[ComicResponse]

class ArtistWorktypeIssueResponse(BaseModel):
    worktype: WorktypeResponse
    issues: List[IssueDetailsResponse]

class ArtistDetailResponse(BaseModel):
    id: str
    name: str
    weblink: str
    comic_works: List[ArtistWorktypeComicResponse]
    issue_works: List[ArtistWorktypeIssueResponse]