from django import forms from django.http import HttpResponseRedirect from django.shortcuts import render, get_object_or_404 # Create your views here. from django.urls import reverse_lazy, reverse from django.views import generic from django.views.generic import CreateView, UpdateView, RedirectView from writingtogether.models import Story, StoryPart, StoryRound, Participant class IndexView(generic.ListView): template_name = 'writingtogether/index.html' context_object_name = 'open_story_round_list' def get_queryset(self): # TODO: show open & finished rounds return StoryRound.objects.order_by('-created')[:5] class DetailView(generic.DetailView): model = StoryRound template_name = 'writingtogether/detail.html' class StoryRoundCreate(CreateView): model = StoryRound fields = ['name', 'participants', 'number_of_rounds'] def form_valid(self, form): self.object = form.save() for user in form.cleaned_data['participants']: participant = Participant.objects.create( user=user, order_by=1, # TODO: get order by from form story_round=self.object ) Story.objects.create( part_of_round=self.object, started_by=participant ) return HttpResponseRedirect(self.get_success_url()) class StoryUpdate(UpdateView): model = Story fields = ['name'] class StoryPartUpdate(UpdateView): model = StoryPart fields = ['text'] template_name = 'writingtogether/story_part.html' #def form_valid(self, form): # form.instance.created_by = self.request.user # form.instance.previous_part_id = self.kwargs['previous'] # form.instance.part_of_id = self.kwargs['story_pk'] # return super().form_valid(form) class RedirectToNextOpenPart(RedirectView): permanent = False query_string = True pattern_name = 'writing:update_story_part' def get_redirect_url(self, *args, **kwargs): story_round = get_object_or_404(StoryRound, pk=kwargs['story_round_pk']) story_part = story_round.get_next_story_part(user=self.request.user) kwargs['story_pk'] = story_part.part_of.pk kwargs['story_part_pk'] = story_part.pk return super().get_redirect_url(*args, **kwargs)