X-Git-Url: https://git.mdrn.pl/prawokultury.git/blobdiff_plain/7a47f04940f28745309716d725b63a8278b29e62..fceff8de9a4a304549e5b4b3094f66071b9912ae:/events/models.py diff --git a/events/models.py b/events/models.py index e20ae53..1151a11 100644 --- a/events/models.py +++ b/events/models.py @@ -6,11 +6,13 @@ from django.conf import settings from django.core.exceptions import ValidationError from django.db import models from django.utils.translation import ugettext_lazy as _, ugettext -from migdal.helpers import add_translatable +from fnpdjango.utils.models.translation import add_translatable class Event(models.Model): date = models.DateTimeField(_('date'), max_length=255, db_index=True) + date_end = models.DateTimeField(_('end date'), max_length=255, + db_index=True, blank=True) link = models.URLField(_('link')) class Meta: @@ -22,6 +24,12 @@ class Event(models.Model): return self.title def clean(self): + if self.date_end: + if self.date_end < self.date: + raise ValidationError( + ugettext("End date must not be earlier than start.")) + else: + self.date_end = self.date for lc, ln in settings.LANGUAGES: if (getattr(self, "published_%s" % lc) and not getattr(self, "title_%s" % lc)):