Import Vector lab project

2026-06-06 18:31:55 +00:00
commit fcc22063ed
105 changed files with 31639 additions and 0 deletions
@@ -0,0 +1,125 @@
+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())