Some additional info while generating keys
[edumed.git] / catalogue / publish.py
1 # -*- coding: utf-8
2 from django.core.files.base import ContentFile
3 from django.core.files import File
4 from librarian import DocProvider, IOFile
5 from librarian.pyhtml import EduModuleFormat
6 from librarian.pypdf import EduModulePDFFormat
7 from .models import Lesson, Attachment
8
9
10 class HtmlFormat(EduModuleFormat):
11     def find_attachment(self, slug, fmt):
12         lesson_slug = self.wldoc.book_info.url.slug
13         try:
14             # If already saved, use it.
15             att = Attachment.objects.get(lesson__slug=lesson_slug,
16                                          slug=slug, ext=fmt)
17         except Attachment.DoesNotExist, e:
18             # If attached to source IOFile, save now.
19             att_name = "%s.%s" % (slug, fmt)
20             try:
21                 att_file = self.wldoc.source.attachments[att_name]
22             except KeyError:
23                 print u"ATTACHMENT MISSING:", att_name
24                 raise self.MaterialNotFound()
25             else:
26                 lesson = Lesson.objects.get(slug=lesson_slug)
27                 att = lesson.attachment_set.create(slug=slug, ext=fmt)
28                 att.file.save(att_name, File(att_file.get_file()))
29                 return att
30         else:
31             return att
32
33     def url_for_material(self, slug, fmt):
34         return self.find_attachment(slug, fmt).file.url
35
36     def url_for_image(self, slug, fmt, width=None):
37         if width is None:
38             return self.url_for_material(slug, fmt)
39
40         lesson_slug = self.wldoc.book_info.url.slug
41         th_slug = "thumb/%s__th%d" % (slug, width)
42         try:
43             # If already saved, use it.
44             att = Attachment.objects.get(lesson__slug=lesson_slug,
45                                          slug=th_slug, ext=fmt)
46         except Attachment.DoesNotExist, e:
47             from PIL import Image
48             from StringIO import StringIO
49             # Find full image, create thumbnail, save.
50             src_att = self.find_attachment(slug, fmt)
51             simg = Image.open(src_att.file.path)
52             size = (width, simg.size[1]*width/simg.size[0])
53             simg = simg.resize(size, Image.ANTIALIAS)
54
55             tempfile = StringIO()
56             img_format = "JPEG" if fmt.upper() == "JPG" else fmt
57             simg.save(tempfile, format=img_format)
58             att_name = "%s.%s" % (th_slug, fmt)
59             lesson = Lesson.objects.get(slug=lesson_slug)
60             att = lesson.attachment_set.create(slug=th_slug, ext=fmt)
61             att.file.save(att_name, ContentFile(tempfile.getvalue()))
62         return att.file.url
63
64 class PdfFormat(EduModulePDFFormat):
65     pass
66
67
68 class OrmDocProvider(DocProvider):
69     def by_slug(self, slug):
70         """Should return a file-like object with a WL document XML."""
71         return IOFile.from_filename(Lesson.objects.get(slug=slug).xml_file.path)