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):
        category_part = (self.category.name or '').strip() if self.category_id else ''
        name_part = (self.name or '').strip()
        grade_part = (self.steel_grade.name or '').strip() if self.steel_grade_id else ''

        self.full_name = ' '.join([p for p in [category_part, name_part, grade_part] if p])
        super().save(*args, **kwargs)

    def __str__(self):
        return self.full_name or ' '.join([p for p in [(self.category.name if self.category_id else ''), self.name, (self.steel_grade.name if self.steel_grade_id else '')] if p]).strip()