101 lines
3.4 KiB
Python
101 lines
3.4 KiB
Python
import names
|
|
from django.contrib.auth.decorators import login_required
|
|
from django.shortcuts import HttpResponse, redirect, render
|
|
from rescuer.models import Rescuer
|
|
|
|
from .forms import BirdAddForm, BirdEditForm
|
|
from .models import FallenBird, Bird
|
|
|
|
|
|
@login_required(login_url="account_login")
|
|
def bird_create(request):
|
|
form = BirdAddForm(initial={"bird_identifier": names.get_first_name()})
|
|
# Rescuer for modal usage
|
|
rescuer_id = request.session.get("rescuer_id")
|
|
rescuer = Rescuer.objects.get(id=rescuer_id, user=request.user)
|
|
|
|
# Just show only related rescuers in select field of the form.
|
|
if request.method == "POST":
|
|
form = BirdAddForm(request.POST or None, request.FILES or None)
|
|
rescuer_id = request.session.get("rescuer_id")
|
|
rescuer = Rescuer.objects.get(id=rescuer_id, user=request.user)
|
|
|
|
if form.is_valid():
|
|
fs = form.save(commit=False)
|
|
fs.user = request.user
|
|
fs.rescuer_id = rescuer_id
|
|
fs.save()
|
|
request.session["rescuer_id"] = None
|
|
return redirect("bird_all")
|
|
context = {"form": form, "rescuer": rescuer}
|
|
return render(request, "bird/bird_create.html", context)
|
|
|
|
|
|
@login_required(login_url="account_login")
|
|
def bird_help(request):
|
|
birds = Bird.objects.all()
|
|
context = {"birds": birds}
|
|
return render(request, "bird/bird_help.html", context)
|
|
|
|
|
|
@login_required(login_url="account_login")
|
|
def bird_all(request):
|
|
birds = FallenBird.objects.all()
|
|
# Sum all costs per bird from json
|
|
for bird in birds:
|
|
costs_per_bird = float()
|
|
for item in bird.costs:
|
|
costs_per_bird += float(item['cost_entry'])
|
|
if costs_per_bird == 0.0:
|
|
costs_per_bird = ""
|
|
bird.costs = costs_per_bird
|
|
|
|
rescuer_modal = Rescuer.objects.all()
|
|
context = {"birds": birds, "rescuer_modal": rescuer_modal}
|
|
# Post came from the modal form.
|
|
if request.method == "POST":
|
|
rescuer_id = request._post["rescuer_id"]
|
|
if rescuer_id != "new_rescuer":
|
|
request.session["rescuer_id"] = rescuer_id
|
|
return redirect("bird_create")
|
|
else:
|
|
return redirect("rescuer_create")
|
|
return render(request, "bird/bird_all.html", context)
|
|
|
|
|
|
@login_required(login_url="account_login")
|
|
def bird_recover_all(request):
|
|
return HttpResponse("Show all recovered Birds")
|
|
|
|
|
|
@login_required(login_url="account_login")
|
|
def bird_single(request, id):
|
|
bird = FallenBird.objects.get(id=id)
|
|
form = BirdEditForm(
|
|
request.POST or None,
|
|
request.FILES or None,
|
|
instance=bird)
|
|
if request.method == "POST":
|
|
if form.is_valid():
|
|
fs = form.save(commit=False)
|
|
if fs.status.description != "In Auswilderung":
|
|
fs.aviary = None
|
|
fs.save()
|
|
return redirect("bird_all")
|
|
context = {"form": form, "bird": bird}
|
|
return render(request, "bird/bird_single.html", context)
|
|
|
|
|
|
@login_required(login_url="account_login")
|
|
def bird_delete(request, id):
|
|
bird = FallenBird.objects.get(id=id)
|
|
if request.method == "POST":
|
|
bird.delete()
|
|
return redirect("bird_all")
|
|
context = {"bird": bird}
|
|
return render(request, "bird/bird_delete.html", context)
|
|
|
|
|
|
@login_required(login_url="account_login")
|
|
def bird_recover(request, id):
|
|
return HttpResponse(f"Show recover with ID {id}")
|