mirror of
https://github.com/drygdryg/netbox-plugin-interface-sync
synced 2024-11-25 18:10:52 +03:00
139 lines
4.5 KiB
Python
139 lines
4.5 KiB
Python
import re
|
|
from typing import Iterable
|
|
from django.shortcuts import render, redirect
|
|
from django.contrib import messages
|
|
from django.core.exceptions import ObjectDoesNotExist
|
|
from django.conf import settings
|
|
|
|
config = settings.PLUGINS_CONFIG['netbox_interface_sync']
|
|
|
|
|
|
def split(s):
|
|
for x, y in re.findall(r"(\d*)(\D*)", s):
|
|
yield "", int(x or "0")
|
|
yield y, 0
|
|
|
|
|
|
def natural_keys(c):
|
|
return tuple(split(c))
|
|
|
|
|
|
def human_sorted(iterable: Iterable):
|
|
return sorted(iterable, key=natural_keys)
|
|
|
|
|
|
def get_components(request, device, components, unified_components, unified_component_templates, component_type):
|
|
# List of components and components templates presented in the unified format
|
|
overall_powers = list(set(unified_component_templates + unified_components))
|
|
overall_powers.sort(key=lambda o: natural_keys(o.name))
|
|
|
|
comparison_templates = []
|
|
comparison_components = []
|
|
for i in overall_powers:
|
|
try:
|
|
comparison_templates.append(
|
|
unified_component_templates[unified_component_templates.index(i)]
|
|
)
|
|
except ValueError:
|
|
comparison_templates.append(None)
|
|
|
|
try:
|
|
comparison_components.append(
|
|
unified_components[unified_components.index(i)]
|
|
)
|
|
except ValueError:
|
|
comparison_components.append(None)
|
|
|
|
comparison_items = list(zip(comparison_templates, comparison_components))
|
|
return render(
|
|
request,
|
|
"netbox_interface_sync/components_comparison.html",
|
|
{
|
|
"component_type": component_type,
|
|
"comparison_items": comparison_items,
|
|
"templates_count": len(unified_component_templates),
|
|
"components_count": len(components),
|
|
"device": device,
|
|
},
|
|
)
|
|
|
|
|
|
def post_components(
|
|
request, device, components, component_templates, ObjectType, ObjectTemplateType, unified_component, unified_component_templates, component_type
|
|
):
|
|
# Manually validating components and component templates lists
|
|
add_to_device = filter(
|
|
lambda i: i in component_templates.values_list("id", flat=True),
|
|
map(int, filter(lambda x: x.isdigit(), request.POST.getlist("add_to_device"))),
|
|
)
|
|
remove_from_device = filter(
|
|
lambda i: i in components.values_list("id", flat=True),
|
|
map(
|
|
int,
|
|
filter(lambda x: x.isdigit(), request.POST.getlist("remove_from_device")),
|
|
),
|
|
)
|
|
|
|
# Remove selected component from the device and count them
|
|
deleted = ObjectType.objects.filter(id__in=remove_from_device).delete()[0]
|
|
|
|
# Add selected components to the device and count them
|
|
add_to_device_component = ObjectTemplateType.objects.filter(id__in=add_to_device)
|
|
|
|
bulk_create = []
|
|
updated = 0
|
|
keys_to_avoid = ["id"]
|
|
|
|
if config["exclude_description"]:
|
|
keys_to_avoid.append("description")
|
|
|
|
for i in add_to_device_component.values():
|
|
to_create = False
|
|
|
|
try:
|
|
tmp = components.get(name=i["name"])
|
|
except ObjectDoesNotExist:
|
|
tmp = ObjectType()
|
|
tmp.device = device
|
|
to_create = True
|
|
|
|
for k in i.keys():
|
|
if k not in keys_to_avoid:
|
|
setattr(tmp, k, i[k])
|
|
|
|
if to_create:
|
|
bulk_create.append(tmp)
|
|
else:
|
|
tmp.save()
|
|
updated += 1
|
|
|
|
created = len(ObjectType.objects.bulk_create(bulk_create))
|
|
|
|
# Rename selected components
|
|
fixed = 0
|
|
for component, component_comparison in unified_component:
|
|
try:
|
|
# Try to extract a component template with the corresponding name
|
|
corresponding_template = unified_component_templates[
|
|
unified_component_templates.index(component_comparison)
|
|
]
|
|
component.name = corresponding_template.name
|
|
component.save()
|
|
fixed += 1
|
|
except ValueError:
|
|
pass
|
|
|
|
# Generating result message
|
|
message = []
|
|
if created > 0:
|
|
message.append(f"created {created} {component_type}")
|
|
if updated > 0:
|
|
message.append(f"updated {updated} {component_type}")
|
|
if deleted > 0:
|
|
message.append(f"deleted {deleted} {component_type}")
|
|
if fixed > 0:
|
|
message.append(f"fixed {fixed} {component_type}")
|
|
messages.success(request, "; ".join(message).capitalize())
|
|
|
|
return redirect(request.path)
|