views.py 59 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733
  1. from __future__ import unicode_literals
  2. from copy import deepcopy
  3. import re
  4. from natsort import natsorted
  5. from operator import attrgetter
  6. from django.contrib import messages
  7. from django.contrib.auth.decorators import permission_required
  8. from django.contrib.auth.mixins import PermissionRequiredMixin
  9. from django.core.paginator import EmptyPage, PageNotAnInteger
  10. from django.db.models import Count
  11. from django.http import HttpResponseRedirect
  12. from django.shortcuts import get_object_or_404, redirect, render
  13. from django.urls import reverse
  14. from django.utils.html import escape
  15. from django.utils.http import urlencode
  16. from django.utils.safestring import mark_safe
  17. from django.views.generic import View
  18. from ipam.models import Prefix, Service, VLAN
  19. from circuits.models import Circuit
  20. from extras.models import Graph, TopologyMap, GRAPH_TYPE_INTERFACE, GRAPH_TYPE_SITE, UserAction
  21. from utilities.forms import ConfirmationForm
  22. from utilities.paginator import EnhancedPaginator
  23. from utilities.views import (
  24. BulkDeleteView, BulkEditView, BulkImportView, ObjectDeleteView, ObjectEditView, ObjectListView,
  25. )
  26. from . import filters, forms, tables
  27. from .models import (
  28. CONNECTION_STATUS_CONNECTED, ConsolePort, ConsolePortTemplate, ConsoleServerPort, ConsoleServerPortTemplate, Device,
  29. DeviceBay, DeviceBayTemplate, DeviceRole, DeviceType, Interface, InterfaceConnection, InterfaceTemplate,
  30. Manufacturer, InventoryItem, Platform, PowerOutlet, PowerOutletTemplate, PowerPort, PowerPortTemplate, Rack, RackGroup,
  31. RackReservation, RackRole, Region, Site,
  32. )
  33. EXPANSION_PATTERN = '\[(\d+-\d+)\]'
  34. def xstr(s):
  35. """
  36. Replace None with an empty string (for CSV export)
  37. """
  38. return '' if s is None else str(s)
  39. def expand_pattern(string):
  40. """
  41. Expand a numeric pattern into a list of strings. Examples:
  42. 'ge-0/0/[0-3]' => ['ge-0/0/0', 'ge-0/0/1', 'ge-0/0/2', 'ge-0/0/3']
  43. 'xe-0/[0-3]/[0-7]' => ['xe-0/0/0', 'xe-0/0/1', 'xe-0/0/2', ... 'xe-0/3/5', 'xe-0/3/6', 'xe-0/3/7']
  44. """
  45. lead, pattern, remnant = re.split(EXPANSION_PATTERN, string, maxsplit=1)
  46. x, y = pattern.split('-')
  47. for i in range(int(x), int(y) + 1):
  48. if remnant:
  49. for string in expand_pattern(remnant):
  50. yield "{0}{1}{2}".format(lead, i, string)
  51. else:
  52. yield "{0}{1}".format(lead, i)
  53. class ComponentCreateView(View):
  54. parent_model = None
  55. parent_field = None
  56. model = None
  57. form = None
  58. model_form = None
  59. def get(self, request, pk):
  60. parent = get_object_or_404(self.parent_model, pk=pk)
  61. form = self.form(parent, initial=request.GET)
  62. return render(request, 'dcim/device_component_add.html', {
  63. 'parent': parent,
  64. 'component_type': self.model._meta.verbose_name,
  65. 'form': form,
  66. 'return_url': parent.get_absolute_url(),
  67. })
  68. def post(self, request, pk):
  69. parent = get_object_or_404(self.parent_model, pk=pk)
  70. form = self.form(parent, request.POST)
  71. if form.is_valid():
  72. new_components = []
  73. data = deepcopy(form.cleaned_data)
  74. for name in form.cleaned_data['name_pattern']:
  75. component_data = {
  76. self.parent_field: parent.pk,
  77. 'name': name,
  78. }
  79. # Replace objects with their primary key to keep component_form.clean() happy
  80. for k, v in data.items():
  81. if hasattr(v, 'pk'):
  82. component_data[k] = v.pk
  83. else:
  84. component_data[k] = v
  85. component_form = self.model_form(component_data)
  86. if component_form.is_valid():
  87. new_components.append(component_form.save(commit=False))
  88. else:
  89. for field, errors in component_form.errors.as_data().items():
  90. # Assign errors on the child form's name field to name_pattern on the parent form
  91. if field == 'name':
  92. field = 'name_pattern'
  93. for e in errors:
  94. form.add_error(field, '{}: {}'.format(name, ', '.join(e)))
  95. if not form.errors:
  96. self.model.objects.bulk_create(new_components)
  97. messages.success(request, "Added {} {} to {}.".format(
  98. len(new_components), self.model._meta.verbose_name_plural, parent
  99. ))
  100. if '_addanother' in request.POST:
  101. return redirect(request.path)
  102. else:
  103. return redirect(parent.get_absolute_url())
  104. return render(request, 'dcim/device_component_add.html', {
  105. 'parent': parent,
  106. 'component_type': self.model._meta.verbose_name,
  107. 'form': form,
  108. 'return_url': parent.get_absolute_url(),
  109. })
  110. class ComponentEditView(ObjectEditView):
  111. def get_return_url(self, request, obj):
  112. return obj.device.get_absolute_url()
  113. class ComponentDeleteView(ObjectDeleteView):
  114. def get_return_url(self, request, obj):
  115. return obj.device.get_absolute_url()
  116. #
  117. # Regions
  118. #
  119. class RegionListView(ObjectListView):
  120. queryset = Region.objects.annotate(site_count=Count('sites'))
  121. table = tables.RegionTable
  122. template_name = 'dcim/region_list.html'
  123. class RegionEditView(PermissionRequiredMixin, ObjectEditView):
  124. permission_required = 'dcim.change_region'
  125. model = Region
  126. form_class = forms.RegionForm
  127. def get_return_url(self, request, obj):
  128. return reverse('dcim:region_list')
  129. class RegionBulkDeleteView(PermissionRequiredMixin, BulkDeleteView):
  130. permission_required = 'dcim.delete_region'
  131. cls = Region
  132. default_return_url = 'dcim:region_list'
  133. #
  134. # Sites
  135. #
  136. class SiteListView(ObjectListView):
  137. queryset = Site.objects.select_related('region', 'tenant')
  138. filter = filters.SiteFilter
  139. filter_form = forms.SiteFilterForm
  140. table = tables.SiteTable
  141. template_name = 'dcim/site_list.html'
  142. class SiteView(View):
  143. def get(self, request, slug):
  144. site = get_object_or_404(Site.objects.select_related('region', 'tenant__group'), slug=slug)
  145. stats = {
  146. 'rack_count': Rack.objects.filter(site=site).count(),
  147. 'device_count': Device.objects.filter(site=site).count(),
  148. 'prefix_count': Prefix.objects.filter(site=site).count(),
  149. 'vlan_count': VLAN.objects.filter(site=site).count(),
  150. 'circuit_count': Circuit.objects.filter(terminations__site=site).count(),
  151. }
  152. rack_groups = RackGroup.objects.filter(site=site).annotate(rack_count=Count('racks'))
  153. topology_maps = TopologyMap.objects.filter(site=site)
  154. show_graphs = Graph.objects.filter(type=GRAPH_TYPE_SITE).exists()
  155. return render(request, 'dcim/site.html', {
  156. 'site': site,
  157. 'stats': stats,
  158. 'rack_groups': rack_groups,
  159. 'topology_maps': topology_maps,
  160. 'show_graphs': show_graphs,
  161. })
  162. class SiteEditView(PermissionRequiredMixin, ObjectEditView):
  163. permission_required = 'dcim.change_site'
  164. model = Site
  165. form_class = forms.SiteForm
  166. template_name = 'dcim/site_edit.html'
  167. default_return_url = 'dcim:site_list'
  168. class SiteDeleteView(PermissionRequiredMixin, ObjectDeleteView):
  169. permission_required = 'dcim.delete_site'
  170. model = Site
  171. default_return_url = 'dcim:site_list'
  172. class SiteBulkImportView(PermissionRequiredMixin, BulkImportView):
  173. permission_required = 'dcim.add_site'
  174. form = forms.SiteImportForm
  175. table = tables.SiteTable
  176. template_name = 'dcim/site_import.html'
  177. default_return_url = 'dcim:site_list'
  178. class SiteBulkEditView(PermissionRequiredMixin, BulkEditView):
  179. permission_required = 'dcim.change_site'
  180. cls = Site
  181. filter = filters.SiteFilter
  182. form = forms.SiteBulkEditForm
  183. template_name = 'dcim/site_bulk_edit.html'
  184. default_return_url = 'dcim:site_list'
  185. #
  186. # Rack groups
  187. #
  188. class RackGroupListView(ObjectListView):
  189. queryset = RackGroup.objects.select_related('site').annotate(rack_count=Count('racks'))
  190. filter = filters.RackGroupFilter
  191. filter_form = forms.RackGroupFilterForm
  192. table = tables.RackGroupTable
  193. template_name = 'dcim/rackgroup_list.html'
  194. class RackGroupEditView(PermissionRequiredMixin, ObjectEditView):
  195. permission_required = 'dcim.change_rackgroup'
  196. model = RackGroup
  197. form_class = forms.RackGroupForm
  198. def get_return_url(self, request, obj):
  199. return reverse('dcim:rackgroup_list')
  200. class RackGroupBulkDeleteView(PermissionRequiredMixin, BulkDeleteView):
  201. permission_required = 'dcim.delete_rackgroup'
  202. cls = RackGroup
  203. filter = filters.RackGroupFilter
  204. default_return_url = 'dcim:rackgroup_list'
  205. #
  206. # Rack roles
  207. #
  208. class RackRoleListView(ObjectListView):
  209. queryset = RackRole.objects.annotate(rack_count=Count('racks'))
  210. table = tables.RackRoleTable
  211. template_name = 'dcim/rackrole_list.html'
  212. class RackRoleEditView(PermissionRequiredMixin, ObjectEditView):
  213. permission_required = 'dcim.change_rackrole'
  214. model = RackRole
  215. form_class = forms.RackRoleForm
  216. def get_return_url(self, request, obj):
  217. return reverse('dcim:rackrole_list')
  218. class RackRoleBulkDeleteView(PermissionRequiredMixin, BulkDeleteView):
  219. permission_required = 'dcim.delete_rackrole'
  220. cls = RackRole
  221. default_return_url = 'dcim:rackrole_list'
  222. #
  223. # Racks
  224. #
  225. class RackListView(ObjectListView):
  226. queryset = Rack.objects.select_related(
  227. 'site', 'group', 'tenant', 'role'
  228. ).prefetch_related(
  229. 'devices__device_type'
  230. ).annotate(
  231. device_count=Count('devices', distinct=True)
  232. )
  233. filter = filters.RackFilter
  234. filter_form = forms.RackFilterForm
  235. table = tables.RackTable
  236. template_name = 'dcim/rack_list.html'
  237. class RackElevationListView(View):
  238. """
  239. Display a set of rack elevations side-by-side.
  240. """
  241. def get(self, request):
  242. racks = Rack.objects.select_related(
  243. 'site', 'group', 'tenant', 'role'
  244. ).prefetch_related(
  245. 'devices__device_type'
  246. )
  247. racks = filters.RackFilter(request.GET, racks).qs
  248. total_count = racks.count()
  249. # Pagination
  250. paginator = EnhancedPaginator(racks, 25)
  251. page_number = request.GET.get('page', 1)
  252. try:
  253. page = paginator.page(page_number)
  254. except PageNotAnInteger:
  255. page = paginator.page(1)
  256. except EmptyPage:
  257. page = paginator.page(paginator.num_pages)
  258. # Determine rack face
  259. if request.GET.get('face') == '1':
  260. face_id = 1
  261. else:
  262. face_id = 0
  263. return render(request, 'dcim/rack_elevation_list.html', {
  264. 'paginator': paginator,
  265. 'page': page,
  266. 'total_count': total_count,
  267. 'face_id': face_id,
  268. 'filter_form': forms.RackFilterForm(request.GET),
  269. })
  270. class RackView(View):
  271. def get(self, request, pk):
  272. rack = get_object_or_404(Rack.objects.select_related('site__region', 'tenant__group', 'group', 'role'), pk=pk)
  273. nonracked_devices = Device.objects.filter(rack=rack, position__isnull=True, parent_bay__isnull=True)\
  274. .select_related('device_type__manufacturer')
  275. next_rack = Rack.objects.filter(site=rack.site, name__gt=rack.name).order_by('name').first()
  276. prev_rack = Rack.objects.filter(site=rack.site, name__lt=rack.name).order_by('-name').first()
  277. reservations = RackReservation.objects.filter(rack=rack)
  278. reserved_units = {}
  279. for r in reservations:
  280. for u in r.units:
  281. reserved_units[u] = r
  282. return render(request, 'dcim/rack.html', {
  283. 'rack': rack,
  284. 'reservations': reservations,
  285. 'reserved_units': reserved_units,
  286. 'nonracked_devices': nonracked_devices,
  287. 'next_rack': next_rack,
  288. 'prev_rack': prev_rack,
  289. 'front_elevation': rack.get_front_elevation(),
  290. 'rear_elevation': rack.get_rear_elevation(),
  291. })
  292. class RackEditView(PermissionRequiredMixin, ObjectEditView):
  293. permission_required = 'dcim.change_rack'
  294. model = Rack
  295. form_class = forms.RackForm
  296. template_name = 'dcim/rack_edit.html'
  297. default_return_url = 'dcim:rack_list'
  298. class RackDeleteView(PermissionRequiredMixin, ObjectDeleteView):
  299. permission_required = 'dcim.delete_rack'
  300. model = Rack
  301. default_return_url = 'dcim:rack_list'
  302. class RackBulkImportView(PermissionRequiredMixin, BulkImportView):
  303. permission_required = 'dcim.add_rack'
  304. form = forms.RackImportForm
  305. table = tables.RackImportTable
  306. template_name = 'dcim/rack_import.html'
  307. default_return_url = 'dcim:rack_list'
  308. class RackBulkEditView(PermissionRequiredMixin, BulkEditView):
  309. permission_required = 'dcim.change_rack'
  310. cls = Rack
  311. filter = filters.RackFilter
  312. form = forms.RackBulkEditForm
  313. template_name = 'dcim/rack_bulk_edit.html'
  314. default_return_url = 'dcim:rack_list'
  315. class RackBulkDeleteView(PermissionRequiredMixin, BulkDeleteView):
  316. permission_required = 'dcim.delete_rack'
  317. cls = Rack
  318. filter = filters.RackFilter
  319. default_return_url = 'dcim:rack_list'
  320. #
  321. # Rack reservations
  322. #
  323. class RackReservationListView(ObjectListView):
  324. queryset = RackReservation.objects.all()
  325. filter = filters.RackReservationFilter
  326. filter_form = forms.RackReservationFilterForm
  327. table = tables.RackReservationTable
  328. template_name = 'dcim/rackreservation_list.html'
  329. class RackReservationEditView(PermissionRequiredMixin, ObjectEditView):
  330. permission_required = 'dcim.change_rackreservation'
  331. model = RackReservation
  332. form_class = forms.RackReservationForm
  333. def alter_obj(self, obj, request, args, kwargs):
  334. if not obj.pk:
  335. obj.rack = get_object_or_404(Rack, pk=kwargs['rack'])
  336. obj.user = request.user
  337. return obj
  338. def get_return_url(self, request, obj):
  339. return obj.rack.get_absolute_url()
  340. class RackReservationDeleteView(PermissionRequiredMixin, ObjectDeleteView):
  341. permission_required = 'dcim.delete_rackreservation'
  342. model = RackReservation
  343. def get_return_url(self, request, obj):
  344. return obj.rack.get_absolute_url()
  345. class RackReservationBulkDeleteView(PermissionRequiredMixin, BulkDeleteView):
  346. permission_required = 'dcim.delete_rackreservation'
  347. cls = RackReservation
  348. default_return_url = 'dcim:rackreservation_list'
  349. #
  350. # Manufacturers
  351. #
  352. class ManufacturerListView(ObjectListView):
  353. queryset = Manufacturer.objects.annotate(devicetype_count=Count('device_types'))
  354. table = tables.ManufacturerTable
  355. template_name = 'dcim/manufacturer_list.html'
  356. class ManufacturerEditView(PermissionRequiredMixin, ObjectEditView):
  357. permission_required = 'dcim.change_manufacturer'
  358. model = Manufacturer
  359. form_class = forms.ManufacturerForm
  360. def get_return_url(self, request, obj):
  361. return reverse('dcim:manufacturer_list')
  362. class ManufacturerBulkDeleteView(PermissionRequiredMixin, BulkDeleteView):
  363. permission_required = 'dcim.delete_manufacturer'
  364. cls = Manufacturer
  365. default_return_url = 'dcim:manufacturer_list'
  366. #
  367. # Device types
  368. #
  369. class DeviceTypeListView(ObjectListView):
  370. queryset = DeviceType.objects.select_related('manufacturer').annotate(instance_count=Count('instances'))
  371. filter = filters.DeviceTypeFilter
  372. filter_form = forms.DeviceTypeFilterForm
  373. table = tables.DeviceTypeTable
  374. template_name = 'dcim/devicetype_list.html'
  375. class DeviceTypeView(View):
  376. def get(self, request, pk):
  377. devicetype = get_object_or_404(DeviceType, pk=pk)
  378. # Component tables
  379. consoleport_table = tables.ConsolePortTemplateTable(
  380. natsorted(ConsolePortTemplate.objects.filter(device_type=devicetype), key=attrgetter('name'))
  381. )
  382. consoleserverport_table = tables.ConsoleServerPortTemplateTable(
  383. natsorted(ConsoleServerPortTemplate.objects.filter(device_type=devicetype), key=attrgetter('name'))
  384. )
  385. powerport_table = tables.PowerPortTemplateTable(
  386. natsorted(PowerPortTemplate.objects.filter(device_type=devicetype), key=attrgetter('name'))
  387. )
  388. poweroutlet_table = tables.PowerOutletTemplateTable(
  389. natsorted(PowerOutletTemplate.objects.filter(device_type=devicetype), key=attrgetter('name'))
  390. )
  391. mgmt_interface_table = tables.InterfaceTemplateTable(
  392. list(InterfaceTemplate.objects.order_naturally(devicetype.interface_ordering).filter(
  393. device_type=devicetype, mgmt_only=True
  394. ))
  395. )
  396. interface_table = tables.InterfaceTemplateTable(
  397. list(InterfaceTemplate.objects.order_naturally(devicetype.interface_ordering).filter(
  398. device_type=devicetype, mgmt_only=False
  399. ))
  400. )
  401. devicebay_table = tables.DeviceBayTemplateTable(
  402. natsorted(DeviceBayTemplate.objects.filter(device_type=devicetype), key=attrgetter('name'))
  403. )
  404. if request.user.has_perm('dcim.change_devicetype'):
  405. consoleport_table.base_columns['pk'].visible = True
  406. consoleserverport_table.base_columns['pk'].visible = True
  407. powerport_table.base_columns['pk'].visible = True
  408. poweroutlet_table.base_columns['pk'].visible = True
  409. mgmt_interface_table.base_columns['pk'].visible = True
  410. interface_table.base_columns['pk'].visible = True
  411. devicebay_table.base_columns['pk'].visible = True
  412. return render(request, 'dcim/devicetype.html', {
  413. 'devicetype': devicetype,
  414. 'consoleport_table': consoleport_table,
  415. 'consoleserverport_table': consoleserverport_table,
  416. 'powerport_table': powerport_table,
  417. 'poweroutlet_table': poweroutlet_table,
  418. 'mgmt_interface_table': mgmt_interface_table,
  419. 'interface_table': interface_table,
  420. 'devicebay_table': devicebay_table,
  421. })
  422. class DeviceTypeEditView(PermissionRequiredMixin, ObjectEditView):
  423. permission_required = 'dcim.change_devicetype'
  424. model = DeviceType
  425. form_class = forms.DeviceTypeForm
  426. template_name = 'dcim/devicetype_edit.html'
  427. default_return_url = 'dcim:devicetype_list'
  428. class DeviceTypeDeleteView(PermissionRequiredMixin, ObjectDeleteView):
  429. permission_required = 'dcim.delete_devicetype'
  430. model = DeviceType
  431. default_return_url = 'dcim:devicetype_list'
  432. class DeviceTypeBulkEditView(PermissionRequiredMixin, BulkEditView):
  433. permission_required = 'dcim.change_devicetype'
  434. cls = DeviceType
  435. filter = filters.DeviceTypeFilter
  436. form = forms.DeviceTypeBulkEditForm
  437. template_name = 'dcim/devicetype_bulk_edit.html'
  438. default_return_url = 'dcim:devicetype_list'
  439. class DeviceTypeBulkDeleteView(PermissionRequiredMixin, BulkDeleteView):
  440. permission_required = 'dcim.delete_devicetype'
  441. cls = DeviceType
  442. filter = filters.DeviceTypeFilter
  443. default_return_url = 'dcim:devicetype_list'
  444. #
  445. # Device type components
  446. #
  447. class ConsolePortTemplateAddView(PermissionRequiredMixin, ComponentCreateView):
  448. permission_required = 'dcim.add_consoleporttemplate'
  449. parent_model = DeviceType
  450. parent_field = 'device_type'
  451. model = ConsolePortTemplate
  452. form = forms.ConsolePortTemplateCreateForm
  453. model_form = forms.ConsolePortTemplateForm
  454. class ConsolePortTemplateBulkDeleteView(PermissionRequiredMixin, BulkDeleteView):
  455. permission_required = 'dcim.delete_consoleporttemplate'
  456. parent_model = DeviceType
  457. parent_field = 'device_type'
  458. cls = ConsolePortTemplate
  459. parent_cls = DeviceType
  460. class ConsoleServerPortTemplateAddView(PermissionRequiredMixin, ComponentCreateView):
  461. permission_required = 'dcim.add_consoleserverporttemplate'
  462. parent_model = DeviceType
  463. parent_field = 'device_type'
  464. model = ConsoleServerPortTemplate
  465. form = forms.ConsoleServerPortTemplateCreateForm
  466. model_form = forms.ConsoleServerPortTemplateForm
  467. class ConsoleServerPortTemplateBulkDeleteView(PermissionRequiredMixin, BulkDeleteView):
  468. permission_required = 'dcim.delete_consoleserverporttemplate'
  469. cls = ConsoleServerPortTemplate
  470. parent_cls = DeviceType
  471. class PowerPortTemplateAddView(PermissionRequiredMixin, ComponentCreateView):
  472. permission_required = 'dcim.add_powerporttemplate'
  473. parent_model = DeviceType
  474. parent_field = 'device_type'
  475. model = PowerPortTemplate
  476. form = forms.PowerPortTemplateCreateForm
  477. model_form = forms.PowerPortTemplateForm
  478. class PowerPortTemplateBulkDeleteView(PermissionRequiredMixin, BulkDeleteView):
  479. permission_required = 'dcim.delete_powerporttemplate'
  480. cls = PowerPortTemplate
  481. parent_cls = DeviceType
  482. class PowerOutletTemplateAddView(PermissionRequiredMixin, ComponentCreateView):
  483. permission_required = 'dcim.add_poweroutlettemplate'
  484. parent_model = DeviceType
  485. parent_field = 'device_type'
  486. model = PowerOutletTemplate
  487. form = forms.PowerOutletTemplateCreateForm
  488. model_form = forms.PowerOutletTemplateForm
  489. class PowerOutletTemplateBulkDeleteView(PermissionRequiredMixin, BulkDeleteView):
  490. permission_required = 'dcim.delete_poweroutlettemplate'
  491. cls = PowerOutletTemplate
  492. parent_cls = DeviceType
  493. class InterfaceTemplateAddView(PermissionRequiredMixin, ComponentCreateView):
  494. permission_required = 'dcim.add_interfacetemplate'
  495. parent_model = DeviceType
  496. parent_field = 'device_type'
  497. model = InterfaceTemplate
  498. form = forms.InterfaceTemplateCreateForm
  499. model_form = forms.InterfaceTemplateForm
  500. class InterfaceTemplateBulkEditView(PermissionRequiredMixin, BulkEditView):
  501. permission_required = 'dcim.change_interfacetemplate'
  502. cls = InterfaceTemplate
  503. parent_cls = DeviceType
  504. form = forms.InterfaceTemplateBulkEditForm
  505. template_name = 'dcim/interfacetemplate_bulk_edit.html'
  506. class InterfaceTemplateBulkDeleteView(PermissionRequiredMixin, BulkDeleteView):
  507. permission_required = 'dcim.delete_interfacetemplate'
  508. cls = InterfaceTemplate
  509. parent_cls = DeviceType
  510. class DeviceBayTemplateAddView(PermissionRequiredMixin, ComponentCreateView):
  511. permission_required = 'dcim.add_devicebaytemplate'
  512. parent_model = DeviceType
  513. parent_field = 'device_type'
  514. model = DeviceBayTemplate
  515. form = forms.DeviceBayTemplateCreateForm
  516. model_form = forms.DeviceBayTemplateForm
  517. class DeviceBayTemplateBulkDeleteView(PermissionRequiredMixin, BulkDeleteView):
  518. permission_required = 'dcim.delete_devicebaytemplate'
  519. cls = DeviceBayTemplate
  520. parent_cls = DeviceType
  521. #
  522. # Device roles
  523. #
  524. class DeviceRoleListView(ObjectListView):
  525. queryset = DeviceRole.objects.annotate(device_count=Count('devices'))
  526. table = tables.DeviceRoleTable
  527. template_name = 'dcim/devicerole_list.html'
  528. class DeviceRoleEditView(PermissionRequiredMixin, ObjectEditView):
  529. permission_required = 'dcim.change_devicerole'
  530. model = DeviceRole
  531. form_class = forms.DeviceRoleForm
  532. def get_return_url(self, request, obj):
  533. return reverse('dcim:devicerole_list')
  534. class DeviceRoleBulkDeleteView(PermissionRequiredMixin, BulkDeleteView):
  535. permission_required = 'dcim.delete_devicerole'
  536. cls = DeviceRole
  537. default_return_url = 'dcim:devicerole_list'
  538. #
  539. # Platforms
  540. #
  541. class PlatformListView(ObjectListView):
  542. queryset = Platform.objects.annotate(device_count=Count('devices'))
  543. table = tables.PlatformTable
  544. template_name = 'dcim/platform_list.html'
  545. class PlatformEditView(PermissionRequiredMixin, ObjectEditView):
  546. permission_required = 'dcim.change_platform'
  547. model = Platform
  548. form_class = forms.PlatformForm
  549. def get_return_url(self, request, obj):
  550. return reverse('dcim:platform_list')
  551. class PlatformBulkDeleteView(PermissionRequiredMixin, BulkDeleteView):
  552. permission_required = 'dcim.delete_platform'
  553. cls = Platform
  554. default_return_url = 'dcim:platform_list'
  555. #
  556. # Devices
  557. #
  558. class DeviceListView(ObjectListView):
  559. queryset = Device.objects.select_related('device_type__manufacturer', 'device_role', 'tenant', 'site', 'rack',
  560. 'primary_ip4', 'primary_ip6')
  561. filter = filters.DeviceFilter
  562. filter_form = forms.DeviceFilterForm
  563. table = tables.DeviceTable
  564. template_name = 'dcim/device_list.html'
  565. class DeviceView(View):
  566. def get(self, request, pk):
  567. device = get_object_or_404(Device.objects.select_related(
  568. 'site__region', 'rack__group', 'tenant__group', 'device_role', 'platform'
  569. ), pk=pk)
  570. console_ports = natsorted(
  571. ConsolePort.objects.filter(device=device).select_related('cs_port__device'), key=attrgetter('name')
  572. )
  573. cs_ports = natsorted(
  574. ConsoleServerPort.objects.filter(device=device).select_related('connected_console'), key=attrgetter('name')
  575. )
  576. power_ports = natsorted(
  577. PowerPort.objects.filter(device=device).select_related('power_outlet__device'), key=attrgetter('name')
  578. )
  579. power_outlets = natsorted(
  580. PowerOutlet.objects.filter(device=device).select_related('connected_port'), key=attrgetter('name')
  581. )
  582. interfaces = Interface.objects.order_naturally(device.device_type.interface_ordering).filter(
  583. device=device, mgmt_only=False
  584. ).select_related(
  585. 'connected_as_a__interface_b__device', 'connected_as_b__interface_a__device',
  586. 'circuit_termination__circuit'
  587. ).prefetch_related('ip_addresses')
  588. mgmt_interfaces = Interface.objects.order_naturally(device.device_type.interface_ordering).filter(
  589. device=device, mgmt_only=True
  590. ).select_related(
  591. 'connected_as_a__interface_b__device', 'connected_as_b__interface_a__device',
  592. 'circuit_termination__circuit'
  593. ).prefetch_related('ip_addresses')
  594. device_bays = natsorted(
  595. DeviceBay.objects.filter(device=device).select_related('installed_device__device_type__manufacturer'),
  596. key=attrgetter('name')
  597. )
  598. services = Service.objects.filter(device=device)
  599. secrets = device.secrets.all()
  600. # Find any related devices for convenient linking in the UI
  601. related_devices = []
  602. if device.name:
  603. if re.match('.+[0-9]+$', device.name):
  604. # Strip 1 or more trailing digits (e.g. core-switch1)
  605. base_name = re.match('(.*?)[0-9]+$', device.name).group(1)
  606. elif re.match('.+\d[a-z]$', device.name.lower()):
  607. # Strip a trailing letter if preceded by a digit (e.g. dist-switch3a -> dist-switch3)
  608. base_name = re.match('(.*\d+)[a-z]$', device.name.lower()).group(1)
  609. else:
  610. base_name = None
  611. if base_name:
  612. related_devices = Device.objects.filter(name__istartswith=base_name).exclude(pk=device.pk)\
  613. .select_related('rack', 'device_type__manufacturer')[:10]
  614. # Show graph button on interfaces only if at least one graph has been created.
  615. show_graphs = Graph.objects.filter(type=GRAPH_TYPE_INTERFACE).exists()
  616. return render(request, 'dcim/device.html', {
  617. 'device': device,
  618. 'console_ports': console_ports,
  619. 'cs_ports': cs_ports,
  620. 'power_ports': power_ports,
  621. 'power_outlets': power_outlets,
  622. 'interfaces': interfaces,
  623. 'mgmt_interfaces': mgmt_interfaces,
  624. 'device_bays': device_bays,
  625. 'services': services,
  626. 'secrets': secrets,
  627. 'related_devices': related_devices,
  628. 'show_graphs': show_graphs,
  629. })
  630. class DeviceInventoryView(View):
  631. def get(self, request, pk):
  632. device = get_object_or_404(Device, pk=pk)
  633. inventory_items = InventoryItem.objects.filter(
  634. device=device, parent=None
  635. ).select_related(
  636. 'manufacturer'
  637. ).prefetch_related(
  638. 'child_items'
  639. )
  640. return render(request, 'dcim/device_inventory.html', {
  641. 'device': device,
  642. 'inventory_items': inventory_items,
  643. })
  644. class DeviceLLDPNeighborsView(View):
  645. def get(self, request, pk):
  646. device = get_object_or_404(Device, pk=pk)
  647. interfaces = Interface.objects.order_naturally(
  648. device.device_type.interface_ordering
  649. ).filter(
  650. device=device
  651. ).select_related(
  652. 'connected_as_a', 'connected_as_b'
  653. )
  654. return render(request, 'dcim/device_lldp_neighbors.html', {
  655. 'device': device,
  656. 'interfaces': interfaces,
  657. })
  658. class DeviceEditView(PermissionRequiredMixin, ObjectEditView):
  659. permission_required = 'dcim.change_device'
  660. model = Device
  661. form_class = forms.DeviceForm
  662. template_name = 'dcim/device_edit.html'
  663. default_return_url = 'dcim:device_list'
  664. class DeviceDeleteView(PermissionRequiredMixin, ObjectDeleteView):
  665. permission_required = 'dcim.delete_device'
  666. model = Device
  667. default_return_url = 'dcim:device_list'
  668. class DeviceBulkImportView(PermissionRequiredMixin, BulkImportView):
  669. permission_required = 'dcim.add_device'
  670. form = forms.DeviceImportForm
  671. table = tables.DeviceImportTable
  672. template_name = 'dcim/device_import.html'
  673. default_return_url = 'dcim:device_list'
  674. class ChildDeviceBulkImportView(PermissionRequiredMixin, BulkImportView):
  675. permission_required = 'dcim.add_device'
  676. form = forms.ChildDeviceImportForm
  677. table = tables.DeviceImportTable
  678. template_name = 'dcim/device_import_child.html'
  679. default_return_url = 'dcim:device_list'
  680. def save_obj(self, obj):
  681. # Inherit site and rack from parent device
  682. obj.site = obj.parent_bay.device.site
  683. obj.rack = obj.parent_bay.device.rack
  684. obj.save()
  685. # Save the reverse relation
  686. device_bay = obj.parent_bay
  687. device_bay.installed_device = obj
  688. device_bay.save()
  689. class DeviceBulkEditView(PermissionRequiredMixin, BulkEditView):
  690. permission_required = 'dcim.change_device'
  691. cls = Device
  692. filter = filters.DeviceFilter
  693. form = forms.DeviceBulkEditForm
  694. template_name = 'dcim/device_bulk_edit.html'
  695. default_return_url = 'dcim:device_list'
  696. class DeviceBulkDeleteView(PermissionRequiredMixin, BulkDeleteView):
  697. permission_required = 'dcim.delete_device'
  698. cls = Device
  699. filter = filters.DeviceFilter
  700. default_return_url = 'dcim:device_list'
  701. #
  702. # Console ports
  703. #
  704. class ConsolePortAddView(PermissionRequiredMixin, ComponentCreateView):
  705. permission_required = 'dcim.add_consoleport'
  706. parent_model = Device
  707. parent_field = 'device'
  708. model = ConsolePort
  709. form = forms.ConsolePortCreateForm
  710. model_form = forms.ConsolePortForm
  711. @permission_required('dcim.change_consoleport')
  712. def consoleport_connect(request, pk):
  713. consoleport = get_object_or_404(ConsolePort, pk=pk)
  714. if request.method == 'POST':
  715. form = forms.ConsolePortConnectionForm(request.POST, instance=consoleport)
  716. if form.is_valid():
  717. consoleport = form.save()
  718. msg = 'Connected <a href="{}">{}</a> {} to <a href="{}">{}</a> {}'.format(
  719. consoleport.device.get_absolute_url(),
  720. escape(consoleport.device),
  721. escape(consoleport.name),
  722. consoleport.cs_port.device.get_absolute_url(),
  723. escape(consoleport.cs_port.device),
  724. escape(consoleport.cs_port.name),
  725. )
  726. messages.success(request, mark_safe(msg))
  727. UserAction.objects.log_edit(request.user, consoleport, msg)
  728. return redirect('dcim:device', pk=consoleport.device.pk)
  729. else:
  730. form = forms.ConsolePortConnectionForm(instance=consoleport, initial={
  731. 'site': request.GET.get('site', consoleport.device.site),
  732. 'rack': request.GET.get('rack', None),
  733. 'console_server': request.GET.get('console_server', None),
  734. 'connection_status': CONNECTION_STATUS_CONNECTED,
  735. })
  736. return render(request, 'dcim/consoleport_connect.html', {
  737. 'consoleport': consoleport,
  738. 'form': form,
  739. 'return_url': reverse('dcim:device', kwargs={'pk': consoleport.device.pk}),
  740. })
  741. @permission_required('dcim.change_consoleport')
  742. def consoleport_disconnect(request, pk):
  743. consoleport = get_object_or_404(ConsolePort, pk=pk)
  744. if not consoleport.cs_port:
  745. messages.warning(
  746. request, "Cannot disconnect console port {}: It is not connected to anything.".format(consoleport)
  747. )
  748. return redirect('dcim:device', pk=consoleport.device.pk)
  749. if request.method == 'POST':
  750. form = ConfirmationForm(request.POST)
  751. if form.is_valid():
  752. cs_port = consoleport.cs_port
  753. consoleport.cs_port = None
  754. consoleport.connection_status = None
  755. consoleport.save()
  756. msg = 'Disconnected <a href="{}">{}</a> {} from <a href="{}">{}</a> {}'.format(
  757. consoleport.device.get_absolute_url(),
  758. escape(consoleport.device),
  759. escape(consoleport.name),
  760. cs_port.device.get_absolute_url(),
  761. escape(cs_port.device),
  762. escape(cs_port.name),
  763. )
  764. messages.success(request, mark_safe(msg))
  765. UserAction.objects.log_edit(request.user, consoleport, msg)
  766. return redirect('dcim:device', pk=consoleport.device.pk)
  767. else:
  768. form = ConfirmationForm()
  769. return render(request, 'dcim/consoleport_disconnect.html', {
  770. 'consoleport': consoleport,
  771. 'form': form,
  772. 'return_url': reverse('dcim:device', kwargs={'pk': consoleport.device.pk}),
  773. })
  774. class ConsolePortEditView(PermissionRequiredMixin, ComponentEditView):
  775. permission_required = 'dcim.change_consoleport'
  776. model = ConsolePort
  777. form_class = forms.ConsolePortForm
  778. class ConsolePortDeleteView(PermissionRequiredMixin, ComponentDeleteView):
  779. permission_required = 'dcim.delete_consoleport'
  780. model = ConsolePort
  781. class ConsolePortBulkDeleteView(PermissionRequiredMixin, BulkDeleteView):
  782. permission_required = 'dcim.delete_consoleport'
  783. cls = ConsolePort
  784. parent_cls = Device
  785. class ConsoleConnectionsBulkImportView(PermissionRequiredMixin, BulkImportView):
  786. permission_required = 'dcim.change_consoleport'
  787. form = forms.ConsoleConnectionImportForm
  788. table = tables.ConsoleConnectionTable
  789. template_name = 'dcim/console_connections_import.html'
  790. default_return_url = 'dcim:console_connections_list'
  791. #
  792. # Console server ports
  793. #
  794. class ConsoleServerPortAddView(PermissionRequiredMixin, ComponentCreateView):
  795. permission_required = 'dcim.add_consoleserverport'
  796. parent_model = Device
  797. parent_field = 'device'
  798. model = ConsoleServerPort
  799. form = forms.ConsoleServerPortCreateForm
  800. model_form = forms.ConsoleServerPortForm
  801. @permission_required('dcim.change_consoleserverport')
  802. def consoleserverport_connect(request, pk):
  803. consoleserverport = get_object_or_404(ConsoleServerPort, pk=pk)
  804. if request.method == 'POST':
  805. form = forms.ConsoleServerPortConnectionForm(request.POST)
  806. if form.is_valid():
  807. consoleport = form.cleaned_data['port']
  808. consoleport.cs_port = consoleserverport
  809. consoleport.connection_status = form.cleaned_data['connection_status']
  810. consoleport.save()
  811. msg = 'Connected <a href="{}">{}</a> {} to <a href="{}">{}</a> {}'.format(
  812. consoleport.device.get_absolute_url(),
  813. escape(consoleport.device),
  814. escape(consoleport.name),
  815. consoleserverport.device.get_absolute_url(),
  816. escape(consoleserverport.device),
  817. escape(consoleserverport.name),
  818. )
  819. messages.success(request, mark_safe(msg))
  820. UserAction.objects.log_edit(request.user, consoleport, msg)
  821. return redirect('dcim:device', pk=consoleserverport.device.pk)
  822. else:
  823. form = forms.ConsoleServerPortConnectionForm(initial={
  824. 'site': request.GET.get('site', consoleserverport.device.site),
  825. 'rack': request.GET.get('rack', None),
  826. 'device': request.GET.get('device', None),
  827. 'connection_status': CONNECTION_STATUS_CONNECTED,
  828. })
  829. return render(request, 'dcim/consoleserverport_connect.html', {
  830. 'consoleserverport': consoleserverport,
  831. 'form': form,
  832. 'return_url': reverse('dcim:device', kwargs={'pk': consoleserverport.device.pk}),
  833. })
  834. @permission_required('dcim.change_consoleserverport')
  835. def consoleserverport_disconnect(request, pk):
  836. consoleserverport = get_object_or_404(ConsoleServerPort, pk=pk)
  837. if not hasattr(consoleserverport, 'connected_console'):
  838. messages.warning(
  839. request, "Cannot disconnect console server port {}: Nothing is connected to it.".format(consoleserverport)
  840. )
  841. return redirect('dcim:device', pk=consoleserverport.device.pk)
  842. if request.method == 'POST':
  843. form = ConfirmationForm(request.POST)
  844. if form.is_valid():
  845. consoleport = consoleserverport.connected_console
  846. consoleport.cs_port = None
  847. consoleport.connection_status = None
  848. consoleport.save()
  849. msg = 'Disconnected <a href="{}">{}</a> {} from <a href="{}">{}</a> {}'.format(
  850. consoleport.device.get_absolute_url(),
  851. escape(consoleport.device),
  852. escape(consoleport.name),
  853. consoleserverport.device.get_absolute_url(),
  854. escape(consoleserverport.device),
  855. escape(consoleserverport.name),
  856. )
  857. messages.success(request, mark_safe(msg))
  858. UserAction.objects.log_edit(request.user, consoleport, msg)
  859. return redirect('dcim:device', pk=consoleserverport.device.pk)
  860. else:
  861. form = ConfirmationForm()
  862. return render(request, 'dcim/consoleserverport_disconnect.html', {
  863. 'consoleserverport': consoleserverport,
  864. 'form': form,
  865. 'return_url': reverse('dcim:device', kwargs={'pk': consoleserverport.device.pk}),
  866. })
  867. class ConsoleServerPortEditView(PermissionRequiredMixin, ComponentEditView):
  868. permission_required = 'dcim.change_consoleserverport'
  869. model = ConsoleServerPort
  870. form_class = forms.ConsoleServerPortForm
  871. class ConsoleServerPortDeleteView(PermissionRequiredMixin, ComponentDeleteView):
  872. permission_required = 'dcim.delete_consoleserverport'
  873. model = ConsoleServerPort
  874. class ConsoleServerPortBulkDeleteView(PermissionRequiredMixin, BulkDeleteView):
  875. permission_required = 'dcim.delete_consoleserverport'
  876. cls = ConsoleServerPort
  877. parent_cls = Device
  878. #
  879. # Power ports
  880. #
  881. class PowerPortAddView(PermissionRequiredMixin, ComponentCreateView):
  882. permission_required = 'dcim.add_powerport'
  883. parent_model = Device
  884. parent_field = 'device'
  885. model = PowerPort
  886. form = forms.PowerPortCreateForm
  887. model_form = forms.PowerPortForm
  888. @permission_required('dcim.change_powerport')
  889. def powerport_connect(request, pk):
  890. powerport = get_object_or_404(PowerPort, pk=pk)
  891. if request.method == 'POST':
  892. form = forms.PowerPortConnectionForm(request.POST, instance=powerport)
  893. if form.is_valid():
  894. powerport = form.save()
  895. msg = 'Connected <a href="{}">{}</a> {} to <a href="{}">{}</a> {}'.format(
  896. powerport.device.get_absolute_url(),
  897. escape(powerport.device),
  898. escape(powerport.name),
  899. powerport.power_outlet.device.get_absolute_url(),
  900. escape(powerport.power_outlet.device),
  901. escape(powerport.power_outlet.name),
  902. )
  903. messages.success(request, mark_safe(msg))
  904. UserAction.objects.log_edit(request.user, powerport, msg)
  905. return redirect('dcim:device', pk=powerport.device.pk)
  906. else:
  907. form = forms.PowerPortConnectionForm(instance=powerport, initial={
  908. 'site': request.GET.get('site', powerport.device.site),
  909. 'rack': request.GET.get('rack', None),
  910. 'pdu': request.GET.get('pdu', None),
  911. 'connection_status': CONNECTION_STATUS_CONNECTED,
  912. })
  913. return render(request, 'dcim/powerport_connect.html', {
  914. 'powerport': powerport,
  915. 'form': form,
  916. 'return_url': reverse('dcim:device', kwargs={'pk': powerport.device.pk}),
  917. })
  918. @permission_required('dcim.change_powerport')
  919. def powerport_disconnect(request, pk):
  920. powerport = get_object_or_404(PowerPort, pk=pk)
  921. if not powerport.power_outlet:
  922. messages.warning(
  923. request, "Cannot disconnect power port {}: It is not connected to an outlet.".format(powerport)
  924. )
  925. return redirect('dcim:device', pk=powerport.device.pk)
  926. if request.method == 'POST':
  927. form = ConfirmationForm(request.POST)
  928. if form.is_valid():
  929. power_outlet = powerport.power_outlet
  930. powerport.power_outlet = None
  931. powerport.connection_status = None
  932. powerport.save()
  933. msg = 'Disconnected <a href="{}">{}</a> {} from <a href="{}">{}</a> {}'.format(
  934. powerport.device.get_absolute_url(),
  935. escape(powerport.device),
  936. escape(powerport.name),
  937. power_outlet.device.get_absolute_url(),
  938. escape(power_outlet.device),
  939. escape(power_outlet.name),
  940. )
  941. messages.success(request, mark_safe(msg))
  942. UserAction.objects.log_edit(request.user, powerport, msg)
  943. return redirect('dcim:device', pk=powerport.device.pk)
  944. else:
  945. form = ConfirmationForm()
  946. return render(request, 'dcim/powerport_disconnect.html', {
  947. 'powerport': powerport,
  948. 'form': form,
  949. 'return_url': reverse('dcim:device', kwargs={'pk': powerport.device.pk}),
  950. })
  951. class PowerPortEditView(PermissionRequiredMixin, ComponentEditView):
  952. permission_required = 'dcim.change_powerport'
  953. model = PowerPort
  954. form_class = forms.PowerPortForm
  955. class PowerPortDeleteView(PermissionRequiredMixin, ComponentDeleteView):
  956. permission_required = 'dcim.delete_powerport'
  957. model = PowerPort
  958. class PowerPortBulkDeleteView(PermissionRequiredMixin, BulkDeleteView):
  959. permission_required = 'dcim.delete_powerport'
  960. cls = PowerPort
  961. parent_cls = Device
  962. class PowerConnectionsBulkImportView(PermissionRequiredMixin, BulkImportView):
  963. permission_required = 'dcim.change_powerport'
  964. form = forms.PowerConnectionImportForm
  965. table = tables.PowerConnectionTable
  966. template_name = 'dcim/power_connections_import.html'
  967. default_return_url = 'dcim:power_connections_list'
  968. #
  969. # Power outlets
  970. #
  971. class PowerOutletAddView(PermissionRequiredMixin, ComponentCreateView):
  972. permission_required = 'dcim.add_poweroutlet'
  973. parent_model = Device
  974. parent_field = 'device'
  975. model = PowerOutlet
  976. form = forms.PowerOutletCreateForm
  977. model_form = forms.PowerOutletForm
  978. @permission_required('dcim.change_poweroutlet')
  979. def poweroutlet_connect(request, pk):
  980. poweroutlet = get_object_or_404(PowerOutlet, pk=pk)
  981. if request.method == 'POST':
  982. form = forms.PowerOutletConnectionForm(request.POST)
  983. if form.is_valid():
  984. powerport = form.cleaned_data['port']
  985. powerport.power_outlet = poweroutlet
  986. powerport.connection_status = form.cleaned_data['connection_status']
  987. powerport.save()
  988. msg = 'Connected <a href="{}">{}</a> {} to <a href="{}">{}</a> {}'.format(
  989. powerport.device.get_absolute_url(),
  990. escape(powerport.device),
  991. escape(powerport.name),
  992. poweroutlet.device.get_absolute_url(),
  993. escape(poweroutlet.device),
  994. escape(poweroutlet.name),
  995. )
  996. messages.success(request, mark_safe(msg))
  997. UserAction.objects.log_edit(request.user, powerport, msg)
  998. return redirect('dcim:device', pk=poweroutlet.device.pk)
  999. else:
  1000. form = forms.PowerOutletConnectionForm(initial={
  1001. 'site': request.GET.get('site', poweroutlet.device.site),
  1002. 'rack': request.GET.get('rack', None),
  1003. 'device': request.GET.get('device', None),
  1004. 'connection_status': CONNECTION_STATUS_CONNECTED,
  1005. })
  1006. return render(request, 'dcim/poweroutlet_connect.html', {
  1007. 'poweroutlet': poweroutlet,
  1008. 'form': form,
  1009. 'return_url': reverse('dcim:device', kwargs={'pk': poweroutlet.device.pk}),
  1010. })
  1011. @permission_required('dcim.change_poweroutlet')
  1012. def poweroutlet_disconnect(request, pk):
  1013. poweroutlet = get_object_or_404(PowerOutlet, pk=pk)
  1014. if not hasattr(poweroutlet, 'connected_port'):
  1015. messages.warning(
  1016. request, "Cannot disconnect power outlet {}: Nothing is connected to it.".format(poweroutlet)
  1017. )
  1018. return redirect('dcim:device', pk=poweroutlet.device.pk)
  1019. if request.method == 'POST':
  1020. form = ConfirmationForm(request.POST)
  1021. if form.is_valid():
  1022. powerport = poweroutlet.connected_port
  1023. powerport.power_outlet = None
  1024. powerport.connection_status = None
  1025. powerport.save()
  1026. msg = 'Disconnected <a href="{}">{}</a> {} from <a href="{}">{}</a> {}'.format(
  1027. powerport.device.get_absolute_url(),
  1028. escape(powerport.device),
  1029. escape(powerport.name),
  1030. poweroutlet.device.get_absolute_url(),
  1031. escape(poweroutlet.device),
  1032. escape(poweroutlet.name),
  1033. )
  1034. messages.success(request, mark_safe(msg))
  1035. UserAction.objects.log_edit(request.user, powerport, msg)
  1036. return redirect('dcim:device', pk=poweroutlet.device.pk)
  1037. else:
  1038. form = ConfirmationForm()
  1039. return render(request, 'dcim/poweroutlet_disconnect.html', {
  1040. 'poweroutlet': poweroutlet,
  1041. 'form': form,
  1042. 'return_url': reverse('dcim:device', kwargs={'pk': poweroutlet.device.pk}),
  1043. })
  1044. class PowerOutletEditView(PermissionRequiredMixin, ComponentEditView):
  1045. permission_required = 'dcim.change_poweroutlet'
  1046. model = PowerOutlet
  1047. form_class = forms.PowerOutletForm
  1048. class PowerOutletDeleteView(PermissionRequiredMixin, ComponentDeleteView):
  1049. permission_required = 'dcim.delete_poweroutlet'
  1050. model = PowerOutlet
  1051. class PowerOutletBulkDeleteView(PermissionRequiredMixin, BulkDeleteView):
  1052. permission_required = 'dcim.delete_poweroutlet'
  1053. cls = PowerOutlet
  1054. parent_cls = Device
  1055. #
  1056. # Interfaces
  1057. #
  1058. class InterfaceAddView(PermissionRequiredMixin, ComponentCreateView):
  1059. permission_required = 'dcim.add_interface'
  1060. parent_model = Device
  1061. parent_field = 'device'
  1062. model = Interface
  1063. form = forms.InterfaceCreateForm
  1064. model_form = forms.InterfaceForm
  1065. class InterfaceEditView(PermissionRequiredMixin, ComponentEditView):
  1066. permission_required = 'dcim.change_interface'
  1067. model = Interface
  1068. form_class = forms.InterfaceForm
  1069. class InterfaceDeleteView(PermissionRequiredMixin, ComponentDeleteView):
  1070. permission_required = 'dcim.delete_interface'
  1071. model = Interface
  1072. class InterfaceBulkEditView(PermissionRequiredMixin, BulkEditView):
  1073. permission_required = 'dcim.change_interface'
  1074. cls = Interface
  1075. parent_cls = Device
  1076. form = forms.InterfaceBulkEditForm
  1077. template_name = 'dcim/interface_bulk_edit.html'
  1078. class InterfaceBulkDeleteView(PermissionRequiredMixin, BulkDeleteView):
  1079. permission_required = 'dcim.delete_interface'
  1080. cls = Interface
  1081. parent_cls = Device
  1082. #
  1083. # Device bays
  1084. #
  1085. class DeviceBayAddView(PermissionRequiredMixin, ComponentCreateView):
  1086. permission_required = 'dcim.add_devicebay'
  1087. parent_model = Device
  1088. parent_field = 'device'
  1089. model = DeviceBay
  1090. form = forms.DeviceBayCreateForm
  1091. model_form = forms.DeviceBayForm
  1092. class DeviceBayEditView(PermissionRequiredMixin, ComponentEditView):
  1093. permission_required = 'dcim.change_devicebay'
  1094. model = DeviceBay
  1095. form_class = forms.DeviceBayForm
  1096. class DeviceBayDeleteView(PermissionRequiredMixin, ComponentDeleteView):
  1097. permission_required = 'dcim.delete_devicebay'
  1098. model = DeviceBay
  1099. @permission_required('dcim.change_devicebay')
  1100. def devicebay_populate(request, pk):
  1101. device_bay = get_object_or_404(DeviceBay, pk=pk)
  1102. if request.method == 'POST':
  1103. form = forms.PopulateDeviceBayForm(device_bay, request.POST)
  1104. if form.is_valid():
  1105. device_bay.installed_device = form.cleaned_data['installed_device']
  1106. device_bay.save()
  1107. if not form.errors:
  1108. messages.success(request, "Added {} to {}.".format(device_bay.installed_device, device_bay))
  1109. return redirect('dcim:device', pk=device_bay.device.pk)
  1110. else:
  1111. form = forms.PopulateDeviceBayForm(device_bay)
  1112. return render(request, 'dcim/devicebay_populate.html', {
  1113. 'device_bay': device_bay,
  1114. 'form': form,
  1115. 'return_url': reverse('dcim:device', kwargs={'pk': device_bay.device.pk}),
  1116. })
  1117. @permission_required('dcim.change_devicebay')
  1118. def devicebay_depopulate(request, pk):
  1119. device_bay = get_object_or_404(DeviceBay, pk=pk)
  1120. if request.method == 'POST':
  1121. form = ConfirmationForm(request.POST)
  1122. if form.is_valid():
  1123. removed_device = device_bay.installed_device
  1124. device_bay.installed_device = None
  1125. device_bay.save()
  1126. messages.success(request, "{} has been removed from {}.".format(removed_device, device_bay))
  1127. return redirect('dcim:device', pk=device_bay.device.pk)
  1128. else:
  1129. form = ConfirmationForm()
  1130. return render(request, 'dcim/devicebay_depopulate.html', {
  1131. 'device_bay': device_bay,
  1132. 'form': form,
  1133. 'return_url': reverse('dcim:device', kwargs={'pk': device_bay.device.pk}),
  1134. })
  1135. class DeviceBayBulkDeleteView(PermissionRequiredMixin, BulkDeleteView):
  1136. permission_required = 'dcim.delete_devicebay'
  1137. cls = DeviceBay
  1138. parent_cls = Device
  1139. #
  1140. # Bulk device component creation
  1141. #
  1142. class DeviceBulkAddComponentView(View):
  1143. """
  1144. Add one or more components (e.g. interfaces) to a selected set of Devices.
  1145. """
  1146. form = forms.DeviceBulkAddComponentForm
  1147. model = None
  1148. model_form = None
  1149. def get(self):
  1150. return redirect('dcim:device_list')
  1151. def post(self, request):
  1152. # Are we editing *all* objects in the queryset or just a selected subset?
  1153. if request.POST.get('_all'):
  1154. pk_list = [obj.pk for obj in filters.DeviceFilter(request.GET, Device.objects.all())]
  1155. else:
  1156. pk_list = [int(pk) for pk in request.POST.getlist('pk')]
  1157. if '_create' in request.POST:
  1158. form = self.form(request.POST)
  1159. if form.is_valid():
  1160. new_components = []
  1161. data = deepcopy(form.cleaned_data)
  1162. for device in data['pk']:
  1163. names = data['name_pattern']
  1164. for name in names:
  1165. component_data = {
  1166. 'device': device.pk,
  1167. 'name': name,
  1168. }
  1169. component_data.update(data)
  1170. component_form = self.model_form(component_data)
  1171. if component_form.is_valid():
  1172. new_components.append(component_form.save(commit=False))
  1173. else:
  1174. for field, errors in component_form.errors.as_data().items():
  1175. for e in errors:
  1176. form.add_error(field, '{} {}: {}'.format(device, name, ', '.join(e)))
  1177. if not form.errors:
  1178. self.model.objects.bulk_create(new_components)
  1179. messages.success(request, "Added {} {} to {} devices.".format(
  1180. len(new_components), self.model._meta.verbose_name_plural, len(form.cleaned_data['pk'])
  1181. ))
  1182. return redirect('dcim:device_list')
  1183. else:
  1184. form = self.form(initial={'pk': pk_list})
  1185. selected_devices = Device.objects.filter(pk__in=pk_list)
  1186. if not selected_devices:
  1187. messages.warning(request, "No devices were selected.")
  1188. return redirect('dcim:device_list')
  1189. return render(request, 'dcim/device_bulk_add_component.html', {
  1190. 'form': form,
  1191. 'component_name': self.model._meta.verbose_name_plural,
  1192. 'selected_devices': selected_devices,
  1193. 'return_url': reverse('dcim:device_list'),
  1194. })
  1195. class DeviceBulkAddConsolePortView(PermissionRequiredMixin, DeviceBulkAddComponentView):
  1196. permission_required = 'dcim.add_consoleport'
  1197. model = ConsolePort
  1198. model_form = forms.ConsolePortForm
  1199. class DeviceBulkAddConsoleServerPortView(PermissionRequiredMixin, DeviceBulkAddComponentView):
  1200. permission_required = 'dcim.add_consoleserverport'
  1201. model = ConsoleServerPort
  1202. model_form = forms.ConsoleServerPortForm
  1203. class DeviceBulkAddPowerPortView(PermissionRequiredMixin, DeviceBulkAddComponentView):
  1204. permission_required = 'dcim.add_powerport'
  1205. model = PowerPort
  1206. model_form = forms.PowerPortForm
  1207. class DeviceBulkAddPowerOutletView(PermissionRequiredMixin, DeviceBulkAddComponentView):
  1208. permission_required = 'dcim.add_poweroutlet'
  1209. model = PowerOutlet
  1210. model_form = forms.PowerOutletForm
  1211. class DeviceBulkAddInterfaceView(PermissionRequiredMixin, DeviceBulkAddComponentView):
  1212. permission_required = 'dcim.add_interface'
  1213. form = forms.DeviceBulkAddInterfaceForm
  1214. model = Interface
  1215. model_form = forms.InterfaceForm
  1216. class DeviceBulkAddDeviceBayView(PermissionRequiredMixin, DeviceBulkAddComponentView):
  1217. permission_required = 'dcim.add_devicebay'
  1218. model = DeviceBay
  1219. model_form = forms.DeviceBayForm
  1220. #
  1221. # Interface connections
  1222. #
  1223. @permission_required('dcim.add_interfaceconnection')
  1224. def interfaceconnection_add(request, pk):
  1225. device = get_object_or_404(Device, pk=pk)
  1226. if request.method == 'POST':
  1227. form = forms.InterfaceConnectionForm(device, request.POST)
  1228. if form.is_valid():
  1229. interfaceconnection = form.save()
  1230. msg = 'Connected <a href="{}">{}</a> {} to <a href="{}">{}</a> {}'.format(
  1231. interfaceconnection.interface_a.device.get_absolute_url(),
  1232. escape(interfaceconnection.interface_a.device),
  1233. escape(interfaceconnection.interface_a.name),
  1234. interfaceconnection.interface_b.device.get_absolute_url(),
  1235. escape(interfaceconnection.interface_b.device),
  1236. escape(interfaceconnection.interface_b.name),
  1237. )
  1238. messages.success(request, mark_safe(msg))
  1239. UserAction.objects.log_edit(request.user, interfaceconnection, msg)
  1240. if '_addanother' in request.POST:
  1241. base_url = reverse('dcim:interfaceconnection_add', kwargs={'pk': device.pk})
  1242. device_b = interfaceconnection.interface_b.device
  1243. params = urlencode({
  1244. 'rack_b': device_b.rack.pk if device_b.rack else '',
  1245. 'device_b': device_b.pk,
  1246. })
  1247. return HttpResponseRedirect('{}?{}'.format(base_url, params))
  1248. else:
  1249. return redirect('dcim:device', pk=device.pk)
  1250. else:
  1251. form = forms.InterfaceConnectionForm(device, initial={
  1252. 'interface_a': request.GET.get('interface_a', None),
  1253. 'site_b': request.GET.get('site_b', device.site),
  1254. 'rack_b': request.GET.get('rack_b', None),
  1255. 'device_b': request.GET.get('device_b', None),
  1256. 'interface_b': request.GET.get('interface_b', None),
  1257. })
  1258. return render(request, 'dcim/interfaceconnection_edit.html', {
  1259. 'device': device,
  1260. 'form': form,
  1261. 'return_url': reverse('dcim:device', kwargs={'pk': device.pk}),
  1262. })
  1263. @permission_required('dcim.delete_interfaceconnection')
  1264. def interfaceconnection_delete(request, pk):
  1265. interfaceconnection = get_object_or_404(InterfaceConnection, pk=pk)
  1266. device_id = request.GET.get('device', None)
  1267. if request.method == 'POST':
  1268. form = forms.InterfaceConnectionDeletionForm(request.POST)
  1269. if form.is_valid():
  1270. interfaceconnection.delete()
  1271. msg = 'Disconnected <a href="{}">{}</a> {} from <a href="{}">{}</a> {}'.format(
  1272. interfaceconnection.interface_a.device.get_absolute_url(),
  1273. escape(interfaceconnection.interface_a.device),
  1274. escape(interfaceconnection.interface_a.name),
  1275. interfaceconnection.interface_b.device.get_absolute_url(),
  1276. escape(interfaceconnection.interface_b.device),
  1277. escape(interfaceconnection.interface_b.name),
  1278. )
  1279. messages.success(request, mark_safe(msg))
  1280. UserAction.objects.log_edit(request.user, interfaceconnection, msg)
  1281. if form.cleaned_data['device']:
  1282. return redirect('dcim:device', pk=form.cleaned_data['device'].pk)
  1283. else:
  1284. return redirect('dcim:device_list')
  1285. else:
  1286. form = forms.InterfaceConnectionDeletionForm(initial={
  1287. 'device': device_id,
  1288. })
  1289. # Determine where to direct user upon cancellation
  1290. if device_id:
  1291. return_url = reverse('dcim:device', kwargs={'pk': device_id})
  1292. else:
  1293. return_url = reverse('dcim:device_list')
  1294. return render(request, 'dcim/interfaceconnection_delete.html', {
  1295. 'interfaceconnection': interfaceconnection,
  1296. 'device_id': device_id,
  1297. 'form': form,
  1298. 'return_url': return_url,
  1299. })
  1300. class InterfaceConnectionsBulkImportView(PermissionRequiredMixin, BulkImportView):
  1301. permission_required = 'dcim.change_interface'
  1302. form = forms.InterfaceConnectionImportForm
  1303. table = tables.InterfaceConnectionTable
  1304. template_name = 'dcim/interface_connections_import.html'
  1305. default_return_url = 'dcim:interface_connections_list'
  1306. #
  1307. # Connections
  1308. #
  1309. class ConsoleConnectionsListView(ObjectListView):
  1310. queryset = ConsolePort.objects.select_related('device', 'cs_port__device').filter(cs_port__isnull=False)\
  1311. .order_by('cs_port__device__name', 'cs_port__name')
  1312. filter = filters.ConsoleConnectionFilter
  1313. filter_form = forms.ConsoleConnectionFilterForm
  1314. table = tables.ConsoleConnectionTable
  1315. template_name = 'dcim/console_connections_list.html'
  1316. class PowerConnectionsListView(ObjectListView):
  1317. queryset = PowerPort.objects.select_related('device', 'power_outlet__device').filter(power_outlet__isnull=False)\
  1318. .order_by('power_outlet__device__name', 'power_outlet__name')
  1319. filter = filters.PowerConnectionFilter
  1320. filter_form = forms.PowerConnectionFilterForm
  1321. table = tables.PowerConnectionTable
  1322. template_name = 'dcim/power_connections_list.html'
  1323. class InterfaceConnectionsListView(ObjectListView):
  1324. queryset = InterfaceConnection.objects.select_related('interface_a__device', 'interface_b__device')\
  1325. .order_by('interface_a__device__name', 'interface_a__name')
  1326. filter = filters.InterfaceConnectionFilter
  1327. filter_form = forms.InterfaceConnectionFilterForm
  1328. table = tables.InterfaceConnectionTable
  1329. template_name = 'dcim/interface_connections_list.html'
  1330. #
  1331. # Inventory items
  1332. #
  1333. class InventoryItemEditView(PermissionRequiredMixin, ComponentEditView):
  1334. permission_required = 'dcim.change_inventoryitem'
  1335. model = InventoryItem
  1336. form_class = forms.InventoryItemForm
  1337. def alter_obj(self, obj, request, url_args, url_kwargs):
  1338. if 'device' in url_kwargs:
  1339. obj.device = get_object_or_404(Device, pk=url_kwargs['device'])
  1340. return obj
  1341. class InventoryItemDeleteView(PermissionRequiredMixin, ComponentDeleteView):
  1342. permission_required = 'dcim.delete_inventoryitem'
  1343. model = InventoryItem