4 from django.db import models
5 from django.contrib.auth.models import User
6 from django.core.exceptions import ValidationError
7 from django.utils.translation import ugettext as _
8 from jsonfield import JSONField
10 from contact.models import Contact
15 class Submission(models.Model):
16 contact = models.ForeignKey(Contact, null = True)
17 key = models.CharField(max_length = 30, unique = True)
18 first_name = models.CharField(max_length = 100)
19 last_name = models.CharField(max_length = 100)
20 email = models.EmailField(max_length = 100, unique = True)
21 answers = models.CharField(max_length = 65536, null = True, blank = True)
22 key_sent = models.BooleanField(default = False)
23 def __unicode__(self):
24 return ', '.join((self.last_name, self.first_name, self.email))
27 def generate_key(cls):
29 while not key or key in [record['key'] for record in cls.objects.values('key')]:
30 key = ''.join(random.choice(string.ascii_uppercase + string.ascii_lowercase + string.digits) for i in range(30))
34 def create(cls, first_name, last_name, email, key = None, contact = None):
37 key = key if key else Submission.generate_key(),
38 first_name = first_name,
39 last_name = last_name,
47 class Attachment(models.Model):
48 submission = models.ForeignKey(Submission)
49 name = models.CharField(max_length=100)
50 file = models.FileField(upload_to = 'wtem/attachment')
53 class Assignment(models.Model):
54 user = models.ForeignKey(User, unique = True)
55 exercises = JSONField()
58 if not isinstance(self.exercises, list):
59 raise ValidationError(_('Assigned exercises must be declared in a list format'))
60 for exercise in self.exercises:
61 if not isinstance(exercise, int) or exercise < 1:
62 raise ValidationError(_('Invalid exercise id: %s' % exercise))
64 def __unicode__(self):
65 return self.user.username + ': ' + ','.join(map(str,self.exercises))