from django.conf import settings
from django.contrib.contenttypes.fields import GenericRelation
from django.core.exceptions import ValidationError
from django.db import models
from django.db.models.signals import pre_save
from django.dispatch import receiver

from vergabe_teilnahme.apps.core.models import FlexibleModel, Freigabe

ALLOWED_EXTENSIONS = {'.pdf', '.docx', '.xlsx', '.zip', '.png', '.jpg', '.jpeg'}


class Dokument(FlexibleModel):
    STATUS_CHOICES = [
        ('entwurf', 'Entwurf'),
        ('in_pruefung', 'In Prüfung'),
        ('freigegeben', 'Freigegeben'),
        ('final_abgegeben', 'Final abgegeben'),
        ('ersetzt', 'Ersetzt'),
        ('archiviert', 'Archiviert'),
    ]
    KATEGORIE_CHOICES = [
        ('angebotsunterlage', 'Angebotsunterlage'),
        ('leistungsbeschreibung', 'Leistungsbeschreibung'),
        ('preisblatt', 'Preisblatt'),
        ('referenz', 'Referenz'),
        ('nachweis', 'Nachweis'),
        ('abgabenachweis', 'Abgabenachweis'),
        ('kommunikation', 'Kommunikation'),
        ('intern', 'Intern'),
        ('sonstiges', 'Sonstiges'),
    ]

    ausschreibung = models.ForeignKey(
        'ausschreibungen.Ausschreibung', on_delete=models.SET_NULL,
        null=True, blank=True, related_name='dokumente'
    )
    los = models.ForeignKey(
        'lose.Los', on_delete=models.SET_NULL,
        null=True, blank=True, related_name='dokumente'
    )
    datei = models.FileField(upload_to='dokumente/%Y/%m/', null=True, blank=True)
    dateiname = models.CharField(max_length=300, blank=True)
    quelle = models.CharField(max_length=300, blank=True)
    bibliothek_nachweis = models.ForeignKey(
        'bibliothek.Nachweis', on_delete=models.SET_NULL,
        null=True, blank=True, related_name='verwendete_dokumente'
    )
    kategorie = models.CharField(max_length=30, choices=KATEGORIE_CHOICES, default='sonstiges')
    status = models.CharField(max_length=20, choices=STATUS_CHOICES, default='entwurf')
    version = models.CharField(max_length=20, default='1.0')
    beschreibung = models.TextField(blank=True)
    verantwortlicher = models.ForeignKey(
        'accounts.Mitarbeiter', on_delete=models.SET_NULL,
        null=True, blank=True, related_name='verantwortete_dokumente'
    )
    pruefer = models.ForeignKey(
        'accounts.Mitarbeiter', on_delete=models.SET_NULL,
        null=True, blank=True, related_name='zu_pruefende_dokumente'
    )
    finale_abgabeversion = models.BooleanField(default=False)
    upload_datum = models.DateTimeField(auto_now_add=True)
    freigaben_rel = GenericRelation(Freigabe)

    class Meta:
        ordering = ['-upload_datum']
        verbose_name = 'Dokument'
        verbose_name_plural = 'Dokumente'

    def __str__(self):
        return self.dateiname or self.datei.name

    def clean(self):
        if self.datei:
            import os
            ext = os.path.splitext(self.datei.name)[1].lower()
            if ext not in ALLOWED_EXTENSIONS:
                raise ValidationError(
                    f'Dateityp "{ext}" nicht erlaubt. Erlaubt: {", ".join(ALLOWED_EXTENSIONS)}'
                )
            max_size = getattr(settings, 'MAX_UPLOAD_SIZE', 52428800)
            if hasattr(self.datei, 'size') and self.datei.size > max_size:
                raise ValidationError(
                    f'Datei zu groß. Maximum: {max_size // 1024 // 1024} MB'
                )

    def save(self, *args, **kwargs):
        if self.datei and not self.dateiname:
            import os
            self.dateiname = os.path.basename(self.datei.name)
        super().save(*args, **kwargs)


@receiver(pre_save, sender=Dokument)
def set_final_status(sender, instance, **kwargs):
    if instance.pk:
        try:
            old = Dokument.objects.get(pk=instance.pk)
            if not old.finale_abgabeversion and instance.finale_abgabeversion:
                instance.status = 'final_abgegeben'
        except Dokument.DoesNotExist:
            pass