#132 | #121: backoffice / dev-1.0-121 (#131)

cfr #121

Co-authored-by: Christophe Siraut <d@tobald.eu.org>
Co-authored-by: bkfox <thomas bkfox net>
Co-authored-by: Thomas Kairos <thomas@bkfox.net>
Reviewed-on: rc/aircox#131
Co-authored-by: Chris Tactic <ctactic@noreply.git.radiocampus.be>
Co-committed-by: Chris Tactic <ctactic@noreply.git.radiocampus.be>

aircox/filters.py

@@ -1,16 +1,30 @@
-import django_filters as filters
+from django.contrib.auth.models import User
+from django.db.models import Q
 from django.utils.translation import gettext_lazy as _
+import django_filters as filters
 
-from .models import Episode, Page
+from . import models
+
+
+__all__ = (
+    "PageFilters",
+    "EpisodeFilters",
+    "ImageFilterSet",
+    "SoundFilterSet",
+    "TrackFilterSet",
+    "UserFilterSet",
+    "GroupFilterSet",
+    "UserGroupFilterSet",
+)
 
 
 class PageFilters(filters.FilterSet):
     q = filters.CharFilter(method="search_filter", label=_("Search"))
 
     class Meta:
-        model = Page
+        model = models.Page
         fields = {
-            "category__id": ["in"],
+            "category__id": ["in", "exact"],
             "pub_date": ["exact", "gte", "lte"],
         }
 
@@ -22,10 +36,81 @@ class EpisodeFilters(PageFilters):
     podcast = filters.BooleanFilter(method="podcast_filter", label=_("Podcast"))
 
     class Meta:
-        model = Episode
+        model = models.Episode
         fields = PageFilters.Meta.fields.copy()
 
     def podcast_filter(self, queryset, name, value):
         if value:
             return queryset.filter(sound__is_public=True).distinct()
         return queryset.filter(sound__isnull=True)
+
+
+class ImageFilterSet(filters.FilterSet):
+    search = filters.CharFilter(field_name="search", method="search_filter")
+
+    def search_filter(self, queryset, name, value):
+        return queryset.filter(original_filename__icontains=value)
+
+
+class SoundFilterSet(filters.FilterSet):
+    station = filters.NumberFilter(field_name="program__station__id")
+    program = filters.NumberFilter(field_name="program_id")
+    # episode = filters.NumberFilter(field_name="episode_id")
+    search = filters.CharFilter(field_name="search", method="search_filter")
+
+    class Meta:
+        model = models.Sound
+        fields = {
+            #    "episode": ["in", "exact", "isnull"],
+        }
+
+    def search_filter(self, queryset, name, value):
+        return queryset.search(value)
+
+
+class TrackFilterSet(filters.FilterSet):
+    artist = filters.CharFilter(field_name="artist", lookup_expr="icontains")
+    album = filters.CharFilter(field_name="album", lookup_expr="icontains")
+    title = filters.CharFilter(field_name="title", lookup_expr="icontains")
+
+
+class UserFilterSet(filters.FilterSet):
+    search = filters.CharFilter(field_name="search", method="search_filter")
+    in_group = filters.NumberFilter(field_name="in_group", method="in_group_filter")
+    not_in_group = filters.NumberFilter(field_name="not_in_group", method="not_in_group_filter")
+
+    def in_group_filter(self, queryset, name, value):
+        return queryset.filter(groups__in=[value])
+
+    def not_in_group_filter(self, queryset, name, value):
+        return queryset.exclude(groups__in=[value])
+
+    def search_filter(self, queryset, name, value):
+        return queryset.filter(
+            Q(username__icontains=value) | Q(first_name__icontains=value) | Q(last_name__icontains=value)
+        )
+
+
+class GroupFilterSet(filters.FilterSet):
+    search = filters.CharFilter(field_name="search", method="search_filter")
+    no_user = filters.NumberFilter(field_name="no_user", method="no_user_filter")
+
+    def no_user_filter(self, queryset, name, value):
+        return queryset.exclude(user__in=[value])
+
+    def search_filter(self, queryset, name, value):
+        return queryset.filter(Q(name__icontains=value) | Q(program__title__icontains=value))
+
+
+class UserGroupFilterSet(filters.FilterSet):
+    class Meta:
+        model = User.groups.through
+        fields = ["group", "user"]
+
+    def filter_queryset(self, queryset):
+        queryset = super().filter_queryset(queryset)
+        if self.form.cleaned_data.get("user"):
+            queryset = queryset.order_by("group__name")
+        elif self.form.cleaned_data.get("group"):
+            queryset = queryset.order_by("user__first_name")
+        return queryset