root
126 lines
4.3 KiB
Python
126 lines
4.3 KiB
Python
from sqlalchemy import Column, Integer, String, Float, Boolean, DateTime, Text, ARRAY, ForeignKey
|
||
from sqlalchemy.dialects.postgresql import UUID, JSONB
|
||
from sqlalchemy.sql import func
|
||
from sqlalchemy.orm import relationship
|
||
import uuid
|
||
from database import Base
|
||
|
||
|
||
class User(Base):
|
||
"""User model for authentication"""
|
||
__tablename__ = "users"
|
||
|
||
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
||
username = Column(String(100), unique=True, nullable=False)
|
||
email = Column(String(255), unique=True, nullable=False)
|
||
hashed_password = Column(String(255), nullable=False)
|
||
full_name = Column(String(255))
|
||
role = Column(String(20), nullable=False, default="operator")
|
||
is_active = Column(Boolean, default=True)
|
||
created_at = Column(DateTime, server_default=func.now())
|
||
last_login = Column(DateTime)
|
||
|
||
|
||
class Case(Base):
|
||
"""Unified case model - combines search case and result"""
|
||
__tablename__ = "cases"
|
||
|
||
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
||
created_at = Column(DateTime, server_default=func.now())
|
||
status = Column(String(20), default="active")
|
||
|
||
# Ребёнок (Шаг 1)
|
||
child_name = Column(String(255))
|
||
age_years = Column(Integer, nullable=False)
|
||
gender = Column(String(1))
|
||
height_build = Column(Text)
|
||
clothes_upper = Column(Text)
|
||
clothes_lower = Column(Text)
|
||
shoes = Column(Text)
|
||
clothes_description = Column(Text)
|
||
special_marks = Column(Text)
|
||
phone_status = Column(String(20))
|
||
|
||
# Здоровье (Шаг 2)
|
||
has_diagnosis = Column(Boolean, default=False)
|
||
diagnosis_type = Column(ARRAY(String))
|
||
fitness_level = Column(String(20))
|
||
has_transport = Column(String(20), default="none")
|
||
cant_swim = Column(Boolean, default=False)
|
||
|
||
# Психотип (Шаг 2б)
|
||
psychotype = Column(String(50))
|
||
psychotype_answers = Column(JSONB)
|
||
|
||
# Обстоятельства (Шаг 3)
|
||
loss_reason = Column(String(100))
|
||
loss_time = Column(DateTime)
|
||
elapsed_hours = Column(Float)
|
||
last_seen_direction = Column(String(10))
|
||
last_seen_reliability = Column(String(20))
|
||
last_seen_description = Column(Text)
|
||
behavior_description = Column(Text)
|
||
familiar_places = Column(Text)
|
||
lost_before = Column(String(20))
|
||
|
||
# Среда (Шаг 4)
|
||
season = Column(String(20))
|
||
temperature_c = Column(Float)
|
||
precipitation = Column(String(20))
|
||
visibility = Column(String(20))
|
||
wind = Column(String(20))
|
||
terrain = Column(ARRAY(String))
|
||
|
||
# GPS (Шаг 4)
|
||
tnp_lat = Column(Float)
|
||
tnp_lon = Column(Float)
|
||
tnp_address = Column(Text)
|
||
|
||
# Ресурсы (Шаг 5)
|
||
teams_count = Column(Integer)
|
||
team_size = Column(Integer)
|
||
has_dog = Column(Boolean, default=False)
|
||
extra_resources = Column(ARRAY(String))
|
||
|
||
# Исход (заполняется после завершения)
|
||
found_alive = Column(Boolean)
|
||
found_distance_km = Column(Float)
|
||
found_direction = Column(String(10))
|
||
found_location_type = Column(String(50))
|
||
found_lat = Column(Float)
|
||
found_lon = Column(Float)
|
||
search_duration_hours = Column(Float)
|
||
who_found = Column(String(50))
|
||
|
||
# Мета
|
||
confidence_avg = Column(Float)
|
||
raw_text = Column(Text)
|
||
analysis_log = Column(JSONB)
|
||
|
||
|
||
class AnalysisLog(Base):
|
||
"""Analysis log for audit and debugging"""
|
||
__tablename__ = "analysis_log"
|
||
|
||
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
||
case_id = Column(UUID(as_uuid=True), ForeignKey("cases.id", ondelete="CASCADE"), nullable=False)
|
||
user_id = Column(UUID(as_uuid=True), ForeignKey("users.id", ondelete="SET NULL"))
|
||
analysis_type = Column(String(50), nullable=False)
|
||
input_data = Column(JSONB)
|
||
output_data = Column(JSONB)
|
||
execution_time = Column(Float)
|
||
status = Column(String(20), default="success")
|
||
error_message = Column(Text)
|
||
created_at = Column(DateTime, server_default=func.now())
|
||
|
||
|
||
class RawDocument(Base):
|
||
"""Raw document storage for parsed reports"""
|
||
__tablename__ = "raw_documents"
|
||
|
||
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
||
filename = Column(String(255), nullable=False)
|
||
raw_text = Column(Text)
|
||
extracted_json = Column(JSONB)
|
||
created_at = Column(DateTime, server_default=func.now())
|