most of stage 2

[edumed.git] / stage2 / forms.py

diff --git a/stage2/forms.py b/stage2/forms.py
index 9ff44f7..f7d248e 100644 (file)
--- a/stage2/forms.py
+++ b/stage2/forms.py
@@ -1,7 +1,10 @@
 # -*- coding: utf-8 -*-
 from django import forms
+from django.conf import settings
+from django.core.exceptions import ValidationError
+from django.template.defaultfilters import filesizeformat
 
-from stage2.models import Attachment
+from stage2.models import Attachment, Mark
 
 
 class AttachmentForm(forms.ModelForm):
@@ -13,3 +16,33 @@ class AttachmentForm(forms.ModelForm):
         prefix = 'att%s-%s' % (assignment.id, file_no)
         super(AttachmentForm, self).__init__(*args, prefix=prefix, **kwargs)
         self.fields['file'].label = label
+
+    def clean_file(self):
+        file = self.cleaned_data['file']
+        if file.size > settings.MAX_UPLOAD_SIZE:
+            raise forms.ValidationError(
+                'Please keep filesize under %s. Current filesize: %s' % (
+                    filesizeformat(settings.MAX_UPLOAD_SIZE), filesizeformat(file.size)))
+        return file
+
+
+class MarkForm(forms.ModelForm):
+    class Meta:
+        model = Mark
+        fields = ['points']
+        widgets = {
+            'points': forms.TextInput(attrs={'type': 'number', 'min': 0, 'step': '0.5'})
+        }
+
+    def __init__(self, answer, *args, **kwargs):
+        super(MarkForm, self).__init__(*args, **kwargs)
+        self.answer = answer
+        self.fields['points'].widget.attrs['max'] = answer.assignment.max_points
+
+    def clean_points(self):
+        points = self.cleaned_data['points']
+        if points > self.answer.assignment.max_points:
+            raise ValidationError('Too many points for this assignment')
+        if points < 0:
+            raise ValidationError('Points cannot be negative')
+        return points