You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
114 lines
3.3 KiB
Python
114 lines
3.3 KiB
Python
3 years ago
|
"""
|
||
|
Author: Michel Peltriaux
|
||
|
Organization: Struktur- und Genehmigungsdirektion Nord, Rhineland-Palatinate, Germany
|
||
2 years ago
|
Contact: ksp-servicestelle@sgdnord.rlp.de
|
||
|
Created on: 18.08.22
|
||
3 years ago
|
|
||
|
"""
|
||
3 years ago
|
from django.db.models import QuerySet, Q
|
||
3 years ago
|
|
||
2 years ago
|
from konova.filters.table_filters import AbstractTableFilter, CheckboxTableFilter, QueryTableFilter, \
|
||
|
SelectionTableFilter
|
||
3 years ago
|
|
||
|
|
||
|
class SelectionCompensationTableFilter(SelectionTableFilter):
|
||
|
""" Specialization of regular SelectionTableFilter for compensation model
|
||
|
|
||
|
"""
|
||
|
def filter_reg_office(self, queryset, name, value):
|
||
|
return queryset.filter(
|
||
|
intervention__responsible__registration_office=value
|
||
|
)
|
||
|
|
||
|
def filter_cons_office(self, queryset, name, value):
|
||
|
return queryset.filter(
|
||
|
intervention__responsible__conservation_office=value
|
||
|
)
|
||
3 years ago
|
|
||
|
|
||
3 years ago
|
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
|
||
|
|
||
|
|
||
3 years ago
|
class CheckboxCompensationTableFilter(CheckboxTableFilter):
|
||
|
""" Specialization of regular CheckboxTableFilter for compensation model
|
||
3 years ago
|
|
||
|
"""
|
||
|
|
||
3 years ago
|
def filter_show_all(self, queryset, name, value) -> QuerySet:
|
||
3 years ago
|
""" Filters queryset depending on value of 'show_all' setting
|
||
|
|
||
|
Args:
|
||
|
queryset ():
|
||
|
name ():
|
||
|
value ():
|
||
|
|
||
|
Returns:
|
||
|
|
||
|
"""
|
||
|
if not value:
|
||
|
return queryset.filter(
|
||
3 years ago
|
Q(intervention__users__in=[self.user]) | # requesting user has access
|
||
2 years ago
|
Q(intervention__teams__in=self.user.shared_teams)
|
||
3 years ago
|
).distinct()
|
||
3 years ago
|
else:
|
||
|
return queryset
|
||
|
|
||
3 years ago
|
def filter_show_recorded(self, queryset, name, value) -> QuerySet:
|
||
3 years ago
|
""" Filters queryset depending on value of 'show_recorded' setting
|
||
|
|
||
|
Args:
|
||
|
queryset ():
|
||
|
name ():
|
||
|
value ():
|
||
|
|
||
|
Returns:
|
||
|
|
||
|
"""
|
||
|
if not value:
|
||
|
return queryset.filter(
|
||
3 years ago
|
intervention__recorded=None,
|
||
3 years ago
|
)
|
||
|
else:
|
||
|
return queryset
|
||
3 years ago
|
|
||
|
|
||
3 years ago
|
class CompensationTableFilter(AbstractTableFilter):
|
||
3 years ago
|
""" TableFilter for compensations
|
||
3 years ago
|
|
||
3 years ago
|
Based widely on InterventionTableFilter.
|
||
|
Just some minor changes for Compensation model.
|
||
3 years ago
|
|
||
|
"""
|
||
3 years ago
|
def __init__(self, user=None, *args, **kwargs):
|
||
|
super().__init__(*args, **kwargs)
|
||
3 years ago
|
|
||
3 years ago
|
self.user = user
|
||
3 years ago
|
qs = kwargs.get("queryset", None)
|
||
|
request_data = kwargs.get("data", None)
|
||
|
|
||
3 years ago
|
# Overwrite all filters for special needs of compensations
|
||
3 years ago
|
self.selection_filter = SelectionCompensationTableFilter(
|
||
3 years ago
|
data=request_data,
|
||
|
queryset=qs,
|
||
|
)
|
||
3 years ago
|
self.query_filter = QueryCompensationTableFilter(
|
||
|
data=request_data,
|
||
|
queryset=self.selection_filter.qs,
|
||
|
)
|
||
3 years ago
|
self.checkbox_filter = CheckboxCompensationTableFilter(
|
||
|
user=user,
|
||
|
data=request_data,
|
||
|
queryset=self.query_filter.qs,
|
||
|
)
|
||
3 years ago
|
# Overwrite final queryset as well
|
||
3 years ago
|
self.qs = self.checkbox_filter.qs
|