stage2/forms.py

   1 # -*- coding: utf-8 -*-
   2 from django import forms
   3 from django.conf import settings
   4 from django.template.defaultfilters import filesizeformat
   5
   6 from stage2.models import Attachment, Mark, FieldOptionSet, FieldOption
   7
   8
   9 class AttachmentForm(forms.ModelForm):
  10     assignment_id = forms.CharField(widget=forms.HiddenInput)
  11
  12     class Meta:
  13         model = Attachment
  14         fields = ['file']
  15
  16     def __init__(self, assignment, file_no, label, extensions=None, *args, **kwargs):
  17         prefix = 'att%s-%s' % (assignment.id, file_no)
  18         super(AttachmentForm, self).__init__(*args, prefix=prefix, **kwargs)
  19         self.fields['assignment_id'].initial = assignment.id
  20         self.fields['file'].label = label
  21         if extensions:
  22             self.fields['file'].widget.attrs = {'data-ext': '|'.join(extensions)}
  23         self.extensions = extensions
  24
  25     def clean_file(self):
  26         file = self.cleaned_data['file']
  27         if file.size > settings.MAX_UPLOAD_SIZE:
  28             raise forms.ValidationError(
  29                 'Please keep filesize under %s. Current filesize: %s' % (
  30                     filesizeformat(settings.MAX_UPLOAD_SIZE), filesizeformat(file.size)))
  31         if self.extensions and ('.' not in file.name or file.name.rsplit('.', 1)[1].lower() not in self.extensions):
  32             raise forms.ValidationError('Incorrect extension, should be one of: %s' % ', '.join(self.extensions))
  33         return file
  34
  35
  36 class AssignmentFieldForm(forms.Form):
  37     value = forms.CharField()
  38     assignment_id = forms.CharField(widget=forms.HiddenInput)
  39
  40     def __init__(self, label, field_no, options, answer, *args, **kwargs):
  41         prefix = 'field%s-%s' % (answer.id, field_no)
  42         super(AssignmentFieldForm, self).__init__(prefix=prefix, *args, **kwargs)
  43         self.answer = answer
  44         self.label = label
  45         self.fields['value'].label = label
  46         self.type = options['type']
  47         self.fields['assignment_id'].initial = answer.assignment.id
  48         if self.type == 'options':
  49             option_set = FieldOptionSet.objects.get(name=options['option_set'])
  50             self.fields['value'].widget = forms.Select(choices=option_set.choices(answer))
  51             options = answer.fieldoption_set.all()
  52             if options:
  53                 self.fields['value'].initial = options.get().id
  54         else:
  55             value = answer.field_values.get(label)
  56             self.fields['value'].initial = value or ''
  57
  58     def clean_value(self):
  59         if self.type == 'options':
  60             option = FieldOption.objects.get(id=int(self.cleaned_data['value']))
  61             if option.answer != self.answer and option.answer is not None:
  62                 raise forms.ValidationError(u'Ta opcja została już wybrana przez kogoś innego.')
  63             return option
  64         return self.cleaned_data['value']
  65
  66     def save(self):
  67         value = self.cleaned_data['value']
  68         if self.type == 'options':
  69             option = value
  70             if option.answer != self.answer:
  71                 # not thread-safe :/
  72                 assert not option.answer
  73                 for opt in self.answer.fieldoption_set.all():
  74                     opt.answer = None
  75                     opt.save()
  76                 option.answer = self.answer
  77                 option.save()
  78         else:
  79             self.answer.field_values[self.label] = value
  80             self.answer.save()
  81
  82
  83 class MarkForm(forms.ModelForm):
  84     class Meta:
  85         model = Mark
  86         fields = ['points']
  87         widgets = {
  88             'points': forms.TextInput(attrs={'type': 'number', 'min': 0, 'step': '0.5'})
  89         }
  90
  91     def __init__(self, answer, *args, **kwargs):
  92         super(MarkForm, self).__init__(*args, **kwargs)
  93         self.answer = answer
  94         self.fields['points'].widget.attrs['max'] = answer.assignment.max_points
  95
  96     def clean_points(self):
  97         points = self.cleaned_data['points']
  98         if points > self.answer.assignment.max_points:
  99             raise forms.ValidationError('Too many points for this assignment')
 100         if points < 0:
 101             raise forms.ValidationError('Points cannot be negative')
 102         return points