# 61 Filter frontend

* adds filtering for file number
* updates translations

compensation/filters.py

@@ -8,11 +8,23 @@ Created on: 29.07.21
 import django_filters
 from django.utils.translation import gettext_lazy as _
 from django import forms
-from django.db.models import QuerySet
+from django.db.models import QuerySet, Q
 
 from konova.filters import QueryTableFilter, CheckboxTableFilter
 
 
+class QueryCompensationTableFilter(QueryTableFilter):
+    """ Specialization of regular QueryTableFilter for compensation model
+
+    """
+    def filter_file_number(self, queryset, name, value) -> QuerySet:
+        queryset = queryset.filter(
+            Q(intervention__responsible__registration_file_number__icontains=value) |
+            Q(intervention__responsible__conservation_file_number__icontains=value)
+        )
+        return queryset
+
+
 class CheckboxCompensationTableFilter(CheckboxTableFilter):
     """ Specialization of regular CheckboxTableFilter for compensation model
 
@@ -71,7 +83,7 @@ class CompensationTableFilter:
         qs = kwargs.get("queryset", None)
         request_data = kwargs.get("data", None)
 
-        self.query_filter = QueryTableFilter(
+        self.query_filter = QueryCompensationTableFilter(
             user=user,
             data=request_data,
             queryset=qs,
@@ -89,7 +101,11 @@ class CheckboxEcoAccountTableFilter(CheckboxTableFilter):
         method='filter_only_show_unrecorded',
         label=_("Show only unrecorded"),
         label_suffix=_(""),
-        widget=forms.CheckboxInput()
+        widget=forms.CheckboxInput(
+            attrs={
+                "class": "form-check-input",
+            }
+        )
     )
 
     def filter_show_all(self, queryset, name, value) -> QuerySet: