diff --git a/seminar/models.py b/seminar/models.py index 566b9d8d..15367d3a 100644 --- a/seminar/models.py +++ b/seminar/models.py @@ -11,6 +11,7 @@ from django.utils.encoding import force_unicode from django.utils.text import slugify from django.core.urlresolvers import reverse from django.core.cache import cache +from django.core.exceptions import ObjectDoesNotExist from imagekit.models import ImageSpecField, ProcessedImageField from imagekit.processors import ResizeToFit, Transpose @@ -28,6 +29,7 @@ from reversion import revisions as reversion from seminar.utils import roman + class SeminarModelBase(models.Model): class Meta: @@ -382,6 +384,15 @@ class Cislo(SeminarModelBase): return None return cs[i] + @classmethod + def get(cls, rocnik, cislo): + try: + r = Rocnik.objects.get(rocnik=rocnik) + c = r.cisla.get(cislo=cislo) + except ObjectDoesNotExist: + return None + return c + @reversion.register(ignore_duplicate_revisions=True) @python_2_unicode_compatible