apps/quiz/models.py

   1 from django.db import models
   2 from django.contrib.sites.models import Site
   3 from django.utils.translation import ugettext_lazy as _
   4
   5 from django.conf import settings
   6
   7
   8 class Quiz(Site):
   9     description = models.TextField()
  10
  11     class Meta:
  12         verbose_name = _('quiz')
  13         verbose_name_plural = _('quizzes')
  14
  15     @classmethod
  16     def current(cls):
  17         return cls.objects.get(id=settings.SITE_ID)
  18
  19     def start(self):
  20         return self.question_set.all()[0]
  21
  22     @models.permalink
  23     def get_absolute_url(self):
  24         return ('quiz', )
  25
  26     def where_to(self):
  27         try:
  28             return Result.objects.get(quiz=self).get_absolute_url()
  29         except Result.DoesNotExist:
  30             # just go to the beginning
  31             return self.get_absolute_url()
  32
  33
  34 class Result(models.Model):
  35     quiz = models.ForeignKey(Quiz)
  36     slug = models.SlugField(db_index=True)
  37     title = models.CharField(max_length=255)
  38     text = models.TextField()
  39
  40     class Meta:
  41         verbose_name = _('result')
  42         verbose_name_plural = _('results')
  43         ordering = ['quiz', 'title']
  44
  45     def __unicode__(self):
  46         return self.title
  47
  48     @models.permalink
  49     def get_absolute_url(self):
  50         return ('quiz_result', [self.slug])
  51
  52
  53 class Question(models.Model):
  54     quiz = models.ForeignKey(Quiz)
  55     slug = models.SlugField(db_index=True)
  56     ordering = models.SmallIntegerField()
  57     title = models.CharField(max_length=255)
  58     text = models.TextField(null=True, blank=True)
  59     description = models.TextField(null=True, blank=True)
  60
  61     class Meta:
  62         verbose_name = _('question')
  63         verbose_name_plural = _('questions')
  64         ordering = ['quiz', 'ordering']
  65         unique_together = [['quiz', 'slug'], ['quiz', 'ordering']]
  66
  67     def __unicode__(self):
  68         return self.title
  69
  70     @models.permalink
  71     def get_absolute_url(self):
  72         return ('quiz', [self.slug])
  73
  74     def where_to(self):
  75         later = self.quiz.question_set.filter(ordering__gt=self.ordering)
  76         if later.exists():
  77             return later[0].get_absolute_url()
  78         else:
  79             return self.quiz.where_to()
  80
  81
  82
  83 class Answer(models.Model):
  84     title = models.CharField(max_length=255)
  85     question = models.ForeignKey(Question)
  86     go_to = models.ForeignKey(Question, null=True, blank=True,
  87             related_name='go_tos')
  88     result = models.ForeignKey(Result, null=True, blank=True)
  89     ordering = models.SmallIntegerField()
  90
  91     class Meta:
  92         verbose_name = _('answer')
  93         verbose_name_plural = _('answers')
  94         ordering = ['question', 'ordering']
  95
  96     def __unicode__(self):
  97         return self.title
  98
  99     def where_to(self):
 100         # follow explicit redirects
 101         if self.result:
 102             return self.result.get_absolute_url()
 103         elif self.go_to:
 104             return self.go_to.get_absolute_url()
 105
 106         # or just get the next question
 107         return self.question.where_to()