Преглед изворни кода

Fix test failure by casting to IPAddress

Daniel Sheppard пре 3 дана
родитељ
комит
6e24ce6a1a
1 измењених фајлова са 2 додато и 2 уклоњено
  1. 2 2
      netbox/ipam/models/ip.py

+ 2 - 2
netbox/ipam/models/ip.py

@@ -744,7 +744,7 @@ class IPRange(ContactsMixin, PrimaryModel):
 
         # Set the parent prefix
         self.prefix = Prefix.find_parent_prefix_range(
-            networks=[self.start_address.ip, self.end_address.ip],
+            networks=[netaddr.IPAddress(self.start_address), netaddr.IPAddress(self.end_address)],
             vrf=self.vrf
         )
 
@@ -1100,7 +1100,7 @@ class IPAddress(ContactsMixin, PrimaryModel):
         self.dns_name = self.dns_name.lower()
 
         # Set the parent prefix
-        self.prefix = Prefix.find_parent_prefix(self.address.ip, vrf=self.vrf)
+        self.prefix = Prefix.find_parent_prefix(netaddr.IPAddress(self.address), vrf=self.vrf)
 
         super().save(*args, **kwargs)