mirror/netbox-community-netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00
Code

Optimize path node representations

Browse Source
This commit is contained in:
Jeremy Stretch
2020-10-06 16:34:03 -04:00
parent a072d40594
commit 2c9ae60dec
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
netbox/dcim/utils.py
View File

@ -5,12 +5,20 @@ from .exceptions import CableTraceSplit
def object_to_path_node(obj):
return f'{obj._meta.model_name}:{obj.pk}'
"""
Return a representation of an object suitable for inclusion in a CablePath path. Node representation is in the
form <ContentType ID>:<Object ID>.
"""
ct = ContentType.objects.get_for_model(obj)
return f'{ct.pk}:{obj.pk}'
def path_node_to_object(repr):
model_name, object_id = repr.split(':')
model_class = ContentType.objects.get(model=model_name).model_class()
"""
Given a path node representation, return the corresponding object.
"""
ct_id, object_id = repr.split(':')
model_class = ContentType.objects.get(pk=ct_id).model_class()
return model_class.objects.get(pk=int(object_id))