mattieserver
318b8495b7
* init * added circuit (or at least part of) * refactor circuits + fix js * small changes * small changes * added power + rework settings * small changes
71 lines
2.7 KiB
Python
71 lines
2.7 KiB
Python
from rest_framework.viewsets import ModelViewSet, ViewSet, ReadOnlyModelViewSet, GenericViewSet
|
|
from rest_framework.decorators import action
|
|
from rest_framework.response import Response
|
|
from django.contrib.contenttypes.models import ContentType
|
|
from rest_framework.routers import APIRootView
|
|
|
|
from .serializers import TopologyDummySerializer
|
|
from django.conf import settings
|
|
|
|
from dcim.models import DeviceRole, Device, Cable , PowerPanel, PowerFeed
|
|
from circuits.models import Circuit
|
|
from extras.models import Tag
|
|
|
|
|
|
class TopologyViewsRootView(APIRootView):
|
|
def get_view_name(self):
|
|
return 'TopologyViews'
|
|
|
|
|
|
class SaveCoordsViewSet(ReadOnlyModelViewSet):
|
|
queryset = Device.objects.all()
|
|
serializer_class = TopologyDummySerializer
|
|
|
|
@action(detail=False, methods=['patch'])
|
|
def save_coords(self, request):
|
|
results = {}
|
|
if settings.PLUGINS_CONFIG["netbox_topology_views"]["allow_coordinates_saving"]:
|
|
device_id = None
|
|
x_coord = None
|
|
y_coord = None
|
|
if "node_id" in request.data:
|
|
if request.data["node_id"]:
|
|
device_id = request.data["node_id"]
|
|
if "x" in request.data:
|
|
if request.data["x"]:
|
|
x_coord = request.data["x"]
|
|
if "y" in request.data:
|
|
if request.data["y"]:
|
|
y_coord = request.data["y"]
|
|
|
|
if device_id.startswith("c"):
|
|
device_id = device_id.lstrip('c')
|
|
actual_device= Circuit.objects.get(id=device_id)
|
|
elif device_id.startswith("p"):
|
|
device_id = device_id.lstrip('p')
|
|
actual_device= PowerPanel.objects.get(id=device_id)
|
|
elif device_id.startswith("f"):
|
|
device_id = device_id.lstrip('f')
|
|
actual_device= PowerFeed.objects.get(id=device_id)
|
|
else:
|
|
actual_device= Device.objects.get(id=device_id)
|
|
|
|
if "coordinates" in actual_device.custom_field_data:
|
|
actual_device.custom_field_data["coordinates"] = "%s;%s" % (x_coord,y_coord)
|
|
actual_device.save()
|
|
results["status"] = "saved coords"
|
|
else:
|
|
try:
|
|
actual_device.custom_field_data["coordinates"] = "%s;%s" % (x_coord,y_coord)
|
|
actual_device.save()
|
|
results["status"] = "saved coords"
|
|
except :
|
|
results["status"] = "coords custom field not created"
|
|
return Response(status=500)
|
|
|
|
return Response(results)
|
|
else:
|
|
results["status"] = "not allowed to save coords"
|
|
return Response(results, status=500)
|
|
|