start atlas app

atlas/forms.py

@@ -0,0 +1,160 @@
+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",
+)