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