apps/quiz/models.py

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