start atlas app

atlas/filters.py

@@ -0,0 +1,42 @@
+import django_filters
+
+from .models import Case, Series
+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_checker").exists():
+        queryset = User.objects.filter(id=request.user.id)
+    else:
+        queryset = User.objects.all()
+
+    return queryset
+
+
+class AtlasFilter(django_filters.FilterSet):
+    author = django_filters.ModelMultipleChoiceFilter(queryset=get_authors, null_label="No author")
+    class Meta:
+        model = Case
+        fields = (#"site", 
+        "created_date", "open_access", "author")
+
+    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(author__id=request.user.id)
+        super(AtlasFilter, self).__init__(data=data, queryset=queryset, prefix=prefix, request=request)
+        pass
+
+class SeriesFilter(django_filters.FilterSet):
+    class Meta:
+        model = Series
+        fields = ("modality","atlas", "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_checker").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