Le-Wagon Workshop

admin.py

from django.contrib import admin
from mysite.models import Survey, Question, Choice

class QuestionInline(admin.TabularInline):
  model = Question
  show_change_link = True

class ChoiceInline(admin.TabularInline):
  model = Choice


class SurveyAdmin(admin.ModelAdmin):
  inlines = [
    QuestionInline
  ]

class QuestionAdmin(admin.ModelAdmin):
  inlines = [
    ChoiceInline
  ]

admin.site.register(Survey, SurveyAdmin)
admin.site.register(Question, QuestionAdmin)
admin.site.register(Choice)

forms.py

from django import forms

class SurveyForm(forms.Form):
    question_1 = forms.ChoiceField(widget=forms.RadioSelect, choices=())

    def __init__(self, survey, *args, **kwargs):
        super().__init__(*args, **kwargs)
        for question in survey.question_set.all():
          answers = [(answer.text, answer.text) for answer in question.choice_set.all()]
          self.fields[f"question_{question.id}"] = forms.ChoiceField(widget=forms.RadioSelect, choices=answers)
          self.fields[f"question_{question.id}"].label = question.text

initial-models.py

from django.db import models


class Question(models.Model):
    question_text = models.CharField(max_length=200)
    pub_date = models.DateTimeField('date published')

    def __str__(self):
      return self.question_text


class Choice(models.Model):
    question = models.ForeignKey(Question, on_delete=models.CASCADE)
    choice_text = models.CharField(max_length=200)
    votes = models.IntegerField(default=0)

    def __str__(self):
      return self.choice_text

shell.py

from mysite.models import Survey, Question, Choice, Submission
survey1 = Survey.objects.create(title="Survey 1")
q1 = Question.objects.create(survey=survey1, text="What is capital of Japan?", pub_date=timezone.now())
q1.choice_set.add(Choice.objects.create(text="Tokyo"))

sub = Submission(survey=s1, participant_email="kamal@lalokalabs.co", participant_phonenumber="+60187884325")

test_forms.py

from mysite.forms import SurveyForm
from mysite.models import Survey

survey = Survey.objects.all()[0]
data = {
    "email": "kamal@lalokalabs.co",
    "phonenumber": "+81919191919",
    "question_1": "1",
    "question_2": "4",
}
form = SurveyForm(survey, data)
if form.is_valid():
    print(form.cleaned_data)
else:
    print(form.errors)

print(form.fields["question_2"].choices)