|
| 1 | +import datetime |
| 2 | + |
| 3 | +from django.contrib import admin, messages |
| 4 | +from django.db import models |
| 5 | +from django.conf import settings |
| 6 | +from django.utils import timezone |
| 7 | +from django.utils.translation import gettext_lazy as _ |
| 8 | + |
| 9 | + |
| 10 | +class DateRangeFilter(admin.FieldListFilter): |
| 11 | + # Reference https://github.com/andreynovikov/django-daterangefilter |
| 12 | + template = 'adminlte/date_range_filter.html' |
| 13 | + date_format = 'YYYY/MM/DD' |
| 14 | + |
| 15 | + def __init__(self, field, request, params, model, model_admin, field_path): |
| 16 | + self.field_name = field_path |
| 17 | + self.lookup_kwarg_gte = '{}__gte'.format(field_path) |
| 18 | + self.lookup_kwarg_lte = '{}__lte'.format(field_path) |
| 19 | + self.lookup_gte = params.get(self.lookup_kwarg_gte) |
| 20 | + self.lookup_lte = params.get(self.lookup_kwarg_lte) |
| 21 | + |
| 22 | + if self.lookup_gte == '': |
| 23 | + params.pop(self.lookup_kwarg_gte) |
| 24 | + |
| 25 | + if self.lookup_lte == '': |
| 26 | + params.pop(self.lookup_kwarg_lte) |
| 27 | + if self.lookup_gte and self.lookup_lte: |
| 28 | + self.lookup_val = '{} - {}'.format(self.lookup_gte, self.lookup_lte) |
| 29 | + # if we are filtering DateTimeField we should add one day to final date |
| 30 | + if "__" in field_path: |
| 31 | + related_model, field = field_path.split("__") |
| 32 | + field = model._meta.get_field(related_model).related_model._meta.get_field(field) |
| 33 | + else: |
| 34 | + field = model._meta.get_field(field_path) |
| 35 | + |
| 36 | + if isinstance(field, models.DateTimeField): |
| 37 | + try: |
| 38 | + gte_date = datetime.datetime.strptime(self.lookup_gte, '%Y-%m-%d') |
| 39 | + lte_date = datetime.datetime.strptime(self.lookup_lte, '%Y-%m-%d') |
| 40 | + lte_date = lte_date + datetime.timedelta(seconds=3600 * 24 - 1) |
| 41 | + if settings.USE_TZ: |
| 42 | + gte_date = timezone.make_aware(gte_date, timezone.get_current_timezone()) |
| 43 | + lte_date = timezone.make_aware(lte_date, timezone.get_current_timezone()) |
| 44 | + params[self.lookup_kwarg_gte] = gte_date.strftime('%Y-%m-%d %H:%M:%S%z') |
| 45 | + params[self.lookup_kwarg_lte] = lte_date.strftime('%Y-%m-%d %H:%M:%S%z') |
| 46 | + except ValueError: |
| 47 | + messages.add_message(request, messages.ERROR, |
| 48 | + _("Invalid date for '%(field_name)s' field range filter") % { |
| 49 | + 'field_name': field.verbose_name}) |
| 50 | + else: |
| 51 | + self.lookup_val = '' |
| 52 | + |
| 53 | + super().__init__(field, request, params, model, model_admin, field_path) |
| 54 | + |
| 55 | + def choices(self, changelist): |
| 56 | + yield { |
| 57 | + 'field_name': self.field_path, |
| 58 | + 'value': self.lookup_val, |
| 59 | + 'date_format': self.date_format, |
| 60 | + 'query_string': changelist.get_query_string(remove=self._get_expected_fields()) |
| 61 | + } |
| 62 | + |
| 63 | + def expected_parameters(self): |
| 64 | + return self._get_expected_fields() |
| 65 | + |
| 66 | + def _get_expected_fields(self): |
| 67 | + return [self.lookup_kwarg_gte, self.lookup_kwarg_lte] |
0 commit comments