42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
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 django.db.models import Q
|
|
from guild.models import Character, Reward
|
|
|
|
|
|
class CharacterListView(LoginRequiredMixin, ListView):
|
|
model = Character
|
|
context_object_name = "characters"
|
|
|
|
|
|
class CreateCharacterView(LoginRequiredMixin, CreateView):
|
|
model = Character
|
|
fields = ["name", "description", "player", "status"]
|
|
|
|
|
|
class CharacterDetailView(LoginRequiredMixin, DetailView):
|
|
model = Character
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
context["rewards"] = Reward.objects.filter(
|
|
Q(character=self.object) | Q(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 CharacterUpdateView(LoginRequiredMixin, UpdateView):
|
|
model = Character
|
|
fields = ["name", "description", "player", "status"]
|
|
|
|
|
|
class CharacterDeleteView(LoginRequiredMixin, DeleteView):
|
|
model = Character
|
|
success_url = "/"
|