diff --git a/inventory/views.py b/inventory/views.py index 18d65b9a..6bfe13af 100644 --- a/inventory/views.py +++ b/inventory/views.py @@ -5867,6 +5867,13 @@ class OpportunityCreateView(CreateView, SuccessMessageMixin, LoginRequiredMixin) initial["stage"] = models.Stage.QUALIFICATION return initial + def get_form(self, form_class=None): + form = super().get_form(form_class) + dealer = get_object_or_404(models.Dealer,slug=self.kwargs.get("dealer_slug")) + form.fields['car'].queryset = models.Car.objects.filter(dealer=dealer) + form.fields['lead'].queryset = models.Lead.objects.filter(dealer=dealer) + return form + def form_valid(self, form): dealer = get_object_or_404(models.Dealer,slug=self.kwargs.get("dealer_slug")) instance = form.save(commit=False) @@ -5909,6 +5916,13 @@ class OpportunityUpdateView(LoginRequiredMixin, SuccessMessageMixin, UpdateView) template_name = "crm/opportunities/opportunity_form.html" success_message = "Opportunity updated successfully." + def get_form(self, form_class=None): + form = super().get_form(form_class) + dealer = get_object_or_404(models.Dealer,slug=self.kwargs.get("dealer_slug")) + form.fields['car'].queryset = models.Car.objects.filter(dealer=dealer) + form.fields['lead'].queryset = models.Lead.objects.filter(dealer=dealer) + return form + def get_success_url(self): return reverse_lazy("opportunity_detail", kwargs={"dealer_slug":self.kwargs.get("dealer_slug"),"slug": self.object.slug})