Add related lessons.

[edumed.git] / curriculum / models.py

diff --git a/curriculum/models.py b/curriculum/models.py
index 399536b..14c916f 100644 (file)
--- a/curriculum/models.py
+++ b/curriculum/models.py
@@ -1,3 +1,6 @@
+# -*- coding: utf-8
+import re
+from django.core.urlresolvers import reverse
 from django.db import models
 from django.utils.translation import ugettext_lazy as _
 
@@ -28,9 +31,21 @@ class Competence(models.Model):
     def __unicode__(self):
         return self.name
 
+    def get_absolute_url(self):
+        return "%s?c=%d" % (reverse("curriculum"), self.pk)
+
+    def for_level(self, level):
+        return self.competencelevel_set.get(level=level)
+
+    @classmethod
+    def from_text(cls, text):
+        name = text.rsplit(u'\u2013', 1)[-1].strip()
+        return cls.objects.get(name__iexact=name)
+
 class Level(models.Model):
+    group = models.CharField(_('group'), max_length=255)
     name = models.CharField(_('name'), max_length=255)
-    slug = models.SlugField(_('slug'))
+    slug = models.CharField(_('slug'), max_length=255)
     order = models.IntegerField(_('order'))
 
     class Meta:
@@ -52,4 +67,76 @@ class CompetenceLevel(models.Model):
         verbose_name_plural = _('competences on levels')
 
     def __unicode__(self):
-        return "%s/%s" % (self.competence, self.level)
+        return u"%s/%s" % (self.competence, self.level)
+
+    def get_absolute_url(self):
+        return "%s?c=%d&level=%s&d=1" % (reverse("curriculum"), self.competence.pk, self.level.slug)
+
+
+
+class CurriculumLevel(models.Model):
+    title = models.CharField(max_length=16, db_index=True)
+
+    class Meta:
+        verbose_name = _("curriculum level")
+        verbose_name_plural = _("curriculum levels")
+
+    def __unicode__(self):
+        return self.title
+
+
+class CurriculumCourse(models.Model):
+    title = models.CharField(max_length=255)
+    slug = models.CharField(max_length=255, db_index=True)
+
+    class Meta:
+        verbose_name = _("curriculum course")
+        verbose_name_plural = _("curriculum courses")
+
+    def __unicode__(self):
+        return self.title
+
+
+class Curriculum(models.Model):
+    """Official curriculum."""
+    TYPES = {'c': u'Cele kształcenia', 't': u'Treści nauczania'}
+
+    identifier = models.CharField(max_length=255, db_index=True)
+    title = models.CharField(max_length=255)
+    course = models.ForeignKey(CurriculumCourse)
+    level = models.ForeignKey(CurriculumLevel)
+    type = models.CharField(max_length=16, choices=TYPES.items())
+
+    class Meta:
+        verbose_name = _("curriculum item")
+        verbose_name_plural = _("curriculum items")
+
+    def __unicode__(self):
+        return self.identifier
+
+    @classmethod
+    def from_text(cls, identifier, title):
+        m = re.match(r"^\d+/(?P<level>[^/]+)/(?P<course>[^/]+)/"
+                     "(?P<type>(?:%s))[^/]+(?P<roz>/roz)?" %
+                        "|".join(cls.TYPES), identifier)
+        assert m is not None, "Curriculum identifier doesn't match template."
+        level, created = CurriculumLevel.objects.get_or_create(
+                                       title=m.group('level'))
+        course, created = CurriculumCourse.objects.get_or_create(
+                                        slug=m.group('course'),
+                                        defaults={'title': m.group('course').title()})
+        type_ = m.group('type')
+        if m.group('roz'):
+            title += " (zakres rozszerzony)"
+
+        try:
+            curr = cls.objects.get(identifier=identifier)
+        except cls.DoesNotExist:
+            curr = cls(identifier=identifier)
+        curr.title = title
+        curr.course = course
+        curr.level = level
+        curr.type = type_
+        curr.save()
+        return curr
+