catalogue/models.py

   1 from django.core.files import File
   2 from django.db import models
   3 from curriculum.models import Level
   4
   5 class Section(models.Model):
   6     title = models.CharField(max_length=255, unique=True)
   7     slug = models.SlugField(unique=True)
   8     order = models.IntegerField()
   9
  10     class Meta:
  11         ordering = ['order']
  12
  13     def __unicode__(self):
  14         return self.title
  15
  16 class Lesson(models.Model):
  17     section = models.ForeignKey(Section)
  18     level = models.ForeignKey(Level)
  19     title = models.CharField(max_length=255)
  20     slug = models.SlugField(unique=True)
  21     depth = models.IntegerField()
  22     order = models.IntegerField()
  23
  24     xml_file = models.FileField(upload_to="catalogue/lesson/xml",
  25         null=True, blank=True) # FIXME: slug in paths
  26     html_file = models.FileField(upload_to="catalogue/lesson/html",
  27         null=True, blank=True)
  28     package = models.FileField(upload_to="catalogue/lesson/pack",
  29         null=True, blank=True)
  30     student_package = models.FileField(upload_to="catalogue/lesson/student_pack",
  31         null=True, blank=True)
  32     pdf = models.FileField(upload_to="catalogue/lesson/pdf",
  33         null=True, blank=True)
  34     student_pdf = models.FileField(upload_to="catalogue/lesson/student_pdf",
  35         null=True, blank=True)
  36
  37     class Meta:
  38         ordering = ['section', 'level', 'depth', 'order']
  39
  40     def __unicode__(self):
  41         return self.title
  42
  43     @models.permalink
  44     def get_absolute_url(self):
  45         return ('catalogue_lesson', [self.slug])
  46
  47     @classmethod
  48     def publish(cls, infile):
  49         from librarian.parser import WLDocument
  50         from django.core.files.base import ContentFile
  51         # infile should be IOFile, now it's a regular file
  52         xml = infile.read()
  53         wldoc = WLDocument.from_string(xml)
  54         slug = wldoc.book_info.url.slug
  55
  56         try:
  57             lesson = cls.objects.get(slug=slug)
  58         except cls.DoesNotExist:
  59             lesson = cls(slug=slug)
  60
  61         # Save XML file
  62         lesson.xml_file.save('%s.xml' % slug, ContentFile(xml), save=False)
  63         lesson.title = wldoc.book_info.title
  64         #book.extra_info = wldoc.book_info.to_dict()
  65
  66         # FIXME:
  67         #lesson.level = Level.objects.get(slug=wldoc.book_info.audience)
  68         lesson.level = Level.objects.get(name=wldoc.book_info.audience)
  69         # TODO: no xml data?
  70         lesson.section = Section.objects.all()[0]
  71         lesson.order = 1
  72         lesson.depth = 1
  73         lesson.save()
  74         lesson.build_html()
  75         return lesson
  76
  77     def build_html(self):
  78         from librarian.parser import WLDocument
  79         wldoc = WLDocument.from_file(self.xml_file.path)
  80         html = wldoc.as_html()
  81         self.html_file.save("%s.html" % self.slug,
  82             File(open(html.get_filename())), save=False)
  83         self.save()
  84
  85 class Attachment(models.Model):
  86     lesson = models.ForeignKey(Lesson)
  87     file = models.FileField(upload_to="catalogue/attachment")
  88
  89
  90 class Part(models.Model):
  91     lesson = models.ForeignKey(Lesson)
  92     pdf = models.FileField(upload_to="catalogue/part/pdf",
  93         null=True, blank=True)
  94     student_pdf = models.FileField(upload_to="catalogue/part/student_pdf",
  95         null=True, blank=True)