forked from rc/aircox
code quality
This commit is contained in:
@ -3,17 +3,16 @@ from django.contrib.auth.mixins import LoginRequiredMixin, UserPassesTestMixin
|
||||
from django.utils.translation import gettext_lazy as _
|
||||
from django.views.generic import ListView
|
||||
|
||||
from .log import LogListView
|
||||
from ..models.log import LogArchiver
|
||||
from .log import LogListView
|
||||
|
||||
|
||||
__all__ = ['AdminMixin', 'StatisticsView']
|
||||
__all__ = ["AdminMixin", "StatisticsView"]
|
||||
|
||||
|
||||
class AdminMixin(LoginRequiredMixin, UserPassesTestMixin):
|
||||
title = ''
|
||||
title = ""
|
||||
init_app = True
|
||||
""" If true, create vue app. """
|
||||
"""If true, create vue app."""
|
||||
|
||||
@property
|
||||
def station(self):
|
||||
@ -24,19 +23,23 @@ class AdminMixin(LoginRequiredMixin, UserPassesTestMixin):
|
||||
|
||||
def get_context_data(self, **kwargs):
|
||||
kwargs.update(admin.site.each_context(self.request))
|
||||
kwargs.setdefault('title', self.title)
|
||||
kwargs.setdefault('station', self.station)
|
||||
kwargs.setdefault('init_app', self.init_app)
|
||||
kwargs.setdefault("title", self.title)
|
||||
kwargs.setdefault("station", self.station)
|
||||
kwargs.setdefault("init_app", self.init_app)
|
||||
return super().get_context_data(**kwargs)
|
||||
|
||||
|
||||
class StatisticsView(AdminMixin, LogListView, ListView):
|
||||
template_name = 'admin/aircox/statistics.html'
|
||||
redirect_date_url = 'admin:tools-stats'
|
||||
title = _('Statistics')
|
||||
template_name = "admin/aircox/statistics.html"
|
||||
redirect_date_url = "admin:tools-stats"
|
||||
title = _("Statistics")
|
||||
date = None
|
||||
|
||||
def get_object_list(self, logs, full=False):
|
||||
if not logs.exists():
|
||||
logs = LogArchiver().load(self.station, self.date) if self.date else []
|
||||
logs = (
|
||||
LogArchiver().load(self.station, self.date)
|
||||
if self.date
|
||||
else []
|
||||
)
|
||||
return super().get_object_list(logs, True)
|
||||
|
Reference in New Issue
Block a user