nuitdelinfo_2021/rescue/views.py

49 lines
1.5 KiB
Python

from django.http import Http404, HttpResponseRedirect, HttpResponseBadRequest
from django.shortcuts import render
from django.urls import reverse
from rescue.forms import SubmitRescue
from rescue.models import Rescue
def get_rescue(rescue_id: int) -> Rescue:
try:
return Rescue.objects.get(pk=rescue_id)
except Rescue.DoesNotExist:
raise Http404("Rescue does not exist")
def index(request, rescue_id: int):
return render(request, "article/article.html", {"rescue": get_rescue(rescue_id)})
def submit(request):
if request.method == "POST":
form = SubmitRescue(request.POST)
if form.is_valid():
rescue = form.save()
return HttpResponseRedirect(reverse(index, args=[rescue.pk]))
else:
form = SubmitRescue()
return render(request, "article/submit.html", {"form": form})
def edit(request, rescue_id: int):
edited_rescue = get_rescue(rescue_id)
if not edited_rescue.validated:
return HttpResponseBadRequest("This entry is not validated, you can't edit it")
if request.method == "POST":
form = SubmitRescue(request.POST)
if form.is_valid():
rescue = form.save()
rescue.pending_edit_of = edited_rescue
rescue.save()
return HttpResponseRedirect(reverse(index, args=[rescue.pk]))
else:
edited_rescue.pk = None
form = SubmitRescue(instance=edited_rescue)
return render(request, "article/edit.html", {"form": form, "edit_id": edited_rescue})