mirror of
https://github.com/nikdoof/limetime.git
synced 2025-12-13 09:42:26 +00:00
39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
from django.views.generic import ListView, CreateView
|
|
from django.utils.timezone import now
|
|
from timer.models import Timer, Location, Station, Moon, System
|
|
from timer.forms import TimerForm
|
|
|
|
|
|
class TimerCreateView(CreateView):
|
|
model = Timer
|
|
form_class = TimerForm
|
|
|
|
|
|
class TimerListView(ListView):
|
|
model = Timer
|
|
template_name = 'timer/timer_list.html'
|
|
|
|
def get_queryset(self):
|
|
qs = super(TimerListView, self).get_queryset().select_related('location', 'station', 'moon', 'system')
|
|
|
|
if int(self.request.GET.get('all', 0)) == 0:
|
|
qs = qs.filter(expiration__gt=now())
|
|
|
|
typ = int(self.request.GET.get('type', 0))
|
|
if typ == 1:
|
|
qs = [m for m in qs if m.location.get_type == 'Station']
|
|
if typ == 2:
|
|
qs = [m for m in qs if m.location.get_type == 'System']
|
|
if typ == 3:
|
|
qs = [m for m in qs if m.location.get_type == 'Moon']
|
|
return qs
|
|
|
|
|
|
def get_context_data(self, **kwargs):
|
|
ctx = super(TimerListView, self).get_context_data(**kwargs)
|
|
ctx.update({
|
|
'list_all': int(self.request.GET.get('all', 0)),
|
|
'list_type': int(self.request.GET.get('type', 0)),
|
|
})
|
|
return ctx
|