from django.db import models


class CareerTestResult(models.Model):
    PROFILE_CHOICES = [
        ('R', 'Практик'),
        ('I', 'Исследователь'),
        ('A', 'Творец'),
        ('S', 'Социальный'),
        ('E', 'Лидер'),
        ('C', 'Организатор'),
    ]

    created_at = models.DateTimeField(auto_now_add=True)
    updated_at = models.DateTimeField(auto_now=True)

    # Данные от пользователя
    ym_client_id = models.CharField(max_length=64, null=True, blank=True, db_index=True)
    ip_address = models.GenericIPAddressField(null=True, blank=True)
    user_agent = models.TextField(null=True, blank=True)

    # Ответы (q1-q24)
    answers = models.JSONField(default=dict)

    # Результаты расчёта
    profile = models.CharField(max_length=1, choices=PROFILE_CHOICES)
    top3_profiles = models.JSONField(default=list)  # ['I', 'E', 'C']

    # Рекомендованные наборы ЕГЭ
    recommended_ege_sets = models.JSONField(default=list)

    # Для отправки в Метрику
    uploaded_to_metrica = models.BooleanField(default=False)

    class Meta:
        verbose_name = "Результат теста"
        verbose_name_plural = "Результаты тестов"
        ordering = ['-created_at']
        indexes = [
            models.Index(fields=['ym_client_id', '-created_at']),
        ]

    def __str__(self):
        return f"Тест {self.get_profile_display()} от {self.created_at.strftime('%d.%m.%Y %H:%M')}"