RPGnotes/characters/views.py

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

from characters.forms import CharacterForm
from characters.models import Character


def list_character_redirect(request, *args, **kwargs):
    first_character: Character = Character.objects.first()
    if not first_character:
        return redirect("characteradd")
    return redirect(first_character)


class CharacterDetailView(generic.DetailView):
    template_name = "characters/detail.jinja"
    model = Character
    context_object_name = "character"

    def get_context_data(self, **kwargs):
        data = super().get_context_data(**kwargs)
        data["player_characters"] = Character.objects.filter(
            player__isnull=False
        ).select_related()
        data["npcs"] = Character.objects.filter(
            player__isnull=True
        ).select_related()
        return data


class CharacterCreateView(generic.CreateView):
    template_name = "loot/edit.html"
    model = Character
    form_class = CharacterForm
    context_object_name = "object"


class CharacterEditView(generic.UpdateView):
    template_name = "loot/edit.html"
    model = Character
    form_class = CharacterForm

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


class CharacterDeleteView(generic.DeleteView):
    template_name = "common/confirm_delete.html"
    model = Character
    success_url = reverse_lazy('characterlist')