code quality

This commit is contained in:
bkfox
2023-03-13 17:47:00 +01:00
parent 934817da8a
commit 112770eddf
162 changed files with 4798 additions and 4069 deletions

View File

@ -1,63 +1,86 @@
from django.db import models
from django.contrib.admin import filters
from django.utils.translation import gettext_lazy as _
from django.db import models
from django.utils.http import urlencode
from django.utils.translation import gettext_lazy as _
__all__ = ('DateFieldFilter', 'DateTimeField')
__all__ = ("DateFieldFilter", "DateTimeFieldFilter")
class DateFieldFilter(filters.FieldListFilter):
""" Display date input """
template = 'admin/aircox/filters/date_filter.html'
input_type = 'date'
"""Display date input."""
template = "admin/aircox/filters/date_filter.html"
input_type = "date"
def __init__(self, field, request, params, model, model_admin, field_path):
self.field_generic = '%s__' % field_path
self.date_params = {k: v for k, v in params.items()
if k.startswith(self.field_generic)}
self.field_generic = "%s__" % field_path
self.date_params = {
k: v for k, v in params.items() if k.startswith(self.field_generic)
}
exact_lookup = 'date' if isinstance(field, models.DateTimeField) else 'exact'
exact_lookup = (
"date" if isinstance(field, models.DateTimeField) else "exact"
)
# links as: (label, param, input_type|None, value)
self.links = [(_('Exact'), self.field_generic + exact_lookup, self.input_type),
(_('Since'), self.field_generic + 'gte', self.input_type),
(_('Until'), self.field_generic + 'lte', self.input_type)]
self.links = [
(_("Exact"), self.field_generic + exact_lookup, self.input_type),
(_("Since"), self.field_generic + "gte", self.input_type),
(_("Until"), self.field_generic + "lte", self.input_type),
]
if field.null:
self.links.insert(0, (_('None'), self.field_generic + 'isnull', None, '1'))
self.query_attrs = {k:v for k,v in request.GET.items()
if k not in self.date_params}
self.links.insert(
0, (_("None"), self.field_generic + "isnull", None, "1")
)
self.query_attrs = {
k: v for k, v in request.GET.items() if k not in self.date_params
}
self.query_string = urlencode(self.query_attrs)
super().__init__(field, request, params, model, model_admin, field_path)
super().__init__(
field, request, params, model, model_admin, field_path
)
def expected_parameters(self):
return [link[1] for link in self.links]
def choices(self, changelist):
yield {'label': _('Any'),
'type': None,
'query_string': self.query_string}
yield {
"label": _("Any"),
"type": None,
"query_string": self.query_string,
}
for link in self.links:
value = len(link) > 3 and link[3] or self.date_params.get(link[1])
yield {
'label': link[0], 'name': link[1], 'value': value,
'type': link[2],
'query_attrs': self.query_attrs,
'query_string': urlencode({link[1]: value}) + '&' + self.query_string
if value else self.query_string,
"label": link[0],
"name": link[1],
"value": value,
"type": link[2],
"query_attrs": self.query_attrs,
"query_string": urlencode({link[1]: value})
+ "&"
+ self.query_string
if value
else self.query_string,
}
class DateTimeFieldFilter(DateFieldFilter):
""" Display datetime input """
input_type = 'datetime-local'
"""Display datetime input."""
input_type = "datetime-local"
filters.FieldListFilter.register(
lambda f: isinstance(f, models.DateField), DateFieldFilter, take_priority=True)
lambda f: isinstance(f, models.DateField),
DateFieldFilter,
take_priority=True,
)
filters.FieldListFilter.register(
lambda f: isinstance(f, models.DateTimeField), DateTimeFieldFilter, take_priority=True)
lambda f: isinstance(f, models.DateTimeField),
DateTimeFieldFilter,
take_priority=True,
)