RPGnotes/notes/views/day_views.py

from django.shortcuts import redirect
from django.urls import reverse_lazy
from django.views import generic

from notes.forms import DayForm
from notes.models import IngameDay


def list_day_redirect(request, *args, **kwargs):
    latest_day: IngameDay = IngameDay.objects.first()
    if not latest_day:
        return redirect("dayadd")
    return redirect(latest_day)


class DayDetailView(generic.DetailView):
    template_name = "notes/day_detail.html"
    model = IngameDay
    context_object_name = "day"

    def get_context_data(self, **kwargs):
        data = super().get_context_data(**kwargs)
        data["days"] = IngameDay.objects.filter(campaign__slug=self.kwargs['campslug'])
        return data

    def get_object(self, queryset=None):
        return IngameDay.objects.get(campaign__slug=self.kwargs['campslug'], day=self.kwargs['day'])


class DayCreateView(generic.CreateView):
    template_name = "notes/loot_edit.html"
    model = IngameDay
    form_class = DayForm
    context_object_name = "object"


class DayEditView(generic.UpdateView):
    template_name = "notes/loot_edit.html"
    model = IngameDay
    form_class = DayForm

    def get_context_data(self, **kwargs):
        data = super().get_context_data(**kwargs)
        data['edit'] = True
        return data

    # def get_object(self, queryset=None):
    #     return IngameDay.objects.get(campaign__slug=self.kwargs['campslug'], day=self.kwargs['day'])


class DayDeleteView(generic.DeleteView):
    template_name = "notes/campaign_confirm_delete.html"
    model = IngameDay
    success_url = reverse_lazy('daylist')

    # def get_object(self, queryset=None):
    #     return IngameDay.objects.get(campaign__slug=self.kwargs['campslug'], day=self.kwargs['day'])