generated from coulomb/repo-seed
98 lines
3.6 KiB
Python
98 lines
3.6 KiB
Python
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
|