7230b23ab496a9678cffb82b8e0a4890549a28a1
[edumed.git] / catalogue / forms.py
1 # -*- coding: utf-8 -*-
2 import json
3 import os.path
4 import shutil
5 import urllib
6 from tempfile import mkdtemp
7
8 from django.forms import Form, CharField
9
10 from librarian import IOFile
11 from catalogue.models import Lesson
12
13
14 class LessonImportForm(Form):
15     lesson_xml = CharField()
16     gallery_url = CharField(required=False)
17     attachments = CharField(required=False)
18
19     def save(self):
20         temp_dir = mkdtemp()
21         attachment_names = json.loads(self.cleaned_data['attachments'])
22         attachments = {}
23         remote_gallery_url = self.cleaned_data['gallery_url']
24         if remote_gallery_url and attachment_names:
25             for attachment_name in attachment_names:
26                 attachment_url = ('%s%s' % (remote_gallery_url, attachment_name)).encode('utf-8')
27                 temp_filename = os.path.join(temp_dir, attachment_name)
28                 urllib.urlretrieve(attachment_url, temp_filename)
29                 attachments[attachment_name] = IOFile.from_filename(temp_filename)
30
31         lesson = Lesson.publish(
32             IOFile.from_string(self.cleaned_data['lesson_xml'], attachments=attachments))
33         if os.path.isdir(temp_dir):
34             shutil.rmtree(temp_dir)
35         return lesson