|
|
from pydantic import BaseModel, Field |
|
|
from typing import Literal, Union, Optional, List |
|
|
|
|
|
|
|
|
|
|
|
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 |
