NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-26 01:06:11 -06:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

make is_oob_ip and is_primary_ip generic for other models

Browse Source
This commit is contained in:
Jamie Murphy 2023-06-26 23:07:20 +01:00
parent 9c681e63dc
commit a3275a9b6b
1 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
netbox/ipam/models/ip.py
View File

@ -852,26 +852,28 @@ class IPAddress(PrimaryModel):
@property @property
def is_oob_ip(self): def is_oob_ip(self):
if self.assigned_object: if self.assigned_object:
parent = getattr(self.assigned_object, 'parent_object', None)
if self.family == 4: if self.family == 4:
if self.assigned_object.device.oob_ip4: if parent.oob_ip4:
if self.assigned_object.device.oob_ip4.pk == self.pk: if parent.oob_ip4.pk == self.pk:
return True return True
if self.family == 6: if self.family == 6:
if self.assigned_object.device.oob_ip6: if parent.oob_ip6:
if self.assigned_object.device.oob_ip6.pk == self.pk: if parent.oob_ip6.pk == self.pk:
return True return True
return False return False
@property @property
def is_primary_ip(self): def is_primary_ip(self):
if self.assigned_object: if self.assigned_object:
parent = getattr(self.assigned_object, 'parent_object', None)
if self.family == 4: if self.family == 4:
if self.assigned_object.device.primary_ip4: if parent.primary_ip4:
if self.assigned_object.device.primary_ip4.pk == self.pk: if parent.primary_ip4.pk == self.pk:
return True return True
if self.family == 6: if self.family == 6:
if self.assigned_object.device.primary_ip6: if parent.primary_ip6:
if self.assigned_object.device.primary_ip6.pk == self.pk: if parent.primary_ip6.pk == self.pk:
return True return True
return False return False