from pydantic import BaseModel, Field
from typing import Literal, Union, Optional, List

# --- Event follow-up classes ---


class AnimalCollectedEvent(BaseModel):
    type: Literal["animal collected"]
    collected: Literal["yes", "no"]


class RecipientEvent(BaseModel):
    type: Literal["recipient"]
    recipient: Literal[
        "veterinary", "care center", "local museum", "national museum", "other"
    ]


class RadiographyEvent(BaseModel):
    type: Literal["radiography"]
    radiography: Literal["yes", "no", "unknown"]


class GivenAnswerEvent(BaseModel):
    type: Literal["given answer"]
    answer: Literal[
        "nothing",
        "complaint against x",
        "complaint",
        "police call",
        "discussion with the speaker",
        "press release",
        "unknown",
    ]


class NameOfRecipientEvent(BaseModel):
    type: Literal["recipient name"]
    name: str


class CollectionReferenceEvent(BaseModel):
    type: Literal["collection reference"]
    reference: str


FollowUpEventType = Union[
    AnimalCollectedEvent,
    RecipientEvent,
    RadiographyEvent,
    GivenAnswerEvent,
    NameOfRecipientEvent,
    CollectionReferenceEvent,
]


class FollowUpEvents(BaseModel):
    follow_up_events: Optional[List[FollowUpEventType]] = None