guild-journal/guild/views/__init__.py

from datetime import date
from django.urls import reverse
from django.views.generic import TemplateView
from django.views.generic.edit import CreateView
from django.contrib.auth.mixins import LoginRequiredMixin
from django.contrib.auth.models import User
from django.contrib.auth.forms import UserCreationForm
from guild.models import NPC, Adventure, Character, Player, Reward


# Create your views here.
class HomeView(LoginRequiredMixin, TemplateView):
    template_name = "guild/home.html"

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        advs = Adventure.objects.prefetch_related("playsession_set", "master").all()
        advs = sorted(
            advs,
            key=lambda x: x.last_session.date if x.last_session else date.min,
            reverse=True,
        )
        context["adventures"] = advs
        context["players"] = Player.objects.all()
        context["characters"] = Character.objects.all()
        context["npcs"] = NPC.objects.all()

        context["rewards"] = Reward.objects.filter(character=None).order_by(
            "-playsession__date"
        )
        context["resources"] = {}
        for reward in context["rewards"]:
            if reward.resource.name not in context["resources"]:
                context["resources"][reward.resource.name] = 0
            context["resources"][reward.resource.name] += reward.amount
        return context


class RegisterView(CreateView):
    model = User
    form_class = UserCreationForm

    def get_success_url(self):
        return reverse("guild:login")