JourneyJoker/auction/views.py

from django.views.generic import CreateView, UpdateView, View, ListView, DetailView
from django.shortcuts import redirect, get_object_or_404
from django.contrib import messages
from django.urls import reverse
from django.contrib.gis.geos import Point
from django.contrib.gis.db.models.functions import Distance
from django.conf import settings

from public.mixins import InConstructionMixin
from partners.mixins import PartnerProfileRequiredMixin
from localauth.helpers import name_to_coords
from partners.models import Establishment
from payment.models import BillingAddress, Invoice, InvoiceItem, InvoicePayment
from clients.models import ClientProfile

from .models import Inquiry, Offer
from .forms import InquiryProcessForm

class InquiryCreateView(CreateView):
    model = Inquiry
    fields = ["destination_name", "budget", "arrival", "min_nights", "adults", "children"]

    def get(self, request, *args, **kwargs):
        return redirect("/")

    def form_valid(self, form):
        form.instance.destination_coords = self.clean_destination_coords()
        form.instance.destination_radius = 5000
        return super().form_valid(form)

    def form_invalid(self, form, *args, **kwargs):
        for field in form:
            for error in field.errors:
                messages.error(self.request, f"{field.name}: {error}")

        return redirect("/")

    def get_success_url(self):
        return reverse("auction:process_inquiry", args=(self.object.uuid,))

    def clean_destination_coords(self):
        lat, lon = name_to_coords(self.request.POST.get("destination_name"))
        return Point(lon, lat)

class InquiryProcessView(InConstructionMixin, UpdateView):
    form_class = InquiryProcessForm
    model = Inquiry
    template_name = "auction/process.html"

    def get_object(self):
        return Inquiry.objects.get(uuid=self.kwargs["uuid"])

    def get_initial(self):
        initial = super().get_initial()

        try:
            initial["country"] = self.request.user.clientprofile.country.code
        except:
            pass

        return initial

    def form_valid(self, form):
        profile, _ = ClientProfile.objects.get_or_create(user=self.request.user)
        profile.first_name = form.cleaned_data["first_name"]
        profile.last_name = form.cleaned_data["last_name"]
        profile.street = form.cleaned_data["street"]
        profile.city = form.cleaned_data["city"]
        profile.zip = form.cleaned_data["zip"]
        profile.state = form.cleaned_data["state"]
        profile.country = form.cleaned_data["country"]
        profile.save()

        form.instance.client = profile

        return super().form_valid(form)

    def form_invalid(self, form, *args, **kwargs):
        for field in form:
            for error in field.errors:
                messages.error(self.request, f"{field.name}: {error}")

        return redirect("/")

    def get_success_url(self):
        return reverse("auction:inquiry_payment", args=(self.object.uuid,))

class InquiryPaymentView(InConstructionMixin, View):
    def get(self, request, *args, **kwargs):
        inquiry = Inquiry.objects.get(uuid=kwargs["uuid"])
        try:
            invoice = inquiry.invoice
        except Invoice.DoesNotExist:
            invoice = Invoice.from_inquiry(inquiry)
        payment_url = InvoicePayment.from_invoice(invoice, inquiry.gateway)

        return redirect(payment_url)

class OfferSelectionView(InConstructionMixin, DetailView):
    model = Inquiry
    template_name = "auction/offer_select.html"

    def get_object(self):
        return get_object_or_404(Inquiry, uuid=self.kwargs["uuid"])

class OfferSelectionTableView(InConstructionMixin, ListView):
    model = Offer
    content_type = "text/javascript"
    template_name = "auction/offer_table.js"

    def get_queryset(self):
        inquiry = get_object_or_404(Inquiry, uuid=self.kwargs["uuid"])

class OfferCreationView(InConstructionMixin, PartnerProfileRequiredMixin, CreateView):
    model = Offer
    template_name = "auction/offer_create.html"
    fields = []

    def dispatch(self, request, *args, **kwargs):
        self.establishment = self.get_establishment()

        if not self.establishment:
            messages.warning(request, "Um bieten zu können, muss zuerst eine Unterkunft im System hinterlegt werden!")
            return redirect("partners:establishment_register")

        return super().dispatch(request, *args, **kwargs)

    def get_establishment(self):
        establishment = self.kwargs.get("establishment", None)
        kwargs = {"owner": self.request.user.partnerprofile}

        if establishment:
            kwargs["id"] = establishment
            return get_object_or_404(Establishment, **kwargs)
        else:
            return Establishment.objects.filter(**kwargs).first()

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context["establishment"] = self.establishment
        context["inquiry"] = get_object_or_404(Inquiry, uuid=self.kwargs.get("inquiry"))
        return context
        
class BiddingListView(InConstructionMixin, PartnerProfileRequiredMixin, ListView):
    model = Inquiry
    template_name = "auction/bidding_list.html"

    def dispatch(self, request, *args, **kwargs):
        self.establishment = self.get_establishment()

        if not self.establishment:
            messages.warning(request, "Um bieten zu können, muss zuerst eine Unterkunft im System hinterlegt werden!")
            return redirect("partners:establishment_register")

        return super().dispatch(request, *args, **kwargs)

    def get_establishment(self):
        establishment = self.kwargs.get("id", None)
        kwargs = {"owner": self.request.user.partnerprofile}

        if establishment:
            kwargs["id"] = establishment
            return get_object_or_404(Establishment, **kwargs)
        else:
            return Establishment.objects.filter(**kwargs).first()

    def get_queryset(self):
        establishment = self.get_establishment()
        return Inquiry.objects.annotate(distance=Distance("destination_coords", establishment.coords)).exclude(activated__isnull=True)

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context["establishment"] = self.establishment
        return context