8cdfe79f90cca75f57f8b5ee02cf489ca10cbe61
[edumed.git] / wtem / forms.py
1 import os
2 import re
3
4 from django import forms
5 from django.utils import simplejson
6
7 from .models import Submission, Attachment, exercises
8
9
10 class WTEMForm(forms.ModelForm):
11     class Meta:
12         model = Submission
13         fields = ('answers',)
14
15     def __init__(self, *args, **kwargs):
16         super(WTEMForm, self).__init__(*args, **kwargs)
17         for exercise in exercises:
18             if exercise['type'] != 'file_upload':
19                 continue
20             self.fields['attachment_for_' + str(exercise['id'])] = forms.FileField(required = False)
21
22     def save(self):
23         submission = super(WTEMForm, self).save()
24         for name, file in self.files.items():
25             m = re.match(r'attachment_for_(\d+)(?:__(.*))?', name)
26             exercise_id = int(m.group(1))
27             tag = m.group(2) or None
28             try:
29                 attachment = Attachment.objects.get(submission = submission, exercise_id = exercise_id, tag=tag)
30             except Attachment.DoesNotExist:
31                 attachment = Attachment(submission = submission, exercise_id = exercise_id, tag=tag)
32             attachment.file = file
33             attachment.save()
34