FallenBirdyForm/app/bird/views.py

import environ
import names

from django.contrib.auth.decorators import login_required
from django.db.models import Q, Sum
from django.shortcuts import redirect, render, HttpResponse
from django.core.mail import send_mail, BadHeaderError
from smtplib import SMTPException

from .forms import BirdAddForm, BirdEditForm, BirdSpeciesForm
from .models import Bird, FallenBird

from sendemail.message import messagebody
from sendemail.models import BirdEmail, Emailadress

env = environ.Env()


@login_required(login_url="account_login")
def bird_create(request):
    form = BirdAddForm(initial={"bird_identifier": names.get_first_name()})
    rescuer_id = None

    # 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 = None

        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

            # Send email to all related email addresses based on bird species notification settings
            bird = Bird.objects.get(id=fs.bird_id)
            
            # Get email addresses that match the bird species' notification settings
            email_addresses = []
            
            # Check each notification category and add matching email addresses
            if bird.melden_an_naturschutzbehoerde:
                naturschutz_emails = Emailadress.objects.filter(is_naturschutzbehoerde=True)
                email_addresses.extend([email.email_address for email in naturschutz_emails])
            
            if bird.melden_an_jagdbehoerde:
                jagd_emails = Emailadress.objects.filter(is_jagdbehoerde=True)
                email_addresses.extend([email.email_address for email in jagd_emails])
            
            if bird.melden_an_wildvogelhilfe_team:
                team_emails = Emailadress.objects.filter(is_wildvogelhilfe_team=True)
                email_addresses.extend([email.email_address for email in team_emails])
            
            # Remove duplicates
            email_addresses = list(set(email_addresses))
            
            if email_addresses:  # Only send if there are recipients
                try:
                    send_mail(
                        subject="Wildvogel gefunden!",
                        message=messagebody(
                            fs.date_found, bird, fs.place, fs.diagnostic_finding
                        ),
                        from_email=env("DEFAULT_FROM_EMAIL"),
                        recipient_list=email_addresses,
                    )
                except BadHeaderError:
                    return HttpResponse("Invalid header found.")
                except SMTPException as e:
                    print("There was an error sending an email: ", e)

            return redirect("bird_all")
    context = {"form": form}
    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")
    )
    context = {"birds": birds}
    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"))
        .order_by("date_found")
    )
    context = {"birds": birds}
    return render(request, "bird/bird_inactive.html", context)


@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_species_list(request):
    """List all bird species with their notification settings."""
    birds = Bird.objects.all().order_by("name")
    context = {"birds": birds}
    return render(request, "bird/bird_species_list.html", context)


@login_required(login_url="account_login")
def bird_species_edit(request, id):
    """Edit bird species notification settings."""
    bird_species = Bird.objects.get(id=id)
    form = BirdSpeciesForm(request.POST or None, instance=bird_species)
    
    if request.method == "POST":
        if form.is_valid():
            form.save()
            return redirect("bird_species_list")
    
    context = {"form": form, "bird_species": bird_species}
    return render(request, "bird/bird_species_edit.html", context)