.

atlas/filters.py

@@ -1,210 +1,210 @@
-import django_filters
-
-from .models import (
-    Case,
-    Condition,
-    Finding,
-    PathologicalProcess,
-    Presentation,
-    Series,
-    Structure,
-    Subspecialty,
-)
-from django.contrib.auth.models import User
-
-
-def get_authors(request):
-    if request is None:
-        return User.objects.none()
-
-    if not request.user.groups.filter(name="atlas_editor").exists():
-        queryset = User.objects.filter(id=request.user.id)
-    else:
-        queryset = User.objects.all()
-
-    return queryset
-
-
-class CaseFilter(django_filters.FilterSet):
-    author = django_filters.ModelMultipleChoiceFilter(
-        queryset=get_authors, null_label="No author"
-    )
-    differential_case = django_filters.ModelMultipleChoiceFilter(
-        label="Differential",
-        field_name="differentialcase__condition",
-        # to_field_name="differential",
-        queryset=Condition.objects.all(),
-    )
-
-    class Meta:
-        model = Case
-        fields = (
-            "created_date",
-            "open_access",
-            "author",
-            "differential_case",
-        )  # "site",
-
-    def __init__(
-        self,
-        data=None,
-        queryset=None,
-        prefix=None,
-        strict=None,
-        user=None,
-        request=None,
-    ):
-        if not request.user.groups.filter(name="atlas_editor").exists():
-            # queryset = queryset.filter(open_access=True) | queryset.filter(author__id=request.user.id)
-            queryset = queryset.filter(author__id=request.user.id)
-        super(CaseFilter, self).__init__(
-            data=data, queryset=queryset, prefix=prefix, request=request
-        )
-        pass
-
-
-class SeriesFilter(django_filters.FilterSet):
-    case = django_filters.ModelMultipleChoiceFilter(
-        queryset=Case.objects.all(), null_label="No case"
-    )
-
-    class Meta:
-        model = Series
-        fields = ("modality", "case", "examination", "author")
-
-    def __init__(
-        self,
-        data=None,
-        queryset=None,
-        prefix=None,
-        strict=None,
-        user=None,
-        request=None,
-    ):
-        if not request.user.groups.filter(name="atlas_editor").exists():
-            # queryset = queryset.filter(open_access=True) | queryset.filter(author__id=request.user.id)
-            queryset = queryset.filter(author__id=request.user.id)
-        super(SeriesFilter, self).__init__(
-            data=data, queryset=queryset, prefix=prefix, request=request
-        )
-        pass
-
-
-class ConditionFilter(django_filters.FilterSet):
-    class Meta:
-        model = Condition
-        fields = ("name", "primary", "synonym", "parent", "subspecialty")
-
-    def __init__(
-        self,
-        data=None,
-        queryset=None,
-        prefix=None,
-        strict=None,
-        user=None,
-        request=None,
-    ):
-        super(ConditionFilter, self).__init__(
-            data=data, queryset=queryset, prefix=prefix, request=request
-        )
-        pass
-
-
-class FindingFilter(django_filters.FilterSet):
-    class Meta:
-        model = Finding
-        fields = ("name", "primary", "synonym")
-
-    def __init__(
-        self,
-        data=None,
-        queryset=None,
-        prefix=None,
-        strict=None,
-        user=None,
-        request=None,
-    ):
-        super(FindingFilter, self).__init__(
-            data=data, queryset=queryset, prefix=prefix, request=request
-        )
-        pass
-
-
-class StructureFilter(django_filters.FilterSet):
-    class Meta:
-        model = Structure
-        fields = {"name": ["icontains"], "primary": ["exact"], "synonym": ["exact"]}
-
-    def __init__(
-        self,
-        data=None,
-        queryset=None,
-        prefix=None,
-        strict=None,
-        user=None,
-        request=None,
-    ):
-        super(StructureFilter, self).__init__(
-            data=data, queryset=queryset, prefix=prefix, request=request
-        )
-        pass
-
-
-class PresentationFilter(django_filters.FilterSet):
-    class Meta:
-        model = Presentation
-        fields = ("name", "subspecialty")
-
-    def __init__(
-        self,
-        data=None,
-        queryset=None,
-        prefix=None,
-        strict=None,
-        user=None,
-        request=None,
-    ):
-        super(PresentationFilter, self).__init__(
-            data=data, queryset=queryset, prefix=prefix, request=request
-        )
-        pass
-
-
-class PathologicalProcessFilter(django_filters.FilterSet):
-    class Meta:
-        model = PathologicalProcess
-        fields = ("name",)
-
-    def __init__(
-        self,
-        data=None,
-        queryset=None,
-        prefix=None,
-        strict=None,
-        user=None,
-        request=None,
-    ):
-        super(PathologicalProcessFilter, self).__init__(
-            data=data, queryset=queryset, prefix=prefix, request=request
-        )
-        pass
-
-
-class SubspecialtyFilter(django_filters.FilterSet):
-    class Meta:
-        model = Subspecialty
-        fields = ("name",)
-
-    def __init__(
-        self,
-        data=None,
-        queryset=None,
-        prefix=None,
-        strict=None,
-        user=None,
-        request=None,
-    ):
-        super(SubspecialtyFilter, self).__init__(
-            data=data, queryset=queryset, prefix=prefix, request=request
-        )
-        pass
+import django_filters
+
+from .models import (
+    Case,
+    Condition,
+    Finding,
+    PathologicalProcess,
+    Presentation,
+    Series,
+    Structure,
+    Subspecialty,
+)
+from django.contrib.auth.models import User
+
+
+def get_authors(request):
+    if request is None:
+        return User.objects.none()
+
+    if not request.user.groups.filter(name="atlas_editor").exists():
+        queryset = User.objects.filter(id=request.user.id)
+    else:
+        queryset = User.objects.all()
+
+    return queryset
+
+
+class CaseFilter(django_filters.FilterSet):
+    author = django_filters.ModelMultipleChoiceFilter(
+        queryset=get_authors, null_label="No author"
+    )
+    differential_case = django_filters.ModelMultipleChoiceFilter(
+        label="Differential",
+        field_name="differentialcase__condition",
+        # to_field_name="differential",
+        queryset=Condition.objects.all(),
+    )
+
+    class Meta:
+        model = Case
+        fields = (
+            "created_date",
+            "open_access",
+            "author",
+            "differential_case",
+        )  # "site",
+
+    def __init__(
+        self,
+        data=None,
+        queryset=None,
+        prefix=None,
+        strict=None,
+        user=None,
+        request=None,
+    ):
+        if not request.user.groups.filter(name="atlas_editor").exists():
+            # queryset = queryset.filter(open_access=True) | queryset.filter(author__id=request.user.id)
+            queryset = queryset.prefetch_related("author").filter(author__id=request.user.id)
+        super(CaseFilter, self).__init__(
+            data=data, queryset=queryset, prefix=prefix, request=request
+        )
+        pass
+
+
+class SeriesFilter(django_filters.FilterSet):
+    case = django_filters.ModelMultipleChoiceFilter(
+        queryset=Case.objects.all(), null_label="No case"
+    )
+
+    class Meta:
+        model = Series
+        fields = ("modality", "case", "examination", "author")
+
+    def __init__(
+        self,
+        data=None,
+        queryset=None,
+        prefix=None,
+        strict=None,
+        user=None,
+        request=None,
+    ):
+        if not request.user.groups.filter(name="atlas_editor").exists():
+            # queryset = queryset.filter(open_access=True) | queryset.filter(author__id=request.user.id)
+            queryset = queryset.filter(author__id=request.user.id)
+        super(SeriesFilter, self).__init__(
+            data=data, queryset=queryset, prefix=prefix, request=request
+        )
+        pass
+
+
+class ConditionFilter(django_filters.FilterSet):
+    class Meta:
+        model = Condition
+        fields = ("name", "primary", "synonym", "parent", "subspecialty")
+
+    def __init__(
+        self,
+        data=None,
+        queryset=None,
+        prefix=None,
+        strict=None,
+        user=None,
+        request=None,
+    ):
+        super(ConditionFilter, self).__init__(
+            data=data, queryset=queryset, prefix=prefix, request=request
+        )
+        pass
+
+
+class FindingFilter(django_filters.FilterSet):
+    class Meta:
+        model = Finding
+        fields = ("name", "primary", "synonym")
+
+    def __init__(
+        self,
+        data=None,
+        queryset=None,
+        prefix=None,
+        strict=None,
+        user=None,
+        request=None,
+    ):
+        super(FindingFilter, self).__init__(
+            data=data, queryset=queryset, prefix=prefix, request=request
+        )
+        pass
+
+
+class StructureFilter(django_filters.FilterSet):
+    class Meta:
+        model = Structure
+        fields = {"name": ["icontains"], "primary": ["exact"], "synonym": ["exact"]}
+
+    def __init__(
+        self,
+        data=None,
+        queryset=None,
+        prefix=None,
+        strict=None,
+        user=None,
+        request=None,
+    ):
+        super(StructureFilter, self).__init__(
+            data=data, queryset=queryset, prefix=prefix, request=request
+        )
+        pass
+
+
+class PresentationFilter(django_filters.FilterSet):
+    class Meta:
+        model = Presentation
+        fields = ("name", "subspecialty")
+
+    def __init__(
+        self,
+        data=None,
+        queryset=None,
+        prefix=None,
+        strict=None,
+        user=None,
+        request=None,
+    ):
+        super(PresentationFilter, self).__init__(
+            data=data, queryset=queryset, prefix=prefix, request=request
+        )
+        pass
+
+
+class PathologicalProcessFilter(django_filters.FilterSet):
+    class Meta:
+        model = PathologicalProcess
+        fields = ("name",)
+
+    def __init__(
+        self,
+        data=None,
+        queryset=None,
+        prefix=None,
+        strict=None,
+        user=None,
+        request=None,
+    ):
+        super(PathologicalProcessFilter, self).__init__(
+            data=data, queryset=queryset, prefix=prefix, request=request
+        )
+        pass
+
+
+class SubspecialtyFilter(django_filters.FilterSet):
+    class Meta:
+        model = Subspecialty
+        fields = ("name",)
+
+    def __init__(
+        self,
+        data=None,
+        queryset=None,
+        prefix=None,
+        strict=None,
+        user=None,
+        request=None,
+    ):
+        super(SubspecialtyFilter, self).__init__(
+            data=data, queryset=queryset, prefix=prefix, request=request
+        )
+        pass