.

atlas/filters.py

@@ -3,6 +3,7 @@ import django_filters
 from .models import Case, Condition, Finding, Series, Structure
 from django.contrib.auth.models import User
 
+
 def get_authors(request):
     if request is None:
         return User.objects.none()
@@ -15,55 +16,116 @@ def get_authors(request):
     return queryset
 
 
-class AtlasFilter(django_filters.FilterSet):
-    author = django_filters.ModelMultipleChoiceFilter(queryset=get_authors, null_label="No author")
+class CaseFilter(django_filters.FilterSet):
+    author = django_filters.ModelMultipleChoiceFilter(
+        queryset=get_authors, null_label="No author"
+    )
+    foo = django_filters.filters.ModelMultipleChoiceFilter(
+        field_name='differential__condition',
+        to_field_name='Diff',
+        queryset=Condition.objects.all(),
+    )
+
     class Meta:
         model = Case
-        fields = (#"site", 
-        "created_date", "open_access", "author")
+        fields = ("created_date", "open_access", "author")  # "site",
 
-    def __init__(self, data=None, queryset=None, prefix=None, strict=None, user=None, request=None):
+    def __init__(
+        self,
+        data=None,
+        queryset=None,
+        prefix=None,
+        strict=None,
+        user=None,
+        request=None,
+    ):
         if not request.user.groups.filter(name="atlas_checker").exists():
-            #queryset = queryset.filter(open_access=True) | queryset.filter(author__id=request.user.id)
+            # queryset = queryset.filter(open_access=True) | queryset.filter(author__id=request.user.id)
             queryset = queryset.filter(author__id=request.user.id)
-        super(AtlasFilter, self).__init__(data=data, queryset=queryset, prefix=prefix, request=request)
+        super(CaseFilter, self).__init__(
+            data=data, queryset=queryset, prefix=prefix, request=request
+        )
         pass
 
+
 class SeriesFilter(django_filters.FilterSet):
     class Meta:
         model = Series
-        fields = ("modality","case", "examination", "author")
+        fields = ("modality", "case", "examination", "author")
 
-    def __init__(self, data=None, queryset=None, prefix=None, strict=None, user=None, request=None):
+    def __init__(
+        self,
+        data=None,
+        queryset=None,
+        prefix=None,
+        strict=None,
+        user=None,
+        request=None,
+    ):
         if not request.user.groups.filter(name="atlas_checker").exists():
-            #queryset = queryset.filter(open_access=True) | queryset.filter(author__id=request.user.id)
+            # 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)
+        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)
+    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)
+    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
+    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