X-Git-Url: https://git.mdrn.pl/edumed.git/blobdiff_plain/cbae43d3842076cf85fcc206e2ed13fce7d0c0a4..83272404e6de588c8b436fed6e5ea41eb7df8e6a:/wtem/models.py

diff --git a/wtem/models.py b/wtem/models.py
index 2b093ae..e3b0656 100644
--- a/wtem/models.py
+++ b/wtem/models.py
@@ -19,11 +19,24 @@ from django.utils import timezone
 from django.utils.translation import ugettext as _
 from jsonfield import JSONField
 
-from contact.models import Contact
 
-f = file(os.path.dirname(__file__) + '/fixtures/exercises.json')
-exercises = json.loads(f.read())
-f.close()
+def prepare_exercises():
+    f = file(os.path.dirname(__file__) + '/fixtures/exercises.json')
+    exercises = json.loads(f.read())
+    for i, exercise in enumerate(exercises, 1):
+        exercise['id'] = i
+        if exercise['type'] == 'edumed_wybor' and exercise['answer_mode'] == 'multi':
+            answer = []
+            for j, option in enumerate(exercise['options'], 1):
+                option['id'] = j
+                if option['answer']:
+                    answer.append(j)
+            exercise['answer'] = answer
+    f.close()
+    return exercises
+
+
+exercises = prepare_exercises()
 
 DEBUG_KEY = 'smerfetka159'
 
@@ -72,7 +85,7 @@ class CompetitionState(models.Model):
 
 
 class Submission(models.Model):
-    contact = models.ForeignKey(Contact, null=True)
+    contact = models.ForeignKey('contact.Contact', null=True)
     key = models.CharField(max_length=30, unique=True)
     first_name = models.CharField(max_length=100)
     last_name = models.CharField(max_length=100)
@@ -156,6 +169,8 @@ class Submission(models.Model):
 
     def get_final_exercise_mark(self, exercise_id):
         exercise = get_exercise_by_id(exercise_id)
+        if exercise.get('excluded'):
+            return 0
         if exercise_checked_manually(exercise):
             marks_by_examiner = self.get_exercise_marks_by_examiner(exercise_id)
             if len(marks_by_examiner):
@@ -265,13 +280,32 @@ class Assignment(models.Model):
         return self.user.username + ': ' + ','.join(map(str, self.exercises))
 
 
-class Confirmation(models.Model):
+class AbstractConfirmation(models.Model):
+    contact = models.ForeignKey('contact.Contact', null=True)
+    key = models.CharField(max_length=30)
+    confirmed = models.BooleanField(default=False)
+
+    class Meta:
+        abstract = True
+
+    def readable_contact(self):
+        return '%s <%s>' % (self.contact.body.get('przewodniczacy'), self.contact.contact)
+
+    def school_phone(self):
+        return '%s, tel. %s' % (self.contact.body.get('school'), self.contact.body.get('school_phone'))
+
+    def age(self):
+        return timezone.now() - self.contact.created_at
+
+    def readable_age(self):
+        td = self.age()
+        return '%s dni, %s godzin' % (td.days, td.seconds/3600)
+
+
+class Confirmation(AbstractConfirmation):
     first_name = models.CharField(max_length=100)
     last_name = models.CharField(max_length=100)
     email = models.EmailField(max_length=100, unique=True)
-    contact = models.ForeignKey(Contact, null=True)
-    key = models.CharField(max_length=30)
-    confirmed = models.BooleanField(default=False)
 
     class Meta:
         ordering = ['contact__contact']
@@ -292,19 +326,6 @@ class Confirmation(models.Model):
     def absolute_url(self):
         return reverse('student_confirmation', args=(self.id, self.key))
 
-    def readable_contact(self):
-        return '%s <%s>' % (self.contact.body.get('przewodniczacy'), self.contact.contact)
-
-    def school_phone(self):
-        return '%s, tel. %s' % (self.contact.body.get('school'), self.contact.body.get('school_phone'))
-
-    def age(self):
-        return timezone.now() - self.contact.created_at
-
-    def readable_age(self):
-        td = self.age()
-        return '%s dni, %s godzin' % (td.days, td.seconds/3600)
-
     def send_mail(self):
         mail_subject = render_to_string('contact/olimpiada/student_mail_subject.html').strip()
         mail_body = render_to_string(
@@ -318,5 +339,23 @@ class Confirmation(models.Model):
                       fail_silently=True)
 
 
+class TeacherConfirmation(AbstractConfirmation):
+
+    class Meta:
+        ordering = ['contact__contact']
+
+    @classmethod
+    def create(cls, contact=None, key=None):
+        confirmation = cls(
+            contact=contact,
+            key=key if key else make_key(30),
+        )
+        confirmation.save()
+        return confirmation
+
+    def absolute_url(self):
+        return reverse('teacher_confirmation', args=(self.id, self.key))
+
+
 def exercise_checked_manually(exercise):
     return (exercise['type'] in ('open', 'file_upload')) or 'open_part' in exercise