views.py 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774
  1. from collections import OrderedDict
  2. import svgwrite
  3. from django.conf import settings
  4. from django.db.models import Count, F
  5. from django.http import HttpResponseForbidden, HttpResponseBadRequest, HttpResponse
  6. from django.shortcuts import get_object_or_404, reverse
  7. from django.utils.http import urlencode
  8. from drf_yasg import openapi
  9. from drf_yasg.openapi import Parameter
  10. from drf_yasg.utils import swagger_auto_schema
  11. from rest_framework.decorators import action
  12. from rest_framework.mixins import ListModelMixin
  13. from rest_framework.response import Response
  14. from rest_framework.viewsets import GenericViewSet, ViewSet
  15. from circuits.models import Circuit
  16. from dcim import constants, filters
  17. from dcim.models import (
  18. Cable, ConsolePort, ConsolePortTemplate, ConsoleServerPort, ConsoleServerPortTemplate, Device, DeviceBay,
  19. DeviceBayTemplate, DeviceRole, DeviceType, FrontPort, FrontPortTemplate, Interface, InterfaceTemplate,
  20. Manufacturer, InventoryItem, Platform, PowerFeed, PowerOutlet, PowerOutletTemplate, PowerPanel, PowerPort,
  21. PowerPortTemplate, Rack, RackGroup, RackReservation, RackRole, RearPort, RearPortTemplate, Region, Site,
  22. VirtualChassis,
  23. )
  24. from extras.api.serializers import RenderedGraphSerializer
  25. from extras.api.views import CustomFieldModelViewSet
  26. from extras.models import Graph
  27. from ipam.models import Prefix, VLAN
  28. from utilities.api import (
  29. get_serializer_for_model, IsAuthenticatedOrLoginNotRequired, FieldChoicesViewSet, ModelViewSet, ServiceUnavailable,
  30. )
  31. from utilities.utils import get_subquery, foreground_color
  32. from virtualization.models import VirtualMachine
  33. from . import serializers
  34. from .exceptions import MissingFilterException
  35. #
  36. # Field choices
  37. #
  38. class DCIMFieldChoicesViewSet(FieldChoicesViewSet):
  39. fields = (
  40. (Cable, ['length_unit', 'status', 'termination_a_type', 'termination_b_type', 'type']),
  41. (ConsolePort, ['type', 'connection_status']),
  42. (ConsolePortTemplate, ['type']),
  43. (ConsoleServerPort, ['type']),
  44. (ConsoleServerPortTemplate, ['type']),
  45. (Device, ['face', 'status']),
  46. (DeviceType, ['subdevice_role']),
  47. (FrontPort, ['type']),
  48. (FrontPortTemplate, ['type']),
  49. (Interface, ['type', 'mode']),
  50. (InterfaceTemplate, ['type']),
  51. (PowerOutlet, ['type', 'feed_leg']),
  52. (PowerOutletTemplate, ['type', 'feed_leg']),
  53. (PowerPort, ['type', 'connection_status']),
  54. (PowerPortTemplate, ['type']),
  55. (Rack, ['outer_unit', 'status', 'type', 'width']),
  56. (RearPort, ['type']),
  57. (RearPortTemplate, ['type']),
  58. (Site, ['status']),
  59. )
  60. # Mixins
  61. class CableTraceMixin(object):
  62. @action(detail=True, url_path='trace')
  63. def trace(self, request, pk):
  64. """
  65. Trace a complete cable path and return each segment as a three-tuple of (termination, cable, termination).
  66. """
  67. obj = get_object_or_404(self.queryset.model, pk=pk)
  68. # Initialize the path array
  69. path = []
  70. for near_end, cable, far_end in obj.trace(follow_circuits=True):
  71. # Serialize each object
  72. serializer_a = get_serializer_for_model(near_end, prefix='Nested')
  73. x = serializer_a(near_end, context={'request': request}).data
  74. if cable is not None:
  75. y = serializers.TracedCableSerializer(cable, context={'request': request}).data
  76. else:
  77. y = None
  78. if far_end is not None:
  79. serializer_b = get_serializer_for_model(far_end, prefix='Nested')
  80. z = serializer_b(far_end, context={'request': request}).data
  81. else:
  82. z = None
  83. path.append((x, y, z))
  84. return Response(path)
  85. #
  86. # Regions
  87. #
  88. class RegionViewSet(ModelViewSet):
  89. queryset = Region.objects.annotate(
  90. site_count=Count('sites')
  91. )
  92. serializer_class = serializers.RegionSerializer
  93. filterset_class = filters.RegionFilter
  94. #
  95. # Sites
  96. #
  97. class SiteViewSet(CustomFieldModelViewSet):
  98. queryset = Site.objects.prefetch_related(
  99. 'region', 'tenant', 'tags'
  100. ).annotate(
  101. device_count=get_subquery(Device, 'site'),
  102. rack_count=get_subquery(Rack, 'site'),
  103. prefix_count=get_subquery(Prefix, 'site'),
  104. vlan_count=get_subquery(VLAN, 'site'),
  105. circuit_count=get_subquery(Circuit, 'terminations__site'),
  106. virtualmachine_count=get_subquery(VirtualMachine, 'cluster__site'),
  107. )
  108. serializer_class = serializers.SiteSerializer
  109. filterset_class = filters.SiteFilter
  110. @action(detail=True)
  111. def graphs(self, request, pk):
  112. """
  113. A convenience method for rendering graphs for a particular site.
  114. """
  115. site = get_object_or_404(Site, pk=pk)
  116. queryset = Graph.objects.filter(type__model='site')
  117. serializer = RenderedGraphSerializer(queryset, many=True, context={'graphed_object': site})
  118. return Response(serializer.data)
  119. #
  120. # Rack groups
  121. #
  122. class RackGroupViewSet(ModelViewSet):
  123. queryset = RackGroup.objects.prefetch_related('site').annotate(
  124. rack_count=Count('racks')
  125. )
  126. serializer_class = serializers.RackGroupSerializer
  127. filterset_class = filters.RackGroupFilter
  128. #
  129. # Rack roles
  130. #
  131. class RackRoleViewSet(ModelViewSet):
  132. queryset = RackRole.objects.annotate(
  133. rack_count=Count('racks')
  134. )
  135. serializer_class = serializers.RackRoleSerializer
  136. filterset_class = filters.RackRoleFilter
  137. #
  138. # Racks
  139. #
  140. class RackViewSet(CustomFieldModelViewSet):
  141. queryset = Rack.objects.prefetch_related(
  142. 'site', 'group__site', 'role', 'tenant', 'tags'
  143. ).annotate(
  144. device_count=get_subquery(Device, 'rack'),
  145. powerfeed_count=get_subquery(PowerFeed, 'rack')
  146. )
  147. serializer_class = serializers.RackSerializer
  148. filterset_class = filters.RackFilter
  149. @action(detail=True)
  150. def units(self, request, pk=None):
  151. """
  152. List rack units (by rack)
  153. """
  154. rack = get_object_or_404(Rack, pk=pk)
  155. face = request.GET.get('face', 'front')
  156. exclude_pk = request.GET.get('exclude', None)
  157. if exclude_pk is not None:
  158. try:
  159. exclude_pk = int(exclude_pk)
  160. except ValueError:
  161. exclude_pk = None
  162. elevation = rack.get_rack_units(face, exclude_pk)
  163. # Enable filtering rack units by ID
  164. q = request.GET.get('q', None)
  165. if q:
  166. elevation = [u for u in elevation if q in str(u['id'])]
  167. page = self.paginate_queryset(elevation)
  168. if page is not None:
  169. rack_units = serializers.RackUnitSerializer(page, many=True, context={'request': request})
  170. return self.get_paginated_response(rack_units.data)
  171. class RackElevationViewSet(ViewSet):
  172. queryset = Rack.objects.prefetch_related(
  173. 'devices'
  174. )
  175. def get_view_name(self):
  176. return "Rack Elevations"
  177. def _add_gradient(self, drawing, id_, color):
  178. gradient = drawing.linearGradient(start=('0', '20%'), end=('0', '40%'), spreadMethod='repeat', id_=id_, gradientTransform='rotate(80)')
  179. gradient.add_stop_color(offset='0%', color='#f7f7f7')
  180. gradient.add_stop_color(offset='50%', color='#f7f7f7')
  181. gradient.add_stop_color(offset='50%', color=color)
  182. gradient.add_stop_color(offset='100%', color=color)
  183. drawing.defs.add(gradient)
  184. def _setup_drawing(self, width, height):
  185. drawing = svgwrite.Drawing(size=(width, height))
  186. # add the stylesheet
  187. drawing.defs.add(drawing.style(constants.RACK_ELEVATION_STYLE))
  188. # add gradients
  189. self._add_gradient(drawing, 'reserved', '#c7c7ff')
  190. self._add_gradient(drawing, 'occupied', '#f0f0f0')
  191. self._add_gradient(drawing, 'blocked', '#ffc7c7')
  192. return drawing
  193. def _draw_device_front(self, drawing, device, start, end, text):
  194. color = device.device_role.color
  195. link = drawing.add(
  196. drawing.a(
  197. reverse('dcim:device', kwargs={'pk': device.pk}), fill='black'
  198. )
  199. )
  200. link.add(drawing.rect(start, end, fill='#{}'.format(color)))
  201. hex_color = '#{}'.format(foreground_color(color))
  202. link.add(drawing.text(device.name, insert=text, fill=hex_color))
  203. def _draw_device_rear(self, drawing, device, start, end, text):
  204. drawing.add(drawing.rect(start, end, class_="blocked"))
  205. drawing.add(drawing.text(device.name, insert=text))
  206. def _draw_empty(self, rack, drawing, start, end, text, id_, face_id, class_):
  207. link = drawing.add(
  208. drawing.a('{}?{}'.format(
  209. reverse('dcim:device_add'),
  210. urlencode({'rack': rack.pk, 'site': rack.site.pk, 'face': face_id, 'position': id_})
  211. ))
  212. )
  213. link.add(drawing.rect(start, end, class_=class_))
  214. link.add(drawing.text("add device", insert=text, class_='add-device'))
  215. def _draw_elevations(self, rack, elevation, reserved, face_id, width, u_height):
  216. drawing = self._setup_drawing(width, u_height * rack.u_height)
  217. i = 0
  218. for u in elevation:
  219. device = u['device']
  220. height = u['height']
  221. start_y = i * u_height
  222. end_y = u_height * height
  223. start = (0, start_y)
  224. end = (width, end_y)
  225. text = (width / 2, start_y + end_y / 2)
  226. if device and device.face == face_id:
  227. self._draw_device_front(drawing, device, start, end, text)
  228. elif device and device.device_type.is_full_depth:
  229. self._draw_device_rear(drawing, device, start, end, text)
  230. else:
  231. class_ = 'slot'
  232. if device:
  233. class_ += ' occupied'
  234. if u["id"] in reserved:
  235. class_ += ' reserved'
  236. self._draw_empty(
  237. rack, drawing, start, end, text, u["id"], face_id, class_
  238. )
  239. i += height
  240. drawing.add(drawing.rect((0, 0), (width, rack.u_height * u_height), class_='rack'))
  241. return drawing
  242. def _get_elevation(self, rack):
  243. elevation = OrderedDict()
  244. for u in rack.units:
  245. elevation[u] = {'id': u, 'device': None, 'height': 1}
  246. for device in Device.objects.prefetch_related('device_role')\
  247. .filter(rack=rack, position__gt=0):
  248. elevation[device.position]['device'] = device
  249. elevation[device.position]['height'] = device.device_type.u_height
  250. for u in range(device.position + 1, device.position + device.device_type.u_height):
  251. elevation.pop(u, None)
  252. return elevation.values()
  253. def retrieve(self, request, pk=None):
  254. """
  255. Render rack
  256. """
  257. rack = get_object_or_404(Rack, pk=pk)
  258. face_id = request.GET.get('face', '0')
  259. if face_id not in ['front', 'rear']:
  260. return HttpResponseBadRequest('face should either be "front" or "rear".')
  261. width = request.GET.get('u_width', '230')
  262. try:
  263. width = int(width)
  264. except ValueError:
  265. return HttpResponseBadRequest('u_width must be numeric.')
  266. u_height = request.GET.get('u_height', '20')
  267. try:
  268. u_height = int(u_height)
  269. except ValueError:
  270. return HttpResponseBadRequest('u_height must be numeric.')
  271. elevation = self._get_elevation(rack)
  272. reserved = rack.get_reserved_units().keys()
  273. drawing = self._draw_elevations(rack, elevation, reserved, face_id, width, u_height)
  274. return HttpResponse(drawing.tostring(), content_type='image/svg+xml')
  275. #
  276. # Rack reservations
  277. #
  278. class RackReservationViewSet(ModelViewSet):
  279. queryset = RackReservation.objects.prefetch_related('rack', 'user', 'tenant')
  280. serializer_class = serializers.RackReservationSerializer
  281. filterset_class = filters.RackReservationFilter
  282. # Assign user from request
  283. def perform_create(self, serializer):
  284. serializer.save(user=self.request.user)
  285. #
  286. # Manufacturers
  287. #
  288. class ManufacturerViewSet(ModelViewSet):
  289. queryset = Manufacturer.objects.annotate(
  290. devicetype_count=get_subquery(DeviceType, 'manufacturer'),
  291. inventoryitem_count=get_subquery(InventoryItem, 'manufacturer'),
  292. platform_count=get_subquery(Platform, 'manufacturer')
  293. )
  294. serializer_class = serializers.ManufacturerSerializer
  295. filterset_class = filters.ManufacturerFilter
  296. #
  297. # Device types
  298. #
  299. class DeviceTypeViewSet(CustomFieldModelViewSet):
  300. queryset = DeviceType.objects.prefetch_related('manufacturer').prefetch_related('tags').annotate(
  301. device_count=Count('instances')
  302. )
  303. serializer_class = serializers.DeviceTypeSerializer
  304. filterset_class = filters.DeviceTypeFilter
  305. #
  306. # Device type components
  307. #
  308. class ConsolePortTemplateViewSet(ModelViewSet):
  309. queryset = ConsolePortTemplate.objects.prefetch_related('device_type__manufacturer')
  310. serializer_class = serializers.ConsolePortTemplateSerializer
  311. filterset_class = filters.ConsolePortTemplateFilter
  312. class ConsoleServerPortTemplateViewSet(ModelViewSet):
  313. queryset = ConsoleServerPortTemplate.objects.prefetch_related('device_type__manufacturer')
  314. serializer_class = serializers.ConsoleServerPortTemplateSerializer
  315. filterset_class = filters.ConsoleServerPortTemplateFilter
  316. class PowerPortTemplateViewSet(ModelViewSet):
  317. queryset = PowerPortTemplate.objects.prefetch_related('device_type__manufacturer')
  318. serializer_class = serializers.PowerPortTemplateSerializer
  319. filterset_class = filters.PowerPortTemplateFilter
  320. class PowerOutletTemplateViewSet(ModelViewSet):
  321. queryset = PowerOutletTemplate.objects.prefetch_related('device_type__manufacturer')
  322. serializer_class = serializers.PowerOutletTemplateSerializer
  323. filterset_class = filters.PowerOutletTemplateFilter
  324. class InterfaceTemplateViewSet(ModelViewSet):
  325. queryset = InterfaceTemplate.objects.prefetch_related('device_type__manufacturer')
  326. serializer_class = serializers.InterfaceTemplateSerializer
  327. filterset_class = filters.InterfaceTemplateFilter
  328. class FrontPortTemplateViewSet(ModelViewSet):
  329. queryset = FrontPortTemplate.objects.prefetch_related('device_type__manufacturer')
  330. serializer_class = serializers.FrontPortTemplateSerializer
  331. filterset_class = filters.FrontPortTemplateFilter
  332. class RearPortTemplateViewSet(ModelViewSet):
  333. queryset = RearPortTemplate.objects.prefetch_related('device_type__manufacturer')
  334. serializer_class = serializers.RearPortTemplateSerializer
  335. filterset_class = filters.RearPortTemplateFilter
  336. class DeviceBayTemplateViewSet(ModelViewSet):
  337. queryset = DeviceBayTemplate.objects.prefetch_related('device_type__manufacturer')
  338. serializer_class = serializers.DeviceBayTemplateSerializer
  339. filterset_class = filters.DeviceBayTemplateFilter
  340. #
  341. # Device roles
  342. #
  343. class DeviceRoleViewSet(ModelViewSet):
  344. queryset = DeviceRole.objects.annotate(
  345. device_count=get_subquery(Device, 'device_role'),
  346. virtualmachine_count=get_subquery(VirtualMachine, 'role')
  347. )
  348. serializer_class = serializers.DeviceRoleSerializer
  349. filterset_class = filters.DeviceRoleFilter
  350. #
  351. # Platforms
  352. #
  353. class PlatformViewSet(ModelViewSet):
  354. queryset = Platform.objects.annotate(
  355. device_count=get_subquery(Device, 'platform'),
  356. virtualmachine_count=get_subquery(VirtualMachine, 'platform')
  357. )
  358. serializer_class = serializers.PlatformSerializer
  359. filterset_class = filters.PlatformFilter
  360. #
  361. # Devices
  362. #
  363. class DeviceViewSet(CustomFieldModelViewSet):
  364. queryset = Device.objects.prefetch_related(
  365. 'device_type__manufacturer', 'device_role', 'tenant', 'platform', 'site', 'rack', 'parent_bay',
  366. 'virtual_chassis__master', 'primary_ip4__nat_outside', 'primary_ip6__nat_outside', 'tags',
  367. )
  368. filterset_class = filters.DeviceFilter
  369. def get_serializer_class(self):
  370. """
  371. Select the specific serializer based on the request context.
  372. If the `brief` query param equates to True, return the NestedDeviceSerializer
  373. If the `exclude` query param includes `config_context` as a value, return the DeviceSerializer
  374. Else, return the DeviceWithConfigContextSerializer
  375. """
  376. request = self.get_serializer_context()['request']
  377. if request.query_params.get('brief', False):
  378. return serializers.NestedDeviceSerializer
  379. elif 'config_context' in request.query_params.get('exclude', []):
  380. return serializers.DeviceSerializer
  381. return serializers.DeviceWithConfigContextSerializer
  382. @action(detail=True)
  383. def graphs(self, request, pk):
  384. """
  385. A convenience method for rendering graphs for a particular Device.
  386. """
  387. device = get_object_or_404(Device, pk=pk)
  388. queryset = Graph.objects.filter(type__model='device')
  389. serializer = RenderedGraphSerializer(queryset, many=True, context={'graphed_object': device})
  390. return Response(serializer.data)
  391. @action(detail=True, url_path='napalm')
  392. def napalm(self, request, pk):
  393. """
  394. Execute a NAPALM method on a Device
  395. """
  396. device = get_object_or_404(Device, pk=pk)
  397. if not device.primary_ip:
  398. raise ServiceUnavailable("This device does not have a primary IP address configured.")
  399. if device.platform is None:
  400. raise ServiceUnavailable("No platform is configured for this device.")
  401. if not device.platform.napalm_driver:
  402. raise ServiceUnavailable("No NAPALM driver is configured for this device's platform ().".format(
  403. device.platform
  404. ))
  405. # Check that NAPALM is installed
  406. try:
  407. import napalm
  408. from napalm.base.exceptions import ModuleImportError
  409. except ImportError:
  410. raise ServiceUnavailable("NAPALM is not installed. Please see the documentation for instructions.")
  411. # Validate the configured driver
  412. try:
  413. driver = napalm.get_network_driver(device.platform.napalm_driver)
  414. except ModuleImportError:
  415. raise ServiceUnavailable("NAPALM driver for platform {} not found: {}.".format(
  416. device.platform, device.platform.napalm_driver
  417. ))
  418. # Verify user permission
  419. if not request.user.has_perm('dcim.napalm_read'):
  420. return HttpResponseForbidden()
  421. # Connect to the device
  422. napalm_methods = request.GET.getlist('method')
  423. response = OrderedDict([(m, None) for m in napalm_methods])
  424. ip_address = str(device.primary_ip.address.ip)
  425. optional_args = settings.NAPALM_ARGS.copy()
  426. if device.platform.napalm_args is not None:
  427. optional_args.update(device.platform.napalm_args)
  428. d = driver(
  429. hostname=ip_address,
  430. username=settings.NAPALM_USERNAME,
  431. password=settings.NAPALM_PASSWORD,
  432. timeout=settings.NAPALM_TIMEOUT,
  433. optional_args=optional_args
  434. )
  435. try:
  436. d.open()
  437. except Exception as e:
  438. raise ServiceUnavailable("Error connecting to the device at {}: {}".format(ip_address, e))
  439. # Validate and execute each specified NAPALM method
  440. for method in napalm_methods:
  441. if not hasattr(driver, method):
  442. response[method] = {'error': 'Unknown NAPALM method'}
  443. continue
  444. if not method.startswith('get_'):
  445. response[method] = {'error': 'Only get_* NAPALM methods are supported'}
  446. continue
  447. try:
  448. response[method] = getattr(d, method)()
  449. except NotImplementedError:
  450. response[method] = {'error': 'Method {} not implemented for NAPALM driver {}'.format(method, driver)}
  451. except Exception as e:
  452. response[method] = {'error': 'Method {} failed: {}'.format(method, e)}
  453. d.close()
  454. return Response(response)
  455. #
  456. # Device components
  457. #
  458. class ConsolePortViewSet(CableTraceMixin, ModelViewSet):
  459. queryset = ConsolePort.objects.prefetch_related('device', 'connected_endpoint__device', 'cable', 'tags')
  460. serializer_class = serializers.ConsolePortSerializer
  461. filterset_class = filters.ConsolePortFilter
  462. class ConsoleServerPortViewSet(CableTraceMixin, ModelViewSet):
  463. queryset = ConsoleServerPort.objects.prefetch_related('device', 'connected_endpoint__device', 'cable', 'tags')
  464. serializer_class = serializers.ConsoleServerPortSerializer
  465. filterset_class = filters.ConsoleServerPortFilter
  466. class PowerPortViewSet(CableTraceMixin, ModelViewSet):
  467. queryset = PowerPort.objects.prefetch_related(
  468. 'device', '_connected_poweroutlet__device', '_connected_powerfeed', 'cable', 'tags'
  469. )
  470. serializer_class = serializers.PowerPortSerializer
  471. filterset_class = filters.PowerPortFilter
  472. class PowerOutletViewSet(CableTraceMixin, ModelViewSet):
  473. queryset = PowerOutlet.objects.prefetch_related('device', 'connected_endpoint__device', 'cable', 'tags')
  474. serializer_class = serializers.PowerOutletSerializer
  475. filterset_class = filters.PowerOutletFilter
  476. class InterfaceViewSet(CableTraceMixin, ModelViewSet):
  477. queryset = Interface.objects.prefetch_related(
  478. 'device', '_connected_interface', '_connected_circuittermination', 'cable', 'ip_addresses', 'tags'
  479. ).filter(
  480. device__isnull=False
  481. )
  482. serializer_class = serializers.InterfaceSerializer
  483. filterset_class = filters.InterfaceFilter
  484. @action(detail=True)
  485. def graphs(self, request, pk):
  486. """
  487. A convenience method for rendering graphs for a particular interface.
  488. """
  489. interface = get_object_or_404(Interface, pk=pk)
  490. queryset = Graph.objects.filter(type__model='interface')
  491. serializer = RenderedGraphSerializer(queryset, many=True, context={'graphed_object': interface})
  492. return Response(serializer.data)
  493. class FrontPortViewSet(ModelViewSet):
  494. queryset = FrontPort.objects.prefetch_related('device__device_type__manufacturer', 'rear_port', 'cable', 'tags')
  495. serializer_class = serializers.FrontPortSerializer
  496. filterset_class = filters.FrontPortFilter
  497. class RearPortViewSet(ModelViewSet):
  498. queryset = RearPort.objects.prefetch_related('device__device_type__manufacturer', 'cable', 'tags')
  499. serializer_class = serializers.RearPortSerializer
  500. filterset_class = filters.RearPortFilter
  501. class DeviceBayViewSet(ModelViewSet):
  502. queryset = DeviceBay.objects.prefetch_related('installed_device').prefetch_related('tags')
  503. serializer_class = serializers.DeviceBaySerializer
  504. filterset_class = filters.DeviceBayFilter
  505. class InventoryItemViewSet(ModelViewSet):
  506. queryset = InventoryItem.objects.prefetch_related('device', 'manufacturer').prefetch_related('tags')
  507. serializer_class = serializers.InventoryItemSerializer
  508. filterset_class = filters.InventoryItemFilter
  509. #
  510. # Connections
  511. #
  512. class ConsoleConnectionViewSet(ListModelMixin, GenericViewSet):
  513. queryset = ConsolePort.objects.prefetch_related(
  514. 'device', 'connected_endpoint__device'
  515. ).filter(
  516. connected_endpoint__isnull=False
  517. )
  518. serializer_class = serializers.ConsolePortSerializer
  519. filterset_class = filters.ConsoleConnectionFilter
  520. class PowerConnectionViewSet(ListModelMixin, GenericViewSet):
  521. queryset = PowerPort.objects.prefetch_related(
  522. 'device', 'connected_endpoint__device'
  523. ).filter(
  524. _connected_poweroutlet__isnull=False
  525. )
  526. serializer_class = serializers.PowerPortSerializer
  527. filterset_class = filters.PowerConnectionFilter
  528. class InterfaceConnectionViewSet(ListModelMixin, GenericViewSet):
  529. queryset = Interface.objects.prefetch_related(
  530. 'device', '_connected_interface__device'
  531. ).filter(
  532. # Avoid duplicate connections by only selecting the lower PK in a connected pair
  533. _connected_interface__isnull=False,
  534. pk__lt=F('_connected_interface')
  535. )
  536. serializer_class = serializers.InterfaceConnectionSerializer
  537. filterset_class = filters.InterfaceConnectionFilter
  538. #
  539. # Cables
  540. #
  541. class CableViewSet(ModelViewSet):
  542. queryset = Cable.objects.prefetch_related(
  543. 'termination_a', 'termination_b'
  544. )
  545. serializer_class = serializers.CableSerializer
  546. filterset_class = filters.CableFilter
  547. #
  548. # Virtual chassis
  549. #
  550. class VirtualChassisViewSet(ModelViewSet):
  551. queryset = VirtualChassis.objects.prefetch_related('tags').annotate(
  552. member_count=Count('members')
  553. )
  554. serializer_class = serializers.VirtualChassisSerializer
  555. filterset_class = filters.VirtualChassisFilter
  556. #
  557. # Power panels
  558. #
  559. class PowerPanelViewSet(ModelViewSet):
  560. queryset = PowerPanel.objects.prefetch_related(
  561. 'site', 'rack_group'
  562. ).annotate(
  563. powerfeed_count=Count('powerfeeds')
  564. )
  565. serializer_class = serializers.PowerPanelSerializer
  566. filterset_class = filters.PowerPanelFilter
  567. #
  568. # Power feeds
  569. #
  570. class PowerFeedViewSet(CustomFieldModelViewSet):
  571. queryset = PowerFeed.objects.prefetch_related('power_panel', 'rack', 'tags')
  572. serializer_class = serializers.PowerFeedSerializer
  573. filterset_class = filters.PowerFeedFilter
  574. #
  575. # Miscellaneous
  576. #
  577. class ConnectedDeviceViewSet(ViewSet):
  578. """
  579. This endpoint allows a user to determine what device (if any) is connected to a given peer device and peer
  580. interface. This is useful in a situation where a device boots with no configuration, but can detect its neighbors
  581. via a protocol such as LLDP. Two query parameters must be included in the request:
  582. * `peer_device`: The name of the peer device
  583. * `peer_interface`: The name of the peer interface
  584. """
  585. permission_classes = [IsAuthenticatedOrLoginNotRequired]
  586. _device_param = Parameter(
  587. name='peer_device',
  588. in_='query',
  589. description='The name of the peer device',
  590. required=True,
  591. type=openapi.TYPE_STRING
  592. )
  593. _interface_param = Parameter(
  594. name='peer_interface',
  595. in_='query',
  596. description='The name of the peer interface',
  597. required=True,
  598. type=openapi.TYPE_STRING
  599. )
  600. def get_view_name(self):
  601. return "Connected Device Locator"
  602. @swagger_auto_schema(
  603. manual_parameters=[_device_param, _interface_param],
  604. responses={'200': serializers.DeviceSerializer}
  605. )
  606. def list(self, request):
  607. peer_device_name = request.query_params.get(self._device_param.name)
  608. peer_interface_name = request.query_params.get(self._interface_param.name)
  609. if not peer_device_name or not peer_interface_name:
  610. raise MissingFilterException(detail='Request must include "peer_device" and "peer_interface" filters.')
  611. # Determine local interface from peer interface's connection
  612. peer_interface = get_object_or_404(Interface, device__name=peer_device_name, name=peer_interface_name)
  613. local_interface = peer_interface._connected_interface
  614. if local_interface is None:
  615. return Response()
  616. return Response(serializers.DeviceSerializer(local_interface.device, context={'request': request}).data)