Custom range django-filters

custom_range_filters.py

from django import forms

from django_filters.filterset import FilterSet
from django_filters.widgets import RangeWidget
from django_filters.filters import Filter, RangeFilter


class CustomMultiValueField(forms.MultiValueField):
    widget = RangeWidget

    def compress(self, data_list):
        if data_list:
            return slice(*data_list)
        return None


class CustomDateRangeField(CustomMultiValueField):
    def __init__(self, *args, **kwargs):
        fields = (
            forms.DateField(), 
            forms.DateField(),
        )
        super(DateRangeField, self).__init__(fields, *args, **kwargs)



class CustomTimeRangeField(CustomMultiValueField):
    def __init__(self, *args, **kwargs):
        fields = (
            forms.TimeField(),
            forms.TimeField(),
        )
        super(TimeRangeField, self).__init__(fields, *args, **kwargs)


class CustomDateTimeRangeField(CustomMultiValueField):
    def __init__(self, *args, **kwargs):
        fields = (
            forms.DateTimeField(),
            forms.DateTimeField(),
        )
        super(DateTimeRangeField, self).__init__(fields, *args, **kwargs)


class CustomRangeFilter(Filter):
    def filter(self, qs, value):
        if value:
            lookup = '%s__range' % self.name
            return qs.filter(**{lookup: (value.start, value.stop)})
        return qs


class CustomDateRangeFilter(CustomRangeFilter):
    field_class = DateRangeField


class CustomTimeRangeFilter(CustomRangeFilter):
    field_class = TimeRangeField


class CustomDateTimeRangeFilter(CustomRangeFilter):
    field_class = DateTimeRangeField