from typing import Any, Dict, Optional, Type from django.forms.models import BaseModelForm from django.http import HttpResponse from django.shortcuts import get_object_or_404 from django.views.generic import TemplateView, ListView, DetailView from django.views.generic.edit import CreateView, UpdateView, DeleteView from django.contrib.auth.mixins import LoginRequiredMixin from guild.forms import PlaySessionForm from guild.models import Adventure, PlaySession class AdventureListView(LoginRequiredMixin, ListView): model = Adventure context_object_name = "adventures" class CreateAdventureView(LoginRequiredMixin, CreateView): model = Adventure fields = ["name", "description", "master"] class AdventureDetailView(LoginRequiredMixin, DetailView): model = Adventure def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) context["sessions"] = PlaySession.objects.filter( adventure=self.object ).order_by("date") return context class AdventureUpdateView(LoginRequiredMixin, UpdateView): model = Adventure fields = ["name", "description", "master"] class AdventureDeleteView(LoginRequiredMixin, DeleteView): model = Adventure success_url = "/" class PlaySessionCreateView(LoginRequiredMixin, CreateView): model = PlaySession form_class = PlaySessionForm def dispatch(self, request, *args, **kwargs): self.adventure = get_object_or_404(Adventure, pk=kwargs["pk"]) return super().dispatch(request, *args, **kwargs) def form_valid(self, form: BaseModelForm) -> HttpResponse: form.instance.adventure = self.adventure return super().form_valid(form) class PlaySessionUpdateView(LoginRequiredMixin, UpdateView): model = PlaySession form_class = PlaySessionForm pk_url_kwarg = "playsession_pk" def dispatch(self, request, *args, **kwargs): self.adventure = get_object_or_404(Adventure, pk=kwargs["pk"]) return super().dispatch(request, *args, **kwargs) class PlaySessionDeleteView(LoginRequiredMixin, DeleteView): model = PlaySession pk_url_kwarg = "playsession_pk" def dispatch(self, request, *args, **kwargs): self.adventure = get_object_or_404(Adventure, pk=kwargs["pk"]) return super().dispatch(request, *args, **kwargs) def get_success_url(self) -> str: return self.adventure.get_absolute_url()