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

from atlas.models import (
    Case,
    Series,
    SeriesImage,
)

from anatomy.models import Modality

from generic.models import Examination, Condition, Sign
#from generic.models import Examination, Site, Condition, Sign

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

from tinymce.widgets import TinyMCE

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

class SeriesForm(ModelForm):
    class Media:
        # Django also includes a few javascript files necessary
        # for the operation of this form element. You need to
        # include <script src="/admin/jsi18n"></script>
        # in the template.
        css = {
            "all": ["css/widgets.css"],
        }
        # Adding this javascript is crucial
        js = ["jsi18n.js", "tesseract.min.js"]

    def __init__(self, *args, **kwargs):
        super(SeriesForm, self).__init__(*args, **kwargs)
        #self.fields["examination"] = ModelChoiceField(
        #    queryset=Examination.objects.all(),
        #    widget=FilteredSelectMultiple(verbose_name="Examination", is_stacked=False),
        #)

        self.fields["modality"] = ModelChoiceField(
            required=True,
            queryset=Modality.objects.all(),
        )

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

    class Meta:
        model = Series
        exclude = ["author"]


class AtlasForm(ModelForm):

    #exams = ModelMultipleChoiceField(required=False, queryset=Exam.objects.all(),widget=FilteredSelectMultiple(verbose_name="Exams", is_stacked=False))

    class Media:
        # Django also includes a few javascript files necessary
        # for the operation of this form element. You need to
        # include <script src="/admin/jsi18n"></script>
        # in the template.
        css = {
            "all": ["css/widgets.css"],
        }
        # Adding this javascript is crucial
        js = ["jsi18n.js", "tesseract.min.js"]

    def __init__(self, *args, **kwargs):
        self.user = kwargs.pop('user')  # To get request.user. Do not use kwargs.pop('user', None) due to potential security hole
        if kwargs.get("instance"):
            # 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["exams"] = [t.pk for t in kwargs["instance"].exams.all()]

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

        super(AtlasForm, self).__init__(*args, **kwargs)


    class Meta:
        model = Case
        # fields = ['due_back']
        # fields = '__all__'
        fields = [
            #"examination",
            #"site",
            "description",
            "history",
            "findings",
            "condition",
            "sign",
            "open_access",
        ]
        # fields = ['question', 'findings', 'subspecialty', 'references']
        widgets = {
            # "normal": RadioSelect(
            # choices=[(True, 'Yes'),
            #     (False, 'No')])
            "findings" : TinyMCE(attrs={'cols': 80, 'rows': 20}),
            "mark_scheme" : TinyMCE(attrs={'cols': 80, 'rows': 30}),
            "description" : Textarea(attrs={'cols': 80, 'rows': 5}),
            "history" : Textarea(attrs={'cols': 80, 'rows': 5}),
        }

    def save(self, commit=True):
        # Get the unsaved Case instance
        instance = ModelForm.save(self, False)

        # Prepare a 'save_m2m' method for the form,
        old_save_m2m = self.save_m2m

        def save_m2m():
            old_save_m2m()
            #instance.exams.clear()
            #for exam in self.cleaned_data["exams"]:
            #    # We need the id here
            #    instance.exams.add(exam.pk)

        self.save_m2m = save_m2m

        # Do we need to save all changes now?
        # if commit:
        instance.save()
        self.save_m2m()

        return instance

SeriesFormSet = inlineformset_factory(
    Case,
    Series.case.through,
    exclude=[],
    can_delete=True,
    extra=0,
    max_num=10,
    field_classes="testing",
)


SeriesImageFormSet = inlineformset_factory(
    Series,
    SeriesImage,
    exclude=[],
    can_delete=True,
    extra=0,
    max_num=2000,
    field_classes="testing",
)