wtem/models.py

   1 # -*- coding: utf-8 -*-
   2 import random
   3 import string
   4 import os
   5 import json
   6
   7 from datetime import datetime
   8
   9 import pytz as pytz
  10 from django.conf import settings
  11 from django.core.validators import validate_email
  12 from django.db import models
  13 from django.contrib.auth.models import User
  14 from django.core.exceptions import ValidationError
  15 from django.core.mail import send_mail
  16 from django.core.urlresolvers import reverse
  17 from django.template.loader import render_to_string
  18 from django.utils import timezone
  19 from django.utils.translation import ugettext as _
  20 from jsonfield import JSONField
  21
  22 from contact.models import Contact
  23
  24 f = file(os.path.dirname(__file__) + '/fixtures/exercises.json')
  25 exercises = json.loads(f.read())
  26 f.close()
  27
  28 DEBUG_KEY = 'smerfetka159'
  29
  30 tz = pytz.timezone(settings.TIME_ZONE)
  31
  32
  33 def get_exercise_by_id(exercise_id):
  34     return [e for e in exercises if str(e['id']) == str(exercise_id)][0]
  35
  36
  37 def make_key(length):
  38     return ''.join(
  39         random.choice(string.ascii_uppercase + string.ascii_lowercase + string.digits)
  40         for i in range(length))
  41
  42
  43 def tuple2dt(time_tuple):
  44     return tz.localize(datetime(*time_tuple))
  45
  46
  47 class CompetitionState(models.Model):
  48     """singleton"""
  49     BEFORE = 'before'
  50     DURING = 'during'
  51     AFTER = 'after'
  52     STATE_CHOICES = (
  53         (BEFORE, u'przed rozpoczęciem'),
  54         (DURING, u'w trakcie'),
  55         (AFTER, u'po zakończeniu'),
  56     )
  57     state = models.CharField(choices=STATE_CHOICES, max_length=16)
  58
  59     start = tuple2dt(settings.OLIMPIADA_START)
  60     end = tuple2dt(settings.OLIMPIADA_END)
  61
  62     @classmethod
  63     def get_state(cls):
  64         now = timezone.now()
  65         if now < cls.start:
  66             return cls.BEFORE
  67         elif now < cls.end:
  68             return cls.DURING
  69         else:
  70             return cls.AFTER
  71         # return cls.objects.get().state
  72
  73
  74 class Submission(models.Model):
  75     contact = models.ForeignKey(Contact, null=True)
  76     key = models.CharField(max_length=30, unique=True)
  77     first_name = models.CharField(max_length=100)
  78     last_name = models.CharField(max_length=100)
  79     email = models.EmailField(max_length=100, unique=True)
  80     answers = models.CharField(max_length=65536, null=True, blank=True)
  81     key_sent = models.BooleanField(default=False)
  82     marks = JSONField(default={})
  83     examiners = models.ManyToManyField(User, null=True, blank=True)
  84     end_time = models.CharField(max_length=5, null=True, blank=True)
  85     random_seed = models.IntegerField()
  86
  87     def __unicode__(self):
  88         return ', '.join((self.last_name, self.first_name, self.email))
  89
  90     @classmethod
  91     def generate_key(cls):
  92         key = ''
  93         while not key or key in [record['key'] for record in cls.objects.values('key')]:
  94             key = make_key(30)
  95         return key
  96
  97     @classmethod
  98     def create(cls, first_name, last_name, email, key=None, contact=None):
  99         submission = cls(
 100             contact=contact,
 101             key=key if key else Submission.generate_key(),
 102             first_name=first_name,
 103             last_name=last_name,
 104             email=email,
 105             random_seed=random.randint(-2147483648, 2147483647)
 106         )
 107
 108         submission.save()
 109         return submission
 110
 111     def competition_link(self):
 112         return reverse('form_single', kwargs={'submission_id': self.id, 'key': self.key})
 113
 114     def get_answers(self):
 115         return json.loads(self.answers) if self.answers else {}
 116
 117     def shuffled_exercise_ids(self):
 118         exercise_ids = [e['id'] for e in exercises]
 119         seeded_random = random.Random(self.random_seed)
 120         seeded_random.shuffle(exercise_ids)
 121         return exercise_ids
 122
 123     def current_exercise(self):
 124         answers = self.get_answers()
 125         for i, id in enumerate(self.shuffled_exercise_ids(), 1):
 126             if str(id) not in answers:
 127                 return i, get_exercise_by_id(id)
 128         return None, None
 129
 130     def get_mark(self, user_id, exercise_id):
 131         mark = None
 132         user_id = str(user_id)
 133         exercise_id = str(exercise_id)
 134         if self.marks and user_id in self.marks:
 135             mark = self.marks[user_id].get(exercise_id, None)
 136         return mark
 137
 138     def set_mark(self, user_id, exercise_id, mark):
 139         user_id = str(user_id)
 140         exercise_id = str(exercise_id)
 141         if not self.marks:
 142             self.marks = dict()
 143
 144         self.marks.setdefault(user_id, {})[exercise_id] = mark
 145         if mark == 'None':
 146             del self.marks[user_id][exercise_id]
 147
 148     def get_exercise_marks_by_examiner(self, exercise_id):
 149         marks = dict()
 150         for examiner_id, examiner_marks in self.marks.items():
 151             mark = examiner_marks.get(exercise_id, None)
 152             if mark is not None:
 153                 marks[examiner_id] = mark
 154         return marks
 155
 156     def get_final_exercise_mark(self, exercise_id):
 157         exercise = get_exercise_by_id(exercise_id)
 158         if exercise_checked_manually(exercise):
 159             marks_by_examiner = self.get_exercise_marks_by_examiner(exercise_id)
 160             if len(marks_by_examiner):
 161                 return sum(map(float, marks_by_examiner.values())) / float(len(marks_by_examiner))
 162             else:
 163                 return None
 164         else:
 165             if not self.answers:
 166                 return None
 167             answer = json.loads(self.answers)[exercise_id]['closed_part']
 168             t = exercise['type']
 169             if t == 'edumed_uporzadkuj':
 170                 return exercise['points'] if map(int, answer) == exercise['answer'] else 0
 171             if t == 'edumed_przyporzadkuj':
 172                 toret = 0
 173                 for bucket_id, items in answer.items():
 174                     for item_id in items:
 175                         is_corect = False
 176                         if exercise.get('answer_mode', None) == 'possible_buckets_for_item':
 177                             is_correct = int(bucket_id) in exercise['answer'].get(item_id)
 178                         else:
 179                             is_correct = int(item_id) in exercise['answer'].get(bucket_id, [])
 180                         if is_correct:
 181                             toret += exercise['points_per_hit']
 182                 return toret
 183             if t == 'edumed_wybor':
 184                 if len(exercise['answer']) == 1:
 185                     if len(answer) and int(answer[0]) == exercise['answer'][0]:
 186                         return exercise['points']
 187                     else:
 188                         return 0
 189                 else:
 190                     toret = 0
 191                     if exercise.get('answer_mode', None) == 'all_or_nothing':
 192                         toret = exercise['points'] if map(int, answer) == exercise['answer'] else 0
 193                     else:
 194                         for answer_id in map(int, answer):
 195                             if answer_id in exercise['answer']:
 196                                 toret += exercise['points_per_hit']
 197                     return toret
 198             if t == 'edumed_prawdafalsz':
 199                 toret = 0
 200                 for idx, statement in enumerate(exercise['statements']):
 201                     if statement[1] == 'ignore':
 202                         continue
 203                     if answer[idx] == 'true':
 204                         given = True
 205                     elif answer[idx] == 'false':
 206                         given = False
 207                     else:
 208                         given = None
 209                     if given == statement[1]:
 210                         toret += exercise['points_per_hit']
 211                 return toret
 212             raise NotImplementedError
 213
 214     @property
 215     def final_result(self):
 216         final = 0
 217         # for exercise_id in map(str,range(1, len(exercises) + 1)):
 218         for exercise_id in [str(x['id']) for x in exercises]:
 219             mark = self.get_final_exercise_mark(exercise_id)
 220             if mark is not None:
 221                 final += mark
 222         return final
 223
 224     @property
 225     def final_result_as_string(self):
 226         return ('%.2f' % self.final_result).rstrip('0').rstrip('.')
 227
 228
 229 class Attachment(models.Model):
 230     submission = models.ForeignKey(Submission)
 231     exercise_id = models.IntegerField()
 232     tag = models.CharField(max_length=128, null=True, blank=True)
 233     file = models.FileField(upload_to='wtem/attachment')
 234
 235
 236 class Assignment(models.Model):
 237     user = models.ForeignKey(User, unique=True)
 238     exercises = JSONField()
 239
 240     def clean(self):
 241         if not isinstance(self.exercises, list):
 242             raise ValidationError(_('Assigned exercises must be declared in a list format'))
 243         # for exercise in self.exercises:
 244         #     if not isinstance(exercise, int) or exercise < 1:
 245         #         raise ValidationError(_('Invalid exercise id: %s' % exercise))
 246
 247     def __unicode__(self):
 248         return self.user.username + ': ' + ','.join(map(str, self.exercises))
 249
 250
 251 class Confirmation(models.Model):
 252     first_name = models.CharField(max_length=100)
 253     last_name = models.CharField(max_length=100)
 254     email = models.EmailField(max_length=100, unique=True)
 255     contact = models.ForeignKey(Contact, null=True)
 256     key = models.CharField(max_length=30)
 257     confirmed = models.BooleanField(default=False)
 258
 259     class Meta:
 260         ordering = ['contact__contact']
 261
 262     @classmethod
 263     def create(cls, first_name, last_name, email, contact=None, key=None):
 264         confirmation = cls(
 265             contact=contact,
 266             key=key if key else make_key(30),
 267             first_name=first_name,
 268             last_name=last_name,
 269             email=email
 270         )
 271
 272         confirmation.save()
 273         return confirmation
 274
 275     def absolute_url(self):
 276         return reverse('student_confirmation', args=(self.id, self.key))
 277
 278     def readable_contact(self):
 279         return '%s <%s>' % (self.contact.body.get('przewodniczacy'), self.contact.contact)
 280
 281     def school_phone(self):
 282         return '%s, tel. %s' % (self.contact.body.get('school'), self.contact.body.get('school_phone'))
 283
 284     def age(self):
 285         return timezone.now() - self.contact.created_at
 286
 287     def readable_age(self):
 288         td = self.age()
 289         return '%s dni, %s godzin' % (td.days, td.seconds/3600)
 290
 291     def send_mail(self):
 292         mail_subject = render_to_string('contact/olimpiada/student_mail_subject.html').strip()
 293         mail_body = render_to_string(
 294             'contact/olimpiada/student_mail_body.html', {'confirmation': self})
 295         try:
 296             validate_email(self.email)
 297         except ValidationError:
 298             pass
 299         else:
 300             send_mail(mail_subject, mail_body, 'olimpiada@nowoczesnapolska.org.pl', [self.email],
 301                       fail_silently=True)
 302
 303
 304 def exercise_checked_manually(exercise):
 305     return (exercise['type'] in ('open', 'file_upload')) or 'open_part' in exercise