import names from django.contrib.auth.decorators import login_required from django.db.models import Q, Sum from django.shortcuts import HttpResponse, redirect, render from .forms import BirdAddForm, BirdEditForm from .models import Bird, FallenBird from rescuer.models import Rescuer @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) # 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) # circumstances = Circumstance.objects.all() rescuer_id = request.session.get("rescuer_id") rescuer = Rescuer.objects.get(id=rescuer_id) 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().order_by("name") context = {"birds": birds} return render(request, "bird/bird_help.html", context) @login_required(login_url="account_login") def bird_help_single(request, id): bird = Bird.objects.all().get(id=id) context = {"bird": bird} return render(request, "bird/bird_help_single.html", context) @login_required(login_url="account_login") def bird_all(request): birds = FallenBird.objects.filter(Q(status="1") | Q(status="2")).annotate( total_costs=Sum("costs__costs") ).order_by("date_found") 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_inactive(request): birds = FallenBird.objects.filter( ~Q(status="1") & ~Q(status="2")).annotate( total_costs=Sum("costs__costs")) context = {"birds": birds} return render(request, "bird/bird_inactive.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}")