from django.forms import (
    Form,
    ModelForm,
    ModelMultipleChoiceField,
    ModelChoiceField,
    ChoiceField,
    CharField,
)
from django.forms import inlineformset_factory

from generic.models import (
    Examination, QuestionNote
)

from django.contrib.admin.widgets import FilteredSelectMultiple
from django.forms.widgets import RadioSelect, TextInput, Textarea

from rapids.models import Rapid
from anatomy.models import AnatomyQuestion
from longs.models import Long

from django.shortcuts import render, get_object_or_404, redirect

class ExaminationForm(ModelForm):
    class Meta:
        model = Examination
        fields = ["examination"]

class QuestionNoteForm(ModelForm):
    class Meta:
        model = QuestionNote
        fields = ["content_type", "object_id", "author", "note_type", "note"]

    def __init__(self, *args, **kwargs):
        question_type = kwargs.pop("question_type")
        pk = kwargs.pop("pk")

        if question_type == "rapid":
            question = Rapid
        elif question_type == "anatomy":
            question = AnatomyQuestion
        elif question_type == "long":
            question = AnatomyQuestion
        else:
            raise PermissionError()
        
        get_object_or_404(question, pk=pk)
        # We get the 'initial' keyword argument or initialize it
        # as a dict if it didn't exist.
        initial = kwargs.setdefault("initial", {})
        # The widget for a ModelMultipleChoiceField expects
        # a list of primary key for the selected data.
        initial["content_type"] = question
        initial["content_id"] = pk

        ModelForm.__init__(self, *args, **kwargs)
        super(QuestionNoteForm(), self).__init__(*args, **kwargs)