101 lines
No EOL
3.5 KiB
Python
101 lines
No EOL
3.5 KiB
Python
from django.views.generic import CreateView, UpdateView, ListView
|
|
from django.urls import reverse_lazy
|
|
from django.http import HttpResponseRedirect
|
|
from django.db.models import F
|
|
from django.shortcuts import get_list_or_404
|
|
|
|
from .models import PartnerProfile, Establishment
|
|
from .mixins import PartnerProfileRequiredMixin
|
|
|
|
from auction.models import Inquiry, Offer
|
|
from public.mixins import InConstructionMixin
|
|
from localauth.mixins import LoginRequiredMixin
|
|
|
|
class PartnerRegistrationView(InConstructionMixin, LoginRequiredMixin, CreateView):
|
|
model = PartnerProfile
|
|
exclude = ["user"]
|
|
template_name = "partners/signup.html"
|
|
fields = ["company", "vat_id", "first_name", "last_name", "street", "city", "zip", "state", "country"]
|
|
|
|
def dispatch(self, request, *args, **kwargs):
|
|
try:
|
|
PartnerProfile.objects.get(user=request.user)
|
|
return HttpResponseRedirect(reverse_lazy("partners:profile"))
|
|
except (PartnerProfile.DoesNotExist, TypeError):
|
|
return super().dispatch(request, *args, **kwargs)
|
|
|
|
def form_valid(self, form):
|
|
form.instance.user = self.request.user
|
|
return super().form_valid(form)
|
|
|
|
def get_success_url(self):
|
|
return reverse_lazy("partners:profile")
|
|
|
|
def get_initial(self):
|
|
try:
|
|
client = self.request.user.clientprofile
|
|
|
|
return {
|
|
"company": client.company,
|
|
"vat_id": client.vat_id,
|
|
"first_name": client.first_name,
|
|
"last_name": client.last_name,
|
|
"street": client.street,
|
|
"city": client.city,
|
|
"zip": client.zip,
|
|
"state": client.state,
|
|
"country": client.country
|
|
}
|
|
|
|
except:
|
|
return {}
|
|
|
|
class PartnerProfileView(PartnerProfileRequiredMixin, UpdateView):
|
|
model = PartnerProfile
|
|
exclude = ["user"]
|
|
template_name = "partners/profile.html"
|
|
fields = ["company", "vat_id", "first_name", "last_name", "street", "city", "zip", "state", "country"]
|
|
|
|
def get_success_url(self):
|
|
return reverse_lazy("partners:profile")
|
|
|
|
def get_object(self, queryset=None):
|
|
return self.request.user.partnerprofile
|
|
|
|
class BiddingListView(PartnerProfileRequiredMixin, ListView):
|
|
model = Inquiry
|
|
template_name = "partners/bidding_list.html"
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.establishment = self.get_establishment()
|
|
|
|
def get_establishment(self):
|
|
establishment = self.kwargs.get("id", None)
|
|
kwargs = {"owner": self.request.user.partnerprofile}
|
|
|
|
if establishment:
|
|
kwargs["id"] = establishment
|
|
|
|
return get_list_or_404(Establishment, **kwargs).first()
|
|
|
|
def get_queryset(self):
|
|
return Inquiry.objects.filter(active=True).distance(self.establishment.coords).filter(distance__lte=F("destination_radius"))
|
|
|
|
class OffersListView(PartnerProfileRequiredMixin, ListView):
|
|
model = Offer
|
|
template_name = "partners/offer_list.html"
|
|
|
|
def get_queryset(self):
|
|
return Offer.objects.filter(establishment__in=self.request.user.partnerprofile.establishment_set.all())
|
|
|
|
class EstablishmentsListView(PartnerProfileRequiredMixin, ListView):
|
|
model = Establishment
|
|
template_name = "partners/establishment_list.html"
|
|
|
|
def get_queryset(self):
|
|
return self.request.user.partnerprofile.establishment_set
|
|
|
|
class EstablishmentRequestView(PartnerProfileRequiredMixin, ListView):
|
|
model = Establishment
|
|
template_name = "partners/establishment_edit.html" |