64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
from django.shortcuts import render
|
|
from django.urls import reverse_lazy, reverse
|
|
from django.views.generic.base import TemplateView
|
|
from django.views.generic.edit import FormView, CreateView, UpdateView, DeleteView
|
|
from django.views.generic.detail import DetailView
|
|
from django.views.generic.list import ListView
|
|
|
|
from charts.models import Chart, Entry
|
|
from charts.forms import EntryForm
|
|
|
|
|
|
from django.db.models import (Exists, OuterRef,
|
|
Prefetch, Subquery, Count, Sum, F, Q, Value)
|
|
from django.db.models.functions import Length, Upper
|
|
|
|
# Entries
|
|
class ListView(ListView):
|
|
model = Entry
|
|
pk_url_kwarg = 'entry_pk'
|
|
|
|
class CreateView(CreateView):
|
|
model = Entry
|
|
pk_url_kwarg = 'entry_pk'
|
|
template_name_suffix = '_create_form'
|
|
form_class = EntryForm
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
context['chart'] = Chart.objects.get(pk=self.kwargs['pk'])
|
|
return context
|
|
|
|
def form_valid(self, form):
|
|
form.instance.chart = Chart.objects.get(pk=self.kwargs['pk'])
|
|
return super().form_valid(form)
|
|
|
|
def get_success_url(self):
|
|
return reverse('chart-detail', kwargs={'pk': self.kwargs['pk']})
|
|
|
|
class DetailView(DetailView):
|
|
model = Entry
|
|
pk_url_kwarg = 'entry_pk'
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
context['chart'] = Chart.objects.get(pk=self.kwargs['pk'])
|
|
return context
|
|
|
|
class UpdateView(UpdateView):
|
|
model = Entry
|
|
pk_url_kwarg = 'entry_pk'
|
|
form_class = EntryForm
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
context['chart'] = Chart.objects.get(pk=self.kwargs['pk'])
|
|
return context
|
|
|
|
class DeleteView(DeleteView):
|
|
model = Entry
|
|
pk_url_kwarg = 'entry_pk'
|
|
|
|
def get_success_url(self):
|
|
return reverse('chart-detail', kwargs={'pk': self.kwargs['pk']})
|