4c3d0ee81a9eecaf97f8947d6e8c379b56a54664
[edumed.git] / wtem / models.py
1 import random
2 import string
3 import os
4
5 from django.db import models
6 from django.contrib.auth.models import User
7 from django.core.exceptions import ValidationError
8 from django.utils import simplejson
9 from django.utils.translation import ugettext as _
10 from jsonfield import JSONField
11
12 from contact.models import Contact
13
14 f = file(os.path.dirname(__file__) + '/fixtures/exercises.json')
15 exercises = simplejson.loads(f.read())
16 f.close()
17
18 DEBUG_KEY = '12345'
19
20 class Submission(models.Model):
21     contact = models.ForeignKey(Contact, null = True)
22     key = models.CharField(max_length = 30, unique = True)
23     first_name = models.CharField(max_length = 100)
24     last_name = models.CharField(max_length = 100)
25     email = models.EmailField(max_length = 100, unique = True)
26     answers = models.CharField(max_length = 65536, null = True, blank = True)
27     key_sent = models.BooleanField(default = False)
28     marks = JSONField()
29     examiners = models.ManyToManyField(User, null = True, blank = True)
30     end_time = models.CharField(max_length = 5, null = True, blank = True)
31
32     def __unicode__(self):
33         return ', '.join((self.last_name, self.first_name, self.email))
34
35     @classmethod
36     def generate_key(cls):
37         key = ''
38         while not key or key in [record['key'] for record in cls.objects.values('key')]:
39             key = ''.join(random.choice(string.ascii_uppercase + string.ascii_lowercase + string.digits) for i in range(30))
40         return key
41
42     @classmethod
43     def create(cls, first_name, last_name, email, key = None, contact = None):
44         submission = cls(
45             contact = contact,
46             key = key if key else Submission.generate_key(),
47             first_name = first_name,
48             last_name = last_name,
49             email = email
50         )
51
52         submission.save()
53         return submission
54
55     def get_mark(self, user_id, exercise_id):
56         mark = None
57         user_id = str(user_id)
58         exercise_id = str(exercise_id)
59         if self.marks and user_id in self.marks:
60             mark = self.marks[user_id].get(exercise_id, None)
61         return mark
62
63     def set_mark(self, user_id, exercise_id, mark):
64         user_id = str(user_id)
65         exercise_id = str(exercise_id)
66         if not self.marks:
67             self.marks = dict()
68         
69         self.marks.setdefault(user_id, {})[exercise_id] = mark
70         if mark == 'None':
71             del self.marks[user_id][exercise_id]
72
73
74     def get_exercise_marks_by_examiner(self, exercise_id):
75         marks = dict()
76         for examiner_id, examiner_marks in self.marks.items():
77             mark = examiner_marks.get(exercise_id, None)
78             if mark is not None:
79                 marks[examiner_id] = mark
80         return marks
81
82     def get_final_exercise_mark(self, exercise_id):
83         exercise = exercises[int(exercise_id)-1]
84         if exercise_checked_manually(exercise):
85             marks_by_examiner = self.get_exercise_marks_by_examiner(exercise_id)
86             if len(marks_by_examiner):
87                 return sum(map(int, marks_by_examiner.values())) / float(len(marks_by_examiner))
88             else:
89                 return None
90         else:
91             if not self.answers:
92                 return None
93             answer = simplejson.loads(self.answers)[exercise_id]['closed_part']
94             t = exercise['type']
95             if t == 'edumed_uporzadkuj':
96                 return exercise['points'] if map(int, answer) == exercise['answer'] else 0
97             if t == 'edumed_przyporzadkuj':
98                 toret = 0
99                 for bucket_id, items in answer.items():
100                     for item_id in items:
101                         is_corect = False
102                         if exercise.get('answer_mode', None) == 'possible_buckets_for_item':
103                             is_correct = int(bucket_id) in exercise['answer'].get(item_id)
104                         else:
105                             is_correct = int(item_id) in exercise['answer'].get(bucket_id, [])
106                         if is_correct:
107                             toret += exercise['points_per_hit']
108                 return toret
109             if t == 'edumed_wybor':
110                 if len(exercise['answer']) == 1:
111                     if len(answer) and int(answer[0]) == exercise['answer'][0]:
112                         return exercise['points']
113                     else:
114                         return 0
115                 else:
116                     toret = 0
117                     if exercise.get('answer_mode', None) == 'all_or_nothing':
118                         toret = exercise['points'] if map(int, answer) == exercise['answer'] else 0
119                     else:
120                         for id in map(int, answer):
121                             if id in exercise['answer']:
122                                 toret += exercise['points_per_hit']
123                     return toret
124             if t == 'edumed_prawdafalsz':
125                 toret = 0
126                 for idx, statement in enumerate(exercise['statements']):
127                     if answer[idx] == 'true':
128                         given = True
129                     elif answer[idx] == 'false':
130                         given = False
131                     else:
132                         given = None
133                     if given == statement[1]:
134                         toret += exercise['points_per_hit']
135                 return toret
136             raise NotImplementedError
137
138     @property
139     def final_result(self):
140         final = 0
141         for exercise_id in map(str,range(1, len(exercises) + 1)):
142             mark = self.get_final_exercise_mark(exercise_id)
143             if mark is not None:
144                 final += mark
145         return final
146
147     @property
148     def final_result_as_string(self):
149         return ('%.2f' % self.final_result).rstrip('0').rstrip('.')
150
151 class Attachment(models.Model):
152     submission = models.ForeignKey(Submission)
153     exercise_id = models.IntegerField()
154     file = models.FileField(upload_to = 'wtem/attachment')
155
156
157 class Assignment(models.Model):
158     user = models.ForeignKey(User, unique = True)
159     exercises = JSONField()
160
161     def clean(self):
162         if not isinstance(self.exercises, list):
163             raise ValidationError(_('Assigned exercises must be declared in a list format'))
164         for exercise in self.exercises:
165             if not isinstance(exercise, int) or exercise < 1:
166                 raise ValidationError(_('Invalid exercise id: %s' % exercise))
167
168     def __unicode__(self):
169         return self.user.username + ': ' + ','.join(map(str,self.exercises))
170
171
172 def exercise_checked_manually(exercise):
173     return (exercise['type'] in ('open', 'file_upload')) or 'open_part' in exercise