NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-07-22 20:12:00 -06:00
Code Issues Actions 8 Packages Projects Releases Wiki Activity

Store a private copy of the pk during init and use that with __str__

Browse Source
This commit is contained in:
Saria Hajjar 2020-01-13 14:57:21 +00:00
parent 49fa243b4f
commit 3d78a67343
1 changed files with 7 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
netbox/dcim/models.py
View File

@ -3027,24 +3027,18 @@ class Cable(ChangeLoggedModel):
('termination_b_type', 'termination_b_id'),
)
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
# A copy of the PK to be used by __str__ in case the object is deleted
self._pk = self.pk
def __str__(self):
if self.label:
return self.label
# Save a copy of the PK on the instance since it's nullified if .delete() is called
if not hasattr(self, 'id_string'):
self.id_string = '#{}'.format(self.pk)
return self.id_string
return self.label or '#{}'.format(self._pk)
def get_absolute_url(self):
return reverse('dcim:cable', args=[self.pk])
def delete(self, *args, **kwargs):
# Trigger the __str__ method to save the pk into `self.id_string`
str(self)
super().delete(*args, **kwargs)
def clean(self):
# Validate that termination A exists