pflaenz.li/pflaenzli/pflaenzli/views.py

from django.shortcuts import render, get_object_or_404, redirect, reverse
from django.utils import timezone
from django.contrib import messages
from django.contrib.auth import login
from django.contrib.auth.decorators import login_required
from django.views.decorators.csrf import csrf_protect
from django.views.decorators.http import require_POST
from django.http import HttpResponseForbidden

from .forms import CreateOfferForm, RegistrationForm, FilterForm, WishForm

from .models import PflaenzliUser, Offer, Wish
from .mail import send_offer_email
from .upload import generate_unique_filename
from .utils.distance import calculate_distance, filter_by_distance


def list_offers(request, filters=None):
    offers = Offer.objects.all()

    if request.method == "POST":
        form = FilterForm(request.POST, request.FILES)
        if form.is_valid():
            offers = filter_offers(offers, form)
    else:
        form = FilterForm()

    return render(request, "offer/search.html", {"offers": offers, "form": form})


@login_required
def create_offer(request):
    if request.method == "POST":
        form = CreateOfferForm(request.POST, request.FILES)
        if form.is_valid():
            offer = form.save(commit=False)
            offer.image.name = generate_unique_filename(form.cleaned_data['image'].name)
            offer.user = request.user
            offer.title = offer.title.title()
            offer.save()
            messages.success(request, "Offer uploaded successfully!")
            return redirect("offer_detail", offer.id)
    else:
        form = CreateOfferForm()

    return render(request, "basic_form.html", {"form": form, "button_label": "Create", "title": "Create Offer"})


def offer_detail(request, offer_id):
    offer = get_object_or_404(Offer, id=offer_id)
    wishes = Wish.objects.filter(user=offer.user)

    if request.user.is_authenticated:
        if offer.zipcode and request.user.zipcode:
            if offer.zipcode == request.user.zipcode:
                dist = 0
            else:
                dist = calculate_distance(offer.zipcode, request.user.zipcode)
        else:
            dist = None
    else:
        dist = None

    return render(request, "offer/detail.html", {"offer": offer, "wishes": wishes, "dist": dist})


@login_required
def offer_delete(request, offer_id):
    offer = get_object_or_404(Offer, id=offer_id)
    if offer.user != request.user:
        return HttpResponseForbidden()

    offer.delete()
    messages.success(request, "Offer deleted successfully!")
    return redirect("list_offers")


@login_required
def offer_edit(request, offer_id):
    offer = get_object_or_404(Offer, id=offer_id)
    if offer.user != request.user:
        return HttpResponseForbidden()

    if request.method == "POST":
        form = CreateOfferForm(request.POST, request.FILES, instance=offer)
        if form.is_valid():
            offer = form.save(commit=False)
            offer.image.name = generate_unique_filename(form.cleaned_data['image'].name)
            offer.user = request.user
            offer.save()
            messages.success(request, "Offer updated successfully!")
            return redirect("offer_detail", offer.id)
    else:
        form = CreateOfferForm(instance=offer)

    return render(request, "basic_form.html", {"form": form, "button_label": "Update", "title": "Edit Offer"})


@login_required
def wishlist(request, user_id):
    wishes = Wish.objects.filter(user=user_id)

    if user_id == request.user.id:
        title = "Your"
        if request.method == "POST":
            form = WishForm(request.POST)
            if form.is_valid():
                wish = form.save(commit=False)
                wish.user = request.user
                wish.save()
                messages.success(request, "Wish added successfully!")
        form = WishForm()
    else:
        form = None
        user = get_object_or_404(PflaenzliUser, id=user_id)
        title = f"{user.username}'s"

    return render(request, "user/wish.html", {"title": title, "form": form, "wishes": wishes, "own": user_id == request.user.id})


@login_required
def user_edit(request):
    if request.method == "POST":
        form = RegistrationForm(request.POST, instance=request.user)
        if form.is_valid():
            form.save()
            messages.success(request, "Account details updated successfully!")
            return redirect("user_profile")
    else:
        form = RegistrationForm(instance=request.user)

    return render(request, "basic_form.html", {"form": form, "button_label": "Save", "title": "Edit Account Details"})


@csrf_protect
@require_POST
@login_required
def delete_wish(request, wish_id):
    wish = get_object_or_404(Wish, id=wish_id)

    if wish.user != request.user:
        messages.error(request, "You don't have permission to delete this wish!")
        return redirect("wishlist", request.user.id)

    wish.delete()

    return redirect("wishlist", request.user.id)


@login_required
def offer_trade(request, offer_id):
    offer = get_object_or_404(Offer, id=offer_id)
    sender = request.user
    recipient = offer.user

    if sender != recipient:
        send_offer_email(request, offer, sender, recipient)
        messages.success(request, f"{recipient.username} was successfully notified")
    else:
        messages.error(request, "You can't trade with yourself!")

    return redirect("offer_detail", offer_id)


def user_detail(request, user_id):
    user = get_object_or_404(PflaenzliUser, id=user_id)
    offers = Offer.objects.filter(user=user_id)
    wishes = Wish.objects.filter(user=user_id)

    return render(request, "user/public.html", {"user": user, "offers": offers, "wishes": wishes})


def register_user(request):
    if request.method == "POST":
        form = RegistrationForm(request.POST)
        if form.is_valid():
            user = form.save()
            login(request, user)
            return redirect("index")
    else:
        form = RegistrationForm()

    return render(request, "basic_form.html", {"form": form, "button_label": "Register", "title": "Registeration"})


def filter_offers(offers, form):
    if form.cleaned_data['text']:
        offers = offers.filter(title__icontains=form.cleaned_data['text'])

    if form.cleaned_data['zipcode']:
        if form.cleaned_data['distance']:
            offers = filter_by_distance(offers, form.cleaned_data['zipcode'], form.cleaned_data['distance'])
        else:
            offers = offers.filter(zipcode=int(form.cleaned_data['zipcode']))

    return offers