Prototype implementation

vergabe_teilnahme/apps/dokumente/models.py

@@ -0,0 +1,97 @@
+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/')
+    dateiname = models.CharField(max_length=300, blank=True)
+    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