root/questionnaire/models.py

from django.db import models

class RegUser(models.Model):
    email = models.EmailField(u"Emailaddress", unique = True)
    password = models.CharField(u"Password", max_length=200, blank = True, help_text = u'If you leave this empty, a password will be generated when you send an invitation to the user')

    class Meta:
        verbose_name = u'User'

    def check_complete(self):
        questions = Question.objects.all()
        for question in questions:
            answer = Answer.objects.filter(question = question, user = self)
            if len(answer) == 0:
                return False
        return True

    def __unicode__(self):
        return u'%s' % (self.email)

class Category(models.Model):
    nr = models.IntegerField(u"Category number", help_text = u"This is used for determining the order of the categories")
    name = models.CharField(u"Category name", max_length=200)
    text = models.TextField(u"Introduction text", help_text = u"This is displayed above the list of questions you add to this category")

    class Meta:
        verbose_name_plural = u'Categories'
        
    def __unicode__(self):
        return u'%s' % (self.name)

    def questions(self):
        return Question.objects.filter(category = self).order_by('nr')

class Question(models.Model):
    nr = models.IntegerField(u"Question number", help_text = "This is used for determining the order of the questions in a particular category")
    text = models.CharField(u"Question text", max_length=200)
    type = models.IntegerField(u"Type of question", choices = ( (1, u'Open'), (2, u'Multiple-choice'), (3, u'Multiple-choice, multiple-answers'), (4, u'Open, textfield') ), help_text = u"Determines the means of answering the question. An open question gives the user a single-line textfield, multiple-choice gives the user a number of choices he/she can choose from. If a question is multiple-choice, enter the choices this user can choose from below'")
    category = models.ForeignKey(Category, help_text = u"The category that this question is part of. The user sees each category in-order on a single page, with all questions belonging to the category on the page")
        
    def __unicode__(self):
        return u'%s %s %s' % (str(self.category), self.nr, self.text)

    def get_choices(self):
        choices = Choice.objects.filter(question = self).order_by('nr')
        if len(choices) == 0:
            return []
        return choices

class Choice(models.Model):
    question = models.ForeignKey(Question)
    nr = models.IntegerField(u"Choice number", help_text = u"Determines the order of the choices")
    name = models.CharField(u"Choice text", max_length=200)

    def __unicode__(self):
        return u'%s %s' % (self.nr, self.name)

class Answer(models.Model):
    user = models.ForeignKey(RegUser, help_text = u"The user who supplied this answer")
    question = models.ForeignKey(Question, help_text = u"The question that this an answer to")
    answer = models.TextField()

    def choice_str(self, secondary = False):
        choice_string = ""
        choices = self.question.get_choices()
        split_answers = self.answer.split()

        for choice in choices:
            for split_answer in split_answers:
                if str(split_answer) == str(choice.nr):
                    choice_string += str(choice.name) + " " 
        return choice_string

    def __unicode__(self):
        return u'%s %s %s' % (str(self.user), str(self.question), str(self.answer))