root
141 lines
4.1 KiB
Python
141 lines
4.1 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
from typing import Any
|
|
|
|
from pydantic import BaseModel, ConfigDict, Field
|
|
|
|
|
|
class CaseBase(BaseModel):
|
|
model_config = ConfigDict(extra='allow')
|
|
|
|
age: int | None = None
|
|
age_years: int | None = None
|
|
gender: str | None = None
|
|
child_name: str | None = None
|
|
height_build: str | None = None
|
|
clothes_upper: str | None = None
|
|
clothes_lower: str | None = None
|
|
clothes_shoes: str | None = None
|
|
shoes: str | None = None
|
|
clothes_description: str | None = None
|
|
special_marks: str | None = None
|
|
phone_status: str | None = None
|
|
health_flags: list[str] = Field(default_factory=list)
|
|
has_diagnosis: bool | None = None
|
|
diagnosis_type: list[str] = Field(default_factory=list)
|
|
fitness_level: str | None = None
|
|
has_transport: str | None = None
|
|
cant_swim: bool | None = None
|
|
psychotype: str | None = None
|
|
psychotype_answers: dict[str, Any] | None = None
|
|
elapsed_hours: float | None = None
|
|
last_known_place: str | None = None
|
|
direction: str | None = None
|
|
reason: str | None = None
|
|
loss_reason: str | None = None
|
|
loss_time: datetime | None = None
|
|
last_seen_direction: str | None = None
|
|
last_seen_reliability: str | None = None
|
|
last_seen_description: str | None = None
|
|
behavior_description: str | None = None
|
|
familiar_places: str | None = None
|
|
lost_before: str | None = None
|
|
terrain_primary: str | None = None
|
|
terrain: list[str] = Field(default_factory=list)
|
|
weather: str | None = None
|
|
season: str | None = None
|
|
temperature_c: float | None = None
|
|
precipitation: str | None = None
|
|
visibility: str | None = None
|
|
wind: str | None = None
|
|
gps: dict[str, float] | None = None
|
|
gps_lat: float | None = None
|
|
gps_lon: float | None = None
|
|
tnp_lat: float | None = None
|
|
tnp_lon: float | None = None
|
|
tnp_address: str | None = None
|
|
resources: list[str] = Field(default_factory=list)
|
|
teams_count: int | None = None
|
|
team_size: int | None = None
|
|
has_dog: bool | None = None
|
|
extra_resources: list[str] = Field(default_factory=list)
|
|
notes: str | None = None
|
|
note: str | None = None
|
|
status: str | None = None
|
|
found_alive: bool | None = None
|
|
found_distance_km: float | None = None
|
|
found_direction: str | None = None
|
|
found_location_type: str | None = None
|
|
found_lat: float | None = None
|
|
found_lon: float | None = None
|
|
search_duration_hours: float | None = None
|
|
who_found: str | None = None
|
|
raw_text: str | None = None
|
|
analysis_log: dict[str, Any] | None = None
|
|
mobile: bool = False
|
|
|
|
|
|
class CaseCreate(CaseBase):
|
|
status: str | None = 'new'
|
|
|
|
|
|
class CaseUpdate(CaseBase):
|
|
pass
|
|
|
|
|
|
class CaseResponse(CaseBase):
|
|
id: str
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
status: str
|
|
health_flags: list[str] = Field(default_factory=list)
|
|
diagnosis_type: list[str] = Field(default_factory=list)
|
|
terrain: list[str] = Field(default_factory=list)
|
|
resources: list[str] = Field(default_factory=list)
|
|
extra_resources: list[str] = Field(default_factory=list)
|
|
result: dict[str, Any] | None = None
|
|
|
|
|
|
class CaseListResponse(BaseModel):
|
|
items: list[CaseResponse]
|
|
total: int
|
|
page: int
|
|
page_size: int
|
|
|
|
|
|
class DashboardResponse(BaseModel):
|
|
total_cases: int
|
|
found_alive_count: int
|
|
found_deceased_count: int
|
|
unknown_outcome_count: int
|
|
median_found_distance_km: float | None
|
|
recent_activity_count: int
|
|
|
|
|
|
class RecommendationRequest(BaseModel):
|
|
age: int | None = None
|
|
gender: str | None = None
|
|
health_flags: list[str] = Field(default_factory=list)
|
|
elapsed_hours: int | None = None
|
|
terrain_primary: str | None = None
|
|
weather: str | None = None
|
|
resources: list[str] = Field(default_factory=list)
|
|
|
|
|
|
class AnalyzeResponse(BaseModel):
|
|
urgency: str
|
|
primary_zones: list[dict[str, Any]]
|
|
immediate_actions: list[str]
|
|
behavioral_prediction: str
|
|
psychotype_recommendations: dict[str, Any]
|
|
key_locations: list[str]
|
|
case_id: str
|
|
|
|
|
|
class ParseDocResponse(BaseModel):
|
|
filename: str
|
|
parsed: bool
|
|
preview: dict[str, Any]
|
|
raw_text: str | None = None
|