nuitdelinfo_2021/rescue/views.py

from django.contrib.auth.decorators import login_required
from django.db.models import Q
from django.http import Http404, HttpResponseRedirect, HttpResponseBadRequest, JsonResponse
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):
    return render(request, "article/articles.html", {"rescues": Rescue.objects.all()})


def details(request, rescue_id: int):
    return render(request, "article/article.html", {"rescue": get_rescue(rescue_id)})


@login_required
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})


@login_required
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})


def ajax_search(request, text: str):
    return JsonResponse(Rescue.objects.filter(Q(name__icontains=text) | Q(date__icontains=text) | Q(resume__icontains=text)))