.

atlas/views.py

@@ -41,6 +41,7 @@ from .models import (
     Series,
     Examination,
     Finding,
+    Structure,
     Subspecialty,
     SeriesFinding,
     SeriesImage,
@@ -599,3 +600,37 @@ class ConditionAutocomplete(autocomplete.Select2QuerySetView):
                 return Condition.objects.none()
 
         return qs
+
+class FindingAutocomplete(autocomplete.Select2QuerySetView):
+    def get_queryset(self):
+        # Don't forget to filter out results depending on the visitor !
+        if not self.request.user.is_authenticated:
+            return Finding.objects.none()
+
+        qs = Finding.objects.all()
+
+        if self.q:
+            # This raises a fielderror which breaks creating a new item if not caught
+            try:
+                qs = qs.filter(name__contains=self.q)
+            except FieldError:
+                return Finding.objects.none()
+
+        return qs
+
+class StructureAutocomplete(autocomplete.Select2QuerySetView):
+    def get_queryset(self):
+        # Don't forget to filter out results depending on the visitor !
+        if not self.request.user.is_authenticated:
+            return Structure.objects.none()
+
+        qs = Structure.objects.all()
+
+        if self.q:
+            # This raises a fielderror which breaks creating a new item if not caught
+            try:
+                qs = qs.filter(name__contains=self.q)
+            except FieldError:
+                return Structure.objects.none()
+
+        return qs