Update model_forms.py

This commit is contained in:
Jamie Murphy 2023-06-20 21:12:57 +01:00
parent 89a78a64a2
commit a9b2d0ebfd

View File

@ -365,10 +365,10 @@ class IPAddressForm(TenancyForm, NetBoxModelForm):
prefix_str = f"{self.instance.address.network}/{self.instance.address.prefixlen}" prefix_str = f"{self.instance.address.network}/{self.instance.address.prefixlen}"
allow_assignment_error = True allow_assignment_error = True
if self.instance.vrf is None: if self.instance.vrf is None:
prefix_obj = Prefix.objects.get(prefix=prefix_str) prefix_obj = Prefix.objects.filter(prefix=prefix_str)
else: else:
prefix_obj = Prefix.objects.get(prefix=prefix_str, vrf=self.vrf) prefix_obj = Prefix.objects.filter(prefix=prefix_str, vrf=self.vrf)
if prefix_obj and prefix_obj.is_pool: if prefix_obj.exists() and prefix_obj[0].is_pool:
allow_assignment_error = False allow_assignment_error = False
if allow_assignment_error: if allow_assignment_error: