diff --git a/inventory/models.py b/inventory/models.py index 3cac8c54..93d36403 100644 --- a/inventory/models.py +++ b/inventory/models.py @@ -954,7 +954,7 @@ class Dealer(models.Model, LocalizedNameMixin): return None def get_vendors(self): - return self.vendors.all() + return VendorModel.objects.filter(entity_model=self.entity) @property def is_staff_exceed_quota_limit(self): quota = self.user_quota diff --git a/inventory/views.py b/inventory/views.py index c02f14fc..d341c21b 100644 --- a/inventory/views.py +++ b/inventory/views.py @@ -1318,12 +1318,12 @@ class CarUpdateView( def get_success_url(self): return reverse("car_detail", kwargs={"pk": self.object.pk}) - def get_form_kwargs(self): - kwargs = super().get_form_kwargs() - kwargs["instance"] = self.get_object() - kwargs["vendor"] = self.get_object().dealer.get_vendors() - return kwargs - + def get_form(self, form_class=None): + form = super().get_form(form_class) + dealer = get_user_type(self.request) + print(dealer.get_vendors()) + form.fields["vendor"].queryset = dealer.get_vendors() + return form class CarDeleteView( LoginRequiredMixin, PermissionRequiredMixin, SuccessMessageMixin, DeleteView