mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
68 lines
1.7 KiB
Python
68 lines
1.7 KiB
Python
from django.contrib.contenttypes.models import ContentType
|
|
from django.db import transaction
|
|
|
|
|
|
def compile_path_node(ct_id, object_id):
|
|
return f'{ct_id}:{object_id}'
|
|
|
|
|
|
def decompile_path_node(repr):
|
|
ct_id, object_id = repr.split(':')
|
|
return int(ct_id), int(object_id)
|
|
|
|
|
|
def object_to_path_node(obj):
|
|
"""
|
|
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 compile_path_node(ct.pk, obj.pk)
|
|
|
|
|
|
def path_node_to_object(repr):
|
|
"""
|
|
Given the string representation of a path node, return the corresponding instance.
|
|
"""
|
|
ct_id, object_id = decompile_path_node(repr)
|
|
ct = ContentType.objects.get_for_id(ct_id)
|
|
return ct.model_class().objects.get(pk=object_id)
|
|
|
|
|
|
def flatten_path(path):
|
|
"""
|
|
Flatten a two-dimensional array (list of lists) into a flat list.
|
|
"""
|
|
ret = []
|
|
for step in path:
|
|
ret.extend(step)
|
|
return ret
|
|
|
|
|
|
def create_cablepath(terminations):
|
|
"""
|
|
Create CablePaths for all paths originating from the specified set of nodes.
|
|
|
|
:param terminations: Iterable of CableTermination objects
|
|
"""
|
|
from dcim.models import CablePath
|
|
|
|
cp = CablePath.from_origin(terminations)
|
|
if cp:
|
|
cp.save()
|
|
|
|
|
|
def rebuild_paths(obj):
|
|
"""
|
|
Rebuild all CablePaths which traverse the specified node
|
|
"""
|
|
from dcim.models import CablePath
|
|
|
|
cable_paths = CablePath.objects.filter(path__contains=obj)
|
|
|
|
with transaction.atomic():
|
|
for cp in cable_paths:
|
|
cp.delete()
|
|
if cp.origin:
|
|
create_cablepath(cp.origin)
|