from django.db import models

class MaterialCategory(models.Model):
    """Категория материала (например, Труба, Лист, Круг)"""
    name = models.CharField("Название категории", max_length=100, unique=True)
    gost_standard = models.CharField("ГОСТ на тип проката", max_length=255, blank=True, help_text="Напр: ГОСТ 8639-82")

    class Meta:
        verbose_name = "Категория материала"
        verbose_name_plural = "Категории материалов"

    def __str__(self):
        return self.name

class SteelGrade(models.Model):
    """Марка стали (например, Ст3сп, 09Г2С) и связанные с ней ГОСТы"""
    name = models.CharField("Марка стали", max_length=100, unique=True)
    gost_standard = models.CharField("ГОСТ/ТУ", max_length=255, blank=True, help_text="Основной стандарт для этой марки")
    certificate_pdf = models.FileField("Сертификат/ГОСТ (PDF)", upload_to='certificates/', blank=True, null=True)

    class Meta:
        verbose_name = "Марка стали"
        verbose_name_plural = "Марки стали"

    def __str__(self):
        return f"{self.name} ({self.gost_standard})" if self.gost_standard else self.name

class Material(models.Model):
    """Конкретная номенклатурная единица (например, Труба 100х100х4)"""
    category = models.ForeignKey(MaterialCategory, on_delete=models.PROTECT, verbose_name="Категория")
    steel_grade = models.ForeignKey(SteelGrade, on_delete=models.PROTECT, verbose_name="Марка стали", null=True, blank=True)
    name = models.CharField("Наименование (размер/характеристики)", max_length=255)
    full_name = models.CharField("Полное наименование", max_length=500, blank=True, help_text="Генерируется автоматически, если пусто")
    
    class Meta:
        verbose_name = "Материал (номенклатура)"
        verbose_name_plural = "Материалы"

    def save(self, *args, **kwargs):
        if not self.full_name:
            grade_str = f" {self.steel_grade.name}" if self.steel_grade else ""
            self.full_name = f"{self.category.name} {self.name}{grade_str}"
        super().save(*args, **kwargs)

    def __str__(self):
        return self.full_name or f"{self.category.name} {self.name}"