forms.py 76 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338
  1. from __future__ import unicode_literals
  2. import re
  3. from django import forms
  4. from django.contrib.auth.models import User
  5. from django.contrib.postgres.forms.array import SimpleArrayField
  6. from django.db.models import Count, Q
  7. from mptt.forms import TreeNodeChoiceField
  8. from timezone_field import TimeZoneFormField
  9. from extras.forms import CustomFieldForm, CustomFieldBulkEditForm, CustomFieldFilterForm
  10. from ipam.models import IPAddress, VLAN, VLANGroup
  11. from tenancy.forms import TenancyForm
  12. from tenancy.models import Tenant
  13. from utilities.forms import (
  14. APISelect, APISelectMultiple, add_blank_choice, ArrayFieldSelectMultiple, BootstrapMixin, BulkEditForm,
  15. BulkEditNullBooleanSelect, ChainedFieldsMixin, ChainedModelChoiceField, ChainedModelMultipleChoiceField,
  16. CommentField, ComponentForm, ConfirmationForm, CSVChoiceField, ExpandableNameField, FilterChoiceField,
  17. FlexibleModelChoiceField, Livesearch, SelectWithDisabled, SmallTextarea, SlugField,
  18. FilterTreeNodeMultipleChoiceField,
  19. )
  20. from virtualization.models import Cluster
  21. from .constants import (
  22. CONNECTION_STATUS_CHOICES, CONNECTION_STATUS_CONNECTED, DEVICE_STATUS_CHOICES, IFACE_FF_CHOICES, IFACE_FF_LAG,
  23. IFACE_MODE_ACCESS, IFACE_MODE_CHOICES, IFACE_MODE_TAGGED_ALL, IFACE_ORDERING_CHOICES, RACK_FACE_CHOICES,
  24. RACK_TYPE_CHOICES, RACK_WIDTH_CHOICES, RACK_WIDTH_19IN, RACK_WIDTH_23IN, SITE_STATUS_CHOICES, SUBDEVICE_ROLE_CHILD,
  25. SUBDEVICE_ROLE_PARENT, SUBDEVICE_ROLE_CHOICES,
  26. )
  27. from .formfields import MACAddressFormField
  28. from .models import (
  29. DeviceBay, DeviceBayTemplate, ConsolePort, ConsolePortTemplate, ConsoleServerPort, ConsoleServerPortTemplate,
  30. Device, DeviceRole, DeviceType, Interface, InterfaceConnection, InterfaceTemplate, Manufacturer, InventoryItem,
  31. Platform, PowerOutlet, PowerOutletTemplate, PowerPort, PowerPortTemplate, Rack, RackGroup, RackReservation,
  32. RackRole, Region, Site, VirtualChassis
  33. )
  34. DEVICE_BY_PK_RE = '{\d+\}'
  35. def get_device_by_name_or_pk(name):
  36. """
  37. Attempt to retrieve a device by either its name or primary key ('{pk}').
  38. """
  39. if re.match(DEVICE_BY_PK_RE, name):
  40. pk = name.strip('{}')
  41. device = Device.objects.get(pk=pk)
  42. else:
  43. device = Device.objects.get(name=name)
  44. return device
  45. class BulkRenameForm(forms.Form):
  46. """
  47. An extendable form to be used for renaming device components in bulk.
  48. """
  49. find = forms.CharField()
  50. replace = forms.CharField()
  51. #
  52. # Regions
  53. #
  54. class RegionForm(BootstrapMixin, forms.ModelForm):
  55. slug = SlugField()
  56. class Meta:
  57. model = Region
  58. fields = ['parent', 'name', 'slug']
  59. class RegionCSVForm(forms.ModelForm):
  60. parent = forms.ModelChoiceField(
  61. queryset=Region.objects.all(),
  62. required=False,
  63. to_field_name='name',
  64. help_text='Name of parent region',
  65. error_messages={
  66. 'invalid_choice': 'Region not found.',
  67. }
  68. )
  69. class Meta:
  70. model = Region
  71. fields = Region.csv_headers
  72. help_texts = {
  73. 'name': 'Region name',
  74. 'slug': 'URL-friendly slug',
  75. }
  76. class RegionFilterForm(BootstrapMixin, forms.Form):
  77. model = Site
  78. q = forms.CharField(required=False, label='Search')
  79. #
  80. # Sites
  81. #
  82. class SiteForm(BootstrapMixin, TenancyForm, CustomFieldForm):
  83. region = TreeNodeChoiceField(queryset=Region.objects.all(), required=False)
  84. slug = SlugField()
  85. comments = CommentField()
  86. class Meta:
  87. model = Site
  88. fields = [
  89. 'name', 'slug', 'status', 'region', 'tenant_group', 'tenant', 'facility', 'asn', 'description',
  90. 'physical_address', 'shipping_address', 'contact_name', 'contact_phone', 'contact_email', 'time_zone',
  91. 'comments',
  92. ]
  93. widgets = {
  94. 'physical_address': SmallTextarea(attrs={'rows': 3}),
  95. 'shipping_address': SmallTextarea(attrs={'rows': 3}),
  96. }
  97. help_texts = {
  98. 'name': "Full name of the site",
  99. 'facility': "Data center provider and facility (e.g. Equinix NY7)",
  100. 'asn': "BGP autonomous system number",
  101. 'physical_address': "Physical location of the building (e.g. for GPS)",
  102. 'shipping_address': "If different from the physical address"
  103. }
  104. class SiteCSVForm(forms.ModelForm):
  105. status = CSVChoiceField(
  106. choices=DEVICE_STATUS_CHOICES,
  107. required=False,
  108. help_text='Operational status'
  109. )
  110. region = forms.ModelChoiceField(
  111. queryset=Region.objects.all(),
  112. required=False,
  113. to_field_name='name',
  114. help_text='Name of assigned region',
  115. error_messages={
  116. 'invalid_choice': 'Region not found.',
  117. }
  118. )
  119. tenant = forms.ModelChoiceField(
  120. queryset=Tenant.objects.all(),
  121. required=False,
  122. to_field_name='name',
  123. help_text='Name of assigned tenant',
  124. error_messages={
  125. 'invalid_choice': 'Tenant not found.',
  126. }
  127. )
  128. class Meta:
  129. model = Site
  130. fields = Site.csv_headers
  131. help_texts = {
  132. 'name': 'Site name',
  133. 'slug': 'URL-friendly slug',
  134. 'asn': '32-bit autonomous system number',
  135. }
  136. class SiteBulkEditForm(BootstrapMixin, CustomFieldBulkEditForm):
  137. pk = forms.ModelMultipleChoiceField(queryset=Site.objects.all(), widget=forms.MultipleHiddenInput)
  138. status = forms.ChoiceField(choices=add_blank_choice(SITE_STATUS_CHOICES), required=False, initial='')
  139. region = TreeNodeChoiceField(queryset=Region.objects.all(), required=False)
  140. tenant = forms.ModelChoiceField(queryset=Tenant.objects.all(), required=False)
  141. asn = forms.IntegerField(min_value=1, max_value=4294967295, required=False, label='ASN')
  142. description = forms.CharField(max_length=100, required=False)
  143. time_zone = TimeZoneFormField(required=False)
  144. class Meta:
  145. nullable_fields = ['region', 'tenant', 'asn', 'description', 'time_zone']
  146. def site_status_choices():
  147. status_counts = {}
  148. for status in Site.objects.values('status').annotate(count=Count('status')).order_by('status'):
  149. status_counts[status['status']] = status['count']
  150. return [(s[0], '{} ({})'.format(s[1], status_counts.get(s[0], 0))) for s in SITE_STATUS_CHOICES]
  151. class SiteFilterForm(BootstrapMixin, CustomFieldFilterForm):
  152. model = Site
  153. q = forms.CharField(required=False, label='Search')
  154. status = forms.MultipleChoiceField(choices=site_status_choices, required=False)
  155. region = FilterTreeNodeMultipleChoiceField(
  156. queryset=Region.objects.annotate(filter_count=Count('sites')),
  157. to_field_name='slug',
  158. required=False,
  159. )
  160. tenant = FilterChoiceField(
  161. queryset=Tenant.objects.annotate(filter_count=Count('sites')),
  162. to_field_name='slug',
  163. null_label='-- None --'
  164. )
  165. #
  166. # Rack groups
  167. #
  168. class RackGroupForm(BootstrapMixin, forms.ModelForm):
  169. slug = SlugField()
  170. class Meta:
  171. model = RackGroup
  172. fields = ['site', 'name', 'slug']
  173. class RackGroupCSVForm(forms.ModelForm):
  174. site = forms.ModelChoiceField(
  175. queryset=Site.objects.all(),
  176. to_field_name='name',
  177. help_text='Name of parent site',
  178. error_messages={
  179. 'invalid_choice': 'Site not found.',
  180. }
  181. )
  182. class Meta:
  183. model = RackGroup
  184. fields = RackGroup.csv_headers
  185. help_texts = {
  186. 'name': 'Name of rack group',
  187. 'slug': 'URL-friendly slug',
  188. }
  189. class RackGroupFilterForm(BootstrapMixin, forms.Form):
  190. site = FilterChoiceField(queryset=Site.objects.annotate(filter_count=Count('rack_groups')), to_field_name='slug')
  191. #
  192. # Rack roles
  193. #
  194. class RackRoleForm(BootstrapMixin, forms.ModelForm):
  195. slug = SlugField()
  196. class Meta:
  197. model = RackRole
  198. fields = ['name', 'slug', 'color']
  199. class RackRoleCSVForm(forms.ModelForm):
  200. slug = SlugField()
  201. class Meta:
  202. model = RackRole
  203. fields = RackRole.csv_headers
  204. help_texts = {
  205. 'name': 'Name of rack role',
  206. 'color': 'RGB color in hexadecimal (e.g. 00ff00)'
  207. }
  208. #
  209. # Racks
  210. #
  211. class RackForm(BootstrapMixin, TenancyForm, CustomFieldForm):
  212. group = ChainedModelChoiceField(
  213. queryset=RackGroup.objects.all(),
  214. chains=(
  215. ('site', 'site'),
  216. ),
  217. required=False,
  218. widget=APISelect(
  219. api_url='/api/dcim/rack-groups/?site_id={{site}}',
  220. )
  221. )
  222. comments = CommentField()
  223. class Meta:
  224. model = Rack
  225. fields = [
  226. 'site', 'group', 'name', 'facility_id', 'tenant_group', 'tenant', 'role', 'serial', 'type', 'width',
  227. 'u_height', 'desc_units', 'comments',
  228. ]
  229. help_texts = {
  230. 'site': "The site at which the rack exists",
  231. 'name': "Organizational rack name",
  232. 'facility_id': "The unique rack ID assigned by the facility",
  233. 'u_height': "Height in rack units",
  234. }
  235. widgets = {
  236. 'site': forms.Select(attrs={'filter-for': 'group'}),
  237. }
  238. class RackCSVForm(forms.ModelForm):
  239. site = forms.ModelChoiceField(
  240. queryset=Site.objects.all(),
  241. to_field_name='name',
  242. help_text='Name of parent site',
  243. error_messages={
  244. 'invalid_choice': 'Site not found.',
  245. }
  246. )
  247. group_name = forms.CharField(
  248. help_text='Name of rack group',
  249. required=False
  250. )
  251. tenant = forms.ModelChoiceField(
  252. queryset=Tenant.objects.all(),
  253. required=False,
  254. to_field_name='name',
  255. help_text='Name of assigned tenant',
  256. error_messages={
  257. 'invalid_choice': 'Tenant not found.',
  258. }
  259. )
  260. role = forms.ModelChoiceField(
  261. queryset=RackRole.objects.all(),
  262. required=False,
  263. to_field_name='name',
  264. help_text='Name of assigned role',
  265. error_messages={
  266. 'invalid_choice': 'Role not found.',
  267. }
  268. )
  269. type = CSVChoiceField(
  270. choices=RACK_TYPE_CHOICES,
  271. required=False,
  272. help_text='Rack type'
  273. )
  274. width = forms.ChoiceField(
  275. choices=(
  276. (RACK_WIDTH_19IN, '19'),
  277. (RACK_WIDTH_23IN, '23'),
  278. ),
  279. help_text='Rail-to-rail width (in inches)'
  280. )
  281. class Meta:
  282. model = Rack
  283. fields = Rack.csv_headers
  284. help_texts = {
  285. 'name': 'Rack name',
  286. 'u_height': 'Height in rack units',
  287. }
  288. def clean(self):
  289. super(RackCSVForm, self).clean()
  290. site = self.cleaned_data.get('site')
  291. group_name = self.cleaned_data.get('group_name')
  292. # Validate rack group
  293. if group_name:
  294. try:
  295. self.instance.group = RackGroup.objects.get(site=site, name=group_name)
  296. except RackGroup.DoesNotExist:
  297. raise forms.ValidationError("Rack group {} not found for site {}".format(group_name, site))
  298. class RackBulkEditForm(BootstrapMixin, CustomFieldBulkEditForm):
  299. pk = forms.ModelMultipleChoiceField(queryset=Rack.objects.all(), widget=forms.MultipleHiddenInput)
  300. site = forms.ModelChoiceField(queryset=Site.objects.all(), required=False, label='Site')
  301. group = forms.ModelChoiceField(queryset=RackGroup.objects.all(), required=False, label='Group')
  302. tenant = forms.ModelChoiceField(queryset=Tenant.objects.all(), required=False)
  303. role = forms.ModelChoiceField(queryset=RackRole.objects.all(), required=False)
  304. serial = forms.CharField(max_length=50, required=False, label='Serial Number')
  305. type = forms.ChoiceField(choices=add_blank_choice(RACK_TYPE_CHOICES), required=False, label='Type')
  306. width = forms.ChoiceField(choices=add_blank_choice(RACK_WIDTH_CHOICES), required=False, label='Width')
  307. u_height = forms.IntegerField(required=False, label='Height (U)')
  308. desc_units = forms.NullBooleanField(required=False, widget=BulkEditNullBooleanSelect, label='Descending units')
  309. comments = CommentField(widget=SmallTextarea)
  310. class Meta:
  311. nullable_fields = ['group', 'tenant', 'role', 'serial', 'comments']
  312. class RackFilterForm(BootstrapMixin, CustomFieldFilterForm):
  313. model = Rack
  314. q = forms.CharField(required=False, label='Search')
  315. site = FilterChoiceField(
  316. queryset=Site.objects.annotate(filter_count=Count('racks')),
  317. to_field_name='slug'
  318. )
  319. group_id = FilterChoiceField(
  320. queryset=RackGroup.objects.select_related('site').annotate(filter_count=Count('racks')),
  321. label='Rack group',
  322. null_label='-- None --'
  323. )
  324. tenant = FilterChoiceField(
  325. queryset=Tenant.objects.annotate(filter_count=Count('racks')),
  326. to_field_name='slug',
  327. null_label='-- None --'
  328. )
  329. role = FilterChoiceField(
  330. queryset=RackRole.objects.annotate(filter_count=Count('racks')),
  331. to_field_name='slug',
  332. null_label='-- None --'
  333. )
  334. #
  335. # Rack reservations
  336. #
  337. class RackReservationForm(BootstrapMixin, TenancyForm, forms.ModelForm):
  338. units = SimpleArrayField(forms.IntegerField(), widget=ArrayFieldSelectMultiple(attrs={'size': 10}))
  339. user = forms.ModelChoiceField(queryset=User.objects.order_by('username'))
  340. class Meta:
  341. model = RackReservation
  342. fields = ['units', 'user', 'tenant_group', 'tenant', 'description']
  343. def __init__(self, *args, **kwargs):
  344. super(RackReservationForm, self).__init__(*args, **kwargs)
  345. # Populate rack unit choices
  346. self.fields['units'].widget.choices = self._get_unit_choices()
  347. def _get_unit_choices(self):
  348. rack = self.instance.rack
  349. reserved_units = []
  350. for resv in rack.reservations.exclude(pk=self.instance.pk):
  351. for u in resv.units:
  352. reserved_units.append(u)
  353. unit_choices = [(u, {'label': str(u), 'disabled': u in reserved_units}) for u in rack.units]
  354. return unit_choices
  355. class RackReservationFilterForm(BootstrapMixin, forms.Form):
  356. q = forms.CharField(required=False, label='Search')
  357. site = FilterChoiceField(
  358. queryset=Site.objects.annotate(filter_count=Count('racks__reservations')),
  359. to_field_name='slug'
  360. )
  361. group_id = FilterChoiceField(
  362. queryset=RackGroup.objects.select_related('site').annotate(filter_count=Count('racks__reservations')),
  363. label='Rack group',
  364. null_label='-- None --'
  365. )
  366. tenant = FilterChoiceField(
  367. queryset=Tenant.objects.annotate(filter_count=Count('rackreservations')),
  368. to_field_name='slug',
  369. null_label='-- None --'
  370. )
  371. class RackReservationBulkEditForm(BootstrapMixin, BulkEditForm):
  372. pk = forms.ModelMultipleChoiceField(queryset=RackReservation.objects.all(), widget=forms.MultipleHiddenInput)
  373. user = forms.ModelChoiceField(queryset=User.objects.order_by('username'), required=False)
  374. tenant = forms.ModelChoiceField(queryset=Tenant.objects.all(), required=False)
  375. description = forms.CharField(max_length=100, required=False)
  376. class Meta:
  377. nullable_fields = []
  378. #
  379. # Manufacturers
  380. #
  381. class ManufacturerForm(BootstrapMixin, forms.ModelForm):
  382. slug = SlugField()
  383. class Meta:
  384. model = Manufacturer
  385. fields = ['name', 'slug']
  386. class ManufacturerCSVForm(forms.ModelForm):
  387. class Meta:
  388. model = Manufacturer
  389. fields = Manufacturer.csv_headers
  390. help_texts = {
  391. 'name': 'Manufacturer name',
  392. 'slug': 'URL-friendly slug',
  393. }
  394. #
  395. # Device types
  396. #
  397. class DeviceTypeForm(BootstrapMixin, CustomFieldForm):
  398. slug = SlugField(slug_source='model')
  399. class Meta:
  400. model = DeviceType
  401. fields = ['manufacturer', 'model', 'slug', 'part_number', 'u_height', 'is_full_depth', 'is_console_server',
  402. 'is_pdu', 'is_network_device', 'subdevice_role', 'interface_ordering', 'comments']
  403. labels = {
  404. 'interface_ordering': 'Order interfaces by',
  405. }
  406. class DeviceTypeCSVForm(forms.ModelForm):
  407. manufacturer = forms.ModelChoiceField(
  408. queryset=Manufacturer.objects.all(),
  409. required=True,
  410. to_field_name='name',
  411. help_text='Manufacturer name',
  412. error_messages={
  413. 'invalid_choice': 'Manufacturer not found.',
  414. }
  415. )
  416. subdevice_role = CSVChoiceField(
  417. choices=SUBDEVICE_ROLE_CHOICES,
  418. required=False,
  419. help_text='Parent/child status'
  420. )
  421. interface_ordering = CSVChoiceField(
  422. choices=IFACE_ORDERING_CHOICES,
  423. required=False,
  424. help_text='Interface ordering'
  425. )
  426. class Meta:
  427. model = DeviceType
  428. fields = DeviceType.csv_headers
  429. help_texts = {
  430. 'model': 'Model name',
  431. 'slug': 'URL-friendly slug',
  432. }
  433. class DeviceTypeBulkEditForm(BootstrapMixin, CustomFieldBulkEditForm):
  434. pk = forms.ModelMultipleChoiceField(queryset=DeviceType.objects.all(), widget=forms.MultipleHiddenInput)
  435. manufacturer = forms.ModelChoiceField(queryset=Manufacturer.objects.all(), required=False)
  436. u_height = forms.IntegerField(min_value=1, required=False)
  437. is_full_depth = forms.NullBooleanField(required=False, widget=BulkEditNullBooleanSelect, label='Is full depth')
  438. interface_ordering = forms.ChoiceField(choices=add_blank_choice(IFACE_ORDERING_CHOICES), required=False)
  439. is_console_server = forms.NullBooleanField(
  440. required=False, widget=BulkEditNullBooleanSelect, label='Is a console server'
  441. )
  442. is_pdu = forms.NullBooleanField(required=False, widget=BulkEditNullBooleanSelect, label='Is a PDU')
  443. is_network_device = forms.NullBooleanField(
  444. required=False, widget=BulkEditNullBooleanSelect, label='Is a network device'
  445. )
  446. class Meta:
  447. nullable_fields = []
  448. class DeviceTypeFilterForm(BootstrapMixin, CustomFieldFilterForm):
  449. model = DeviceType
  450. q = forms.CharField(required=False, label='Search')
  451. manufacturer = FilterChoiceField(
  452. queryset=Manufacturer.objects.annotate(filter_count=Count('device_types')),
  453. to_field_name='slug'
  454. )
  455. is_console_server = forms.BooleanField(
  456. required=False, label='Is a console server', widget=forms.CheckboxInput(attrs={'value': 'True'}))
  457. is_pdu = forms.BooleanField(
  458. required=False, label='Is a PDU', widget=forms.CheckboxInput(attrs={'value': 'True'})
  459. )
  460. is_network_device = forms.BooleanField(
  461. required=False, label='Is a network device', widget=forms.CheckboxInput(attrs={'value': 'True'})
  462. )
  463. subdevice_role = forms.NullBooleanField(
  464. required=False, label='Subdevice role', widget=forms.Select(choices=(
  465. ('', '---------'),
  466. (SUBDEVICE_ROLE_PARENT, 'Parent'),
  467. (SUBDEVICE_ROLE_CHILD, 'Child'),
  468. ))
  469. )
  470. #
  471. # Device component templates
  472. #
  473. class ConsolePortTemplateForm(BootstrapMixin, forms.ModelForm):
  474. class Meta:
  475. model = ConsolePortTemplate
  476. fields = ['device_type', 'name']
  477. widgets = {
  478. 'device_type': forms.HiddenInput(),
  479. }
  480. class ConsolePortTemplateCreateForm(ComponentForm):
  481. name_pattern = ExpandableNameField(label='Name')
  482. class ConsoleServerPortTemplateForm(BootstrapMixin, forms.ModelForm):
  483. class Meta:
  484. model = ConsoleServerPortTemplate
  485. fields = ['device_type', 'name']
  486. widgets = {
  487. 'device_type': forms.HiddenInput(),
  488. }
  489. class ConsoleServerPortTemplateCreateForm(ComponentForm):
  490. name_pattern = ExpandableNameField(label='Name')
  491. class PowerPortTemplateForm(BootstrapMixin, forms.ModelForm):
  492. class Meta:
  493. model = PowerPortTemplate
  494. fields = ['device_type', 'name']
  495. widgets = {
  496. 'device_type': forms.HiddenInput(),
  497. }
  498. class PowerPortTemplateCreateForm(ComponentForm):
  499. name_pattern = ExpandableNameField(label='Name')
  500. class PowerOutletTemplateForm(BootstrapMixin, forms.ModelForm):
  501. class Meta:
  502. model = PowerOutletTemplate
  503. fields = ['device_type', 'name']
  504. widgets = {
  505. 'device_type': forms.HiddenInput(),
  506. }
  507. class PowerOutletTemplateCreateForm(ComponentForm):
  508. name_pattern = ExpandableNameField(label='Name')
  509. class InterfaceTemplateForm(BootstrapMixin, forms.ModelForm):
  510. class Meta:
  511. model = InterfaceTemplate
  512. fields = ['device_type', 'name', 'form_factor', 'mgmt_only']
  513. widgets = {
  514. 'device_type': forms.HiddenInput(),
  515. }
  516. class InterfaceTemplateCreateForm(ComponentForm):
  517. name_pattern = ExpandableNameField(label='Name')
  518. form_factor = forms.ChoiceField(choices=IFACE_FF_CHOICES)
  519. mgmt_only = forms.BooleanField(required=False, label='OOB Management')
  520. class InterfaceTemplateBulkEditForm(BootstrapMixin, BulkEditForm):
  521. pk = forms.ModelMultipleChoiceField(queryset=InterfaceTemplate.objects.all(), widget=forms.MultipleHiddenInput)
  522. form_factor = forms.ChoiceField(choices=add_blank_choice(IFACE_FF_CHOICES), required=False)
  523. mgmt_only = forms.NullBooleanField(required=False, widget=BulkEditNullBooleanSelect, label='Management only')
  524. class Meta:
  525. nullable_fields = []
  526. class DeviceBayTemplateForm(BootstrapMixin, forms.ModelForm):
  527. class Meta:
  528. model = DeviceBayTemplate
  529. fields = ['device_type', 'name']
  530. widgets = {
  531. 'device_type': forms.HiddenInput(),
  532. }
  533. class DeviceBayTemplateCreateForm(ComponentForm):
  534. name_pattern = ExpandableNameField(label='Name')
  535. #
  536. # Device roles
  537. #
  538. class DeviceRoleForm(BootstrapMixin, forms.ModelForm):
  539. slug = SlugField()
  540. class Meta:
  541. model = DeviceRole
  542. fields = ['name', 'slug', 'color', 'vm_role']
  543. class DeviceRoleCSVForm(forms.ModelForm):
  544. slug = SlugField()
  545. class Meta:
  546. model = DeviceRole
  547. fields = DeviceRole.csv_headers
  548. help_texts = {
  549. 'name': 'Name of device role',
  550. 'color': 'RGB color in hexadecimal (e.g. 00ff00)'
  551. }
  552. #
  553. # Platforms
  554. #
  555. class PlatformForm(BootstrapMixin, forms.ModelForm):
  556. slug = SlugField()
  557. class Meta:
  558. model = Platform
  559. fields = ['name', 'slug', 'manufacturer', 'napalm_driver', 'rpc_client']
  560. class PlatformCSVForm(forms.ModelForm):
  561. slug = SlugField()
  562. class Meta:
  563. model = Platform
  564. fields = Platform.csv_headers
  565. help_texts = {
  566. 'name': 'Platform name',
  567. 'manufacturer': 'Manufacturer name',
  568. }
  569. #
  570. # Devices
  571. #
  572. class DeviceForm(BootstrapMixin, TenancyForm, CustomFieldForm):
  573. site = forms.ModelChoiceField(
  574. queryset=Site.objects.all(),
  575. widget=forms.Select(
  576. attrs={'filter-for': 'rack'}
  577. )
  578. )
  579. rack = ChainedModelChoiceField(
  580. queryset=Rack.objects.all(),
  581. chains=(
  582. ('site', 'site'),
  583. ),
  584. required=False,
  585. widget=APISelect(
  586. api_url='/api/dcim/racks/?site_id={{site}}',
  587. display_field='display_name',
  588. attrs={'filter-for': 'position'}
  589. )
  590. )
  591. position = forms.TypedChoiceField(
  592. required=False,
  593. empty_value=None,
  594. help_text="The lowest-numbered unit occupied by the device",
  595. widget=APISelect(
  596. api_url='/api/dcim/racks/{{rack}}/units/?face={{face}}',
  597. disabled_indicator='device'
  598. )
  599. )
  600. manufacturer = forms.ModelChoiceField(
  601. queryset=Manufacturer.objects.all(),
  602. widget=forms.Select(
  603. attrs={'filter-for': 'device_type'}
  604. )
  605. )
  606. device_type = ChainedModelChoiceField(
  607. queryset=DeviceType.objects.all(),
  608. chains=(
  609. ('manufacturer', 'manufacturer'),
  610. ),
  611. label='Device type',
  612. widget=APISelect(
  613. api_url='/api/dcim/device-types/?manufacturer_id={{manufacturer}}',
  614. display_field='model'
  615. )
  616. )
  617. comments = CommentField()
  618. class Meta:
  619. model = Device
  620. fields = [
  621. 'name', 'device_role', 'device_type', 'serial', 'asset_tag', 'site', 'rack', 'position', 'face', 'status',
  622. 'platform', 'primary_ip4', 'primary_ip6', 'tenant_group', 'tenant', 'comments',
  623. ]
  624. help_texts = {
  625. 'device_role': "The function this device serves",
  626. 'serial': "Chassis serial number",
  627. }
  628. widgets = {
  629. 'face': forms.Select(attrs={'filter-for': 'position'}),
  630. }
  631. def __init__(self, *args, **kwargs):
  632. # Initialize helper selectors
  633. instance = kwargs.get('instance')
  634. # Using hasattr() instead of "is not None" to avoid RelatedObjectDoesNotExist on required field
  635. if instance and hasattr(instance, 'device_type'):
  636. initial = kwargs.get('initial', {}).copy()
  637. initial['manufacturer'] = instance.device_type.manufacturer
  638. kwargs['initial'] = initial
  639. super(DeviceForm, self).__init__(*args, **kwargs)
  640. if self.instance.pk:
  641. # Compile list of choices for primary IPv4 and IPv6 addresses
  642. for family in [4, 6]:
  643. ip_choices = [(None, '---------')]
  644. # Gather PKs of all interfaces belonging to this Device or a peer VirtualChassis member
  645. interface_ids = self.instance.vc_interfaces.values('pk')
  646. # Collect interface IPs
  647. interface_ips = IPAddress.objects.select_related('interface').filter(
  648. family=family, interface_id__in=interface_ids
  649. )
  650. if interface_ips:
  651. ip_list = [(ip.id, '{} ({})'.format(ip.address, ip.interface)) for ip in interface_ips]
  652. ip_choices.append(('Interface IPs', ip_list))
  653. # Collect NAT IPs
  654. nat_ips = IPAddress.objects.select_related('nat_inside').filter(
  655. family=family, nat_inside__interface__in=interface_ids
  656. )
  657. if nat_ips:
  658. ip_list = [(ip.id, '{} ({})'.format(ip.address, ip.nat_inside.address)) for ip in nat_ips]
  659. ip_choices.append(('NAT IPs', ip_list))
  660. self.fields['primary_ip{}'.format(family)].choices = ip_choices
  661. # If editing an existing device, exclude it from the list of occupied rack units. This ensures that a device
  662. # can be flipped from one face to another.
  663. self.fields['position'].widget.attrs['api-url'] += '&exclude={}'.format(self.instance.pk)
  664. # Limit platform by manufacturer
  665. self.fields['platform'].queryset = Platform.objects.filter(
  666. Q(manufacturer__isnull=True) | Q(manufacturer=self.instance.device_type.manufacturer)
  667. )
  668. else:
  669. # An object that doesn't exist yet can't have any IPs assigned to it
  670. self.fields['primary_ip4'].choices = []
  671. self.fields['primary_ip4'].widget.attrs['readonly'] = True
  672. self.fields['primary_ip6'].choices = []
  673. self.fields['primary_ip6'].widget.attrs['readonly'] = True
  674. # Rack position
  675. pk = self.instance.pk if self.instance.pk else None
  676. try:
  677. if self.is_bound and self.data.get('rack') and str(self.data.get('face')):
  678. position_choices = Rack.objects.get(pk=self.data['rack']) \
  679. .get_rack_units(face=self.data.get('face'), exclude=pk)
  680. elif self.initial.get('rack') and str(self.initial.get('face')):
  681. position_choices = Rack.objects.get(pk=self.initial['rack']) \
  682. .get_rack_units(face=self.initial.get('face'), exclude=pk)
  683. else:
  684. position_choices = []
  685. except Rack.DoesNotExist:
  686. position_choices = []
  687. self.fields['position'].choices = [('', '---------')] + [
  688. (p['id'], {
  689. 'label': p['name'],
  690. 'disabled': bool(p['device'] and p['id'] != self.initial.get('position')),
  691. }) for p in position_choices
  692. ]
  693. # Disable rack assignment if this is a child device installed in a parent device
  694. if pk and self.instance.device_type.is_child_device and hasattr(self.instance, 'parent_bay'):
  695. self.fields['site'].disabled = True
  696. self.fields['rack'].disabled = True
  697. self.initial['site'] = self.instance.parent_bay.device.site_id
  698. self.initial['rack'] = self.instance.parent_bay.device.rack_id
  699. class BaseDeviceCSVForm(forms.ModelForm):
  700. device_role = forms.ModelChoiceField(
  701. queryset=DeviceRole.objects.all(),
  702. to_field_name='name',
  703. help_text='Name of assigned role',
  704. error_messages={
  705. 'invalid_choice': 'Invalid device role.',
  706. }
  707. )
  708. tenant = forms.ModelChoiceField(
  709. queryset=Tenant.objects.all(),
  710. required=False,
  711. to_field_name='name',
  712. help_text='Name of assigned tenant',
  713. error_messages={
  714. 'invalid_choice': 'Tenant not found.',
  715. }
  716. )
  717. manufacturer = forms.ModelChoiceField(
  718. queryset=Manufacturer.objects.all(),
  719. to_field_name='name',
  720. help_text='Device type manufacturer',
  721. error_messages={
  722. 'invalid_choice': 'Invalid manufacturer.',
  723. }
  724. )
  725. model_name = forms.CharField(
  726. help_text='Device type model name'
  727. )
  728. platform = forms.ModelChoiceField(
  729. queryset=Platform.objects.all(),
  730. required=False,
  731. to_field_name='name',
  732. help_text='Name of assigned platform',
  733. error_messages={
  734. 'invalid_choice': 'Invalid platform.',
  735. }
  736. )
  737. status = CSVChoiceField(
  738. choices=DEVICE_STATUS_CHOICES,
  739. help_text='Operational status'
  740. )
  741. class Meta:
  742. fields = []
  743. model = Device
  744. help_texts = {
  745. 'name': 'Device name',
  746. }
  747. def clean(self):
  748. super(BaseDeviceCSVForm, self).clean()
  749. manufacturer = self.cleaned_data.get('manufacturer')
  750. model_name = self.cleaned_data.get('model_name')
  751. # Validate device type
  752. if manufacturer and model_name:
  753. try:
  754. self.instance.device_type = DeviceType.objects.get(manufacturer=manufacturer, model=model_name)
  755. except DeviceType.DoesNotExist:
  756. raise forms.ValidationError("Device type {} {} not found".format(manufacturer, model_name))
  757. class DeviceCSVForm(BaseDeviceCSVForm):
  758. site = forms.ModelChoiceField(
  759. queryset=Site.objects.all(),
  760. to_field_name='name',
  761. help_text='Name of parent site',
  762. error_messages={
  763. 'invalid_choice': 'Invalid site name.',
  764. }
  765. )
  766. rack_group = forms.CharField(
  767. required=False,
  768. help_text='Parent rack\'s group (if any)'
  769. )
  770. rack_name = forms.CharField(
  771. required=False,
  772. help_text='Name of parent rack'
  773. )
  774. face = CSVChoiceField(
  775. choices=RACK_FACE_CHOICES,
  776. required=False,
  777. help_text='Mounted rack face'
  778. )
  779. cluster = forms.ModelChoiceField(
  780. queryset=Cluster.objects.all(),
  781. to_field_name='name',
  782. required=False,
  783. help_text='Virtualization cluster',
  784. error_messages={
  785. 'invalid_choice': 'Invalid cluster name.',
  786. }
  787. )
  788. class Meta(BaseDeviceCSVForm.Meta):
  789. fields = [
  790. 'name', 'device_role', 'tenant', 'manufacturer', 'model_name', 'platform', 'serial', 'asset_tag', 'status',
  791. 'site', 'rack_group', 'rack_name', 'position', 'face', 'cluster', 'comments',
  792. ]
  793. def clean(self):
  794. super(DeviceCSVForm, self).clean()
  795. site = self.cleaned_data.get('site')
  796. rack_group = self.cleaned_data.get('rack_group')
  797. rack_name = self.cleaned_data.get('rack_name')
  798. # Validate rack
  799. if site and rack_group and rack_name:
  800. try:
  801. self.instance.rack = Rack.objects.get(site=site, group__name=rack_group, name=rack_name)
  802. except Rack.DoesNotExist:
  803. raise forms.ValidationError("Rack {} not found in site {} group {}".format(rack_name, site, rack_group))
  804. elif site and rack_name:
  805. try:
  806. self.instance.rack = Rack.objects.get(site=site, group__isnull=True, name=rack_name)
  807. except Rack.DoesNotExist:
  808. raise forms.ValidationError("Rack {} not found in site {} (no group)".format(rack_name, site))
  809. class ChildDeviceCSVForm(BaseDeviceCSVForm):
  810. parent = FlexibleModelChoiceField(
  811. queryset=Device.objects.all(),
  812. to_field_name='name',
  813. help_text='Name or ID of parent device',
  814. error_messages={
  815. 'invalid_choice': 'Parent device not found.',
  816. }
  817. )
  818. device_bay_name = forms.CharField(
  819. help_text='Name of device bay',
  820. )
  821. cluster = forms.ModelChoiceField(
  822. queryset=Cluster.objects.all(),
  823. to_field_name='name',
  824. required=False,
  825. help_text='Virtualization cluster',
  826. error_messages={
  827. 'invalid_choice': 'Invalid cluster name.',
  828. }
  829. )
  830. class Meta(BaseDeviceCSVForm.Meta):
  831. fields = [
  832. 'name', 'device_role', 'tenant', 'manufacturer', 'model_name', 'platform', 'serial', 'asset_tag', 'status',
  833. 'parent', 'device_bay_name', 'cluster', 'comments',
  834. ]
  835. def clean(self):
  836. super(ChildDeviceCSVForm, self).clean()
  837. parent = self.cleaned_data.get('parent')
  838. device_bay_name = self.cleaned_data.get('device_bay_name')
  839. # Validate device bay
  840. if parent and device_bay_name:
  841. try:
  842. self.instance.parent_bay = DeviceBay.objects.get(device=parent, name=device_bay_name)
  843. # Inherit site and rack from parent device
  844. self.instance.site = parent.site
  845. self.instance.rack = parent.rack
  846. except DeviceBay.DoesNotExist:
  847. raise forms.ValidationError("Parent device/bay ({} {}) not found".format(parent, device_bay_name))
  848. class DeviceBulkEditForm(BootstrapMixin, CustomFieldBulkEditForm):
  849. pk = forms.ModelMultipleChoiceField(queryset=Device.objects.all(), widget=forms.MultipleHiddenInput)
  850. device_type = forms.ModelChoiceField(queryset=DeviceType.objects.all(), required=False, label='Type')
  851. device_role = forms.ModelChoiceField(queryset=DeviceRole.objects.all(), required=False, label='Role')
  852. tenant = forms.ModelChoiceField(queryset=Tenant.objects.all(), required=False)
  853. platform = forms.ModelChoiceField(queryset=Platform.objects.all(), required=False)
  854. status = forms.ChoiceField(choices=add_blank_choice(DEVICE_STATUS_CHOICES), required=False, initial='')
  855. serial = forms.CharField(max_length=50, required=False, label='Serial Number')
  856. class Meta:
  857. nullable_fields = ['tenant', 'platform', 'serial']
  858. def device_status_choices():
  859. status_counts = {}
  860. for status in Device.objects.values('status').annotate(count=Count('status')).order_by('status'):
  861. status_counts[status['status']] = status['count']
  862. return [(s[0], '{} ({})'.format(s[1], status_counts.get(s[0], 0))) for s in DEVICE_STATUS_CHOICES]
  863. class DeviceFilterForm(BootstrapMixin, CustomFieldFilterForm):
  864. model = Device
  865. q = forms.CharField(required=False, label='Search')
  866. site = FilterChoiceField(
  867. queryset=Site.objects.annotate(filter_count=Count('devices')),
  868. to_field_name='slug',
  869. )
  870. rack_group_id = FilterChoiceField(
  871. queryset=RackGroup.objects.select_related('site').annotate(filter_count=Count('racks__devices')),
  872. label='Rack group',
  873. )
  874. rack_id = FilterChoiceField(
  875. queryset=Rack.objects.annotate(filter_count=Count('devices')),
  876. label='Rack',
  877. null_label='-- None --',
  878. )
  879. role = FilterChoiceField(
  880. queryset=DeviceRole.objects.annotate(filter_count=Count('devices')),
  881. to_field_name='slug',
  882. )
  883. tenant = FilterChoiceField(
  884. queryset=Tenant.objects.annotate(filter_count=Count('devices')),
  885. to_field_name='slug',
  886. null_label='-- None --',
  887. )
  888. manufacturer_id = FilterChoiceField(queryset=Manufacturer.objects.all(), label='Manufacturer')
  889. device_type_id = FilterChoiceField(
  890. queryset=DeviceType.objects.select_related('manufacturer').order_by('model').annotate(
  891. filter_count=Count('instances'),
  892. ),
  893. label='Model',
  894. )
  895. platform = FilterChoiceField(
  896. queryset=Platform.objects.annotate(filter_count=Count('devices')),
  897. to_field_name='slug',
  898. null_label='-- None --',
  899. )
  900. status = forms.MultipleChoiceField(choices=device_status_choices, required=False)
  901. mac_address = forms.CharField(required=False, label='MAC address')
  902. #
  903. # Bulk device component creation
  904. #
  905. class DeviceBulkAddComponentForm(BootstrapMixin, forms.Form):
  906. pk = forms.ModelMultipleChoiceField(queryset=Device.objects.all(), widget=forms.MultipleHiddenInput)
  907. name_pattern = ExpandableNameField(label='Name')
  908. class DeviceBulkAddInterfaceForm(DeviceBulkAddComponentForm):
  909. form_factor = forms.ChoiceField(choices=IFACE_FF_CHOICES)
  910. enabled = forms.BooleanField(required=False, initial=True)
  911. mtu = forms.IntegerField(required=False, min_value=1, max_value=32767, label='MTU')
  912. mgmt_only = forms.BooleanField(required=False, label='OOB Management')
  913. description = forms.CharField(max_length=100, required=False)
  914. #
  915. # Console ports
  916. #
  917. class ConsolePortForm(BootstrapMixin, forms.ModelForm):
  918. class Meta:
  919. model = ConsolePort
  920. fields = ['device', 'name']
  921. widgets = {
  922. 'device': forms.HiddenInput(),
  923. }
  924. class ConsolePortCreateForm(ComponentForm):
  925. name_pattern = ExpandableNameField(label='Name')
  926. class ConsoleConnectionCSVForm(forms.ModelForm):
  927. console_server = FlexibleModelChoiceField(
  928. queryset=Device.objects.filter(device_type__is_console_server=True),
  929. to_field_name='name',
  930. help_text='Console server name or ID',
  931. error_messages={
  932. 'invalid_choice': 'Console server not found',
  933. }
  934. )
  935. cs_port = forms.CharField(
  936. help_text='Console server port name'
  937. )
  938. device = FlexibleModelChoiceField(
  939. queryset=Device.objects.all(),
  940. to_field_name='name',
  941. help_text='Device name or ID',
  942. error_messages={
  943. 'invalid_choice': 'Device not found',
  944. }
  945. )
  946. console_port = forms.CharField(
  947. help_text='Console port name'
  948. )
  949. connection_status = CSVChoiceField(
  950. choices=CONNECTION_STATUS_CHOICES,
  951. help_text='Connection status'
  952. )
  953. class Meta:
  954. model = ConsolePort
  955. fields = ['console_server', 'cs_port', 'device', 'console_port', 'connection_status']
  956. def clean_console_port(self):
  957. console_port_name = self.cleaned_data.get('console_port')
  958. if not self.cleaned_data.get('device') or not console_port_name:
  959. return None
  960. try:
  961. # Retrieve console port by name
  962. consoleport = ConsolePort.objects.get(
  963. device=self.cleaned_data['device'], name=console_port_name
  964. )
  965. # Check if the console port is already connected
  966. if consoleport.cs_port is not None:
  967. raise forms.ValidationError("{} {} is already connected".format(
  968. self.cleaned_data['device'], console_port_name
  969. ))
  970. except ConsolePort.DoesNotExist:
  971. raise forms.ValidationError("Invalid console port ({} {})".format(
  972. self.cleaned_data['device'], console_port_name
  973. ))
  974. self.instance = consoleport
  975. return consoleport
  976. def clean_cs_port(self):
  977. cs_port_name = self.cleaned_data.get('cs_port')
  978. if not self.cleaned_data.get('console_server') or not cs_port_name:
  979. return None
  980. try:
  981. # Retrieve console server port by name
  982. cs_port = ConsoleServerPort.objects.get(
  983. device=self.cleaned_data['console_server'], name=cs_port_name
  984. )
  985. # Check if the console server port is already connected
  986. if ConsolePort.objects.filter(cs_port=cs_port).count():
  987. raise forms.ValidationError("{} {} is already connected".format(
  988. self.cleaned_data['console_server'], cs_port_name
  989. ))
  990. except ConsoleServerPort.DoesNotExist:
  991. raise forms.ValidationError("Invalid console server port ({} {})".format(
  992. self.cleaned_data['console_server'], cs_port_name
  993. ))
  994. return cs_port
  995. class ConsolePortConnectionForm(BootstrapMixin, ChainedFieldsMixin, forms.ModelForm):
  996. site = forms.ModelChoiceField(
  997. queryset=Site.objects.all(),
  998. required=False,
  999. widget=forms.Select(
  1000. attrs={'filter-for': 'rack'}
  1001. )
  1002. )
  1003. rack = ChainedModelChoiceField(
  1004. queryset=Rack.objects.all(),
  1005. chains=(
  1006. ('site', 'site'),
  1007. ),
  1008. label='Rack',
  1009. required=False,
  1010. widget=APISelect(
  1011. api_url='/api/dcim/racks/?site_id={{site}}',
  1012. attrs={'filter-for': 'console_server', 'nullable': 'true'}
  1013. )
  1014. )
  1015. console_server = ChainedModelChoiceField(
  1016. queryset=Device.objects.filter(device_type__is_console_server=True),
  1017. chains=(
  1018. ('site', 'site'),
  1019. ('rack', 'rack'),
  1020. ),
  1021. label='Console Server',
  1022. required=False,
  1023. widget=APISelect(
  1024. api_url='/api/dcim/devices/?site_id={{site}}&rack_id={{rack}}&is_console_server=True',
  1025. display_field='display_name',
  1026. attrs={'filter-for': 'cs_port'}
  1027. )
  1028. )
  1029. livesearch = forms.CharField(
  1030. required=False,
  1031. label='Console Server',
  1032. widget=Livesearch(
  1033. query_key='q',
  1034. query_url='dcim-api:device-list',
  1035. field_to_update='console_server',
  1036. )
  1037. )
  1038. cs_port = ChainedModelChoiceField(
  1039. queryset=ConsoleServerPort.objects.all(),
  1040. chains=(
  1041. ('device', 'console_server'),
  1042. ),
  1043. label='Port',
  1044. widget=APISelect(
  1045. api_url='/api/dcim/console-server-ports/?device_id={{console_server}}',
  1046. disabled_indicator='connected_console',
  1047. )
  1048. )
  1049. class Meta:
  1050. model = ConsolePort
  1051. fields = ['site', 'rack', 'console_server', 'livesearch', 'cs_port', 'connection_status']
  1052. labels = {
  1053. 'cs_port': 'Port',
  1054. 'connection_status': 'Status',
  1055. }
  1056. def __init__(self, *args, **kwargs):
  1057. super(ConsolePortConnectionForm, self).__init__(*args, **kwargs)
  1058. if not self.instance.pk:
  1059. raise RuntimeError("ConsolePortConnectionForm must be initialized with an existing ConsolePort instance.")
  1060. #
  1061. # Console server ports
  1062. #
  1063. class ConsoleServerPortForm(BootstrapMixin, forms.ModelForm):
  1064. class Meta:
  1065. model = ConsoleServerPort
  1066. fields = ['device', 'name']
  1067. widgets = {
  1068. 'device': forms.HiddenInput(),
  1069. }
  1070. class ConsoleServerPortCreateForm(ComponentForm):
  1071. name_pattern = ExpandableNameField(label='Name')
  1072. class ConsoleServerPortConnectionForm(BootstrapMixin, ChainedFieldsMixin, forms.Form):
  1073. site = forms.ModelChoiceField(
  1074. queryset=Site.objects.all(),
  1075. required=False,
  1076. widget=forms.Select(
  1077. attrs={'filter-for': 'rack'}
  1078. )
  1079. )
  1080. rack = ChainedModelChoiceField(
  1081. queryset=Rack.objects.all(),
  1082. chains=(
  1083. ('site', 'site'),
  1084. ),
  1085. label='Rack',
  1086. required=False,
  1087. widget=APISelect(
  1088. api_url='/api/dcim/racks/?site_id={{site}}',
  1089. attrs={'filter-for': 'device', 'nullable': 'true'}
  1090. )
  1091. )
  1092. device = ChainedModelChoiceField(
  1093. queryset=Device.objects.all(),
  1094. chains=(
  1095. ('site', 'site'),
  1096. ('rack', 'rack'),
  1097. ),
  1098. label='Device',
  1099. required=False,
  1100. widget=APISelect(
  1101. api_url='/api/dcim/devices/?site_id={{site}}&rack_id={{rack}}',
  1102. display_field='display_name',
  1103. attrs={'filter-for': 'port'}
  1104. )
  1105. )
  1106. livesearch = forms.CharField(
  1107. required=False,
  1108. label='Device',
  1109. widget=Livesearch(
  1110. query_key='q',
  1111. query_url='dcim-api:device-list',
  1112. field_to_update='device'
  1113. )
  1114. )
  1115. port = ChainedModelChoiceField(
  1116. queryset=ConsolePort.objects.all(),
  1117. chains=(
  1118. ('device', 'device'),
  1119. ),
  1120. label='Port',
  1121. widget=APISelect(
  1122. api_url='/api/dcim/console-ports/?device_id={{device}}',
  1123. disabled_indicator='cs_port'
  1124. )
  1125. )
  1126. connection_status = forms.BooleanField(
  1127. required=False,
  1128. initial=CONNECTION_STATUS_CONNECTED,
  1129. label='Status',
  1130. widget=forms.Select(
  1131. choices=CONNECTION_STATUS_CHOICES
  1132. )
  1133. )
  1134. class Meta:
  1135. fields = ['site', 'rack', 'device', 'livesearch', 'port', 'connection_status']
  1136. labels = {
  1137. 'connection_status': 'Status',
  1138. }
  1139. class ConsoleServerPortBulkRenameForm(BulkRenameForm):
  1140. pk = forms.ModelMultipleChoiceField(queryset=ConsoleServerPort.objects.all(), widget=forms.MultipleHiddenInput)
  1141. class ConsoleServerPortBulkDisconnectForm(ConfirmationForm):
  1142. pk = forms.ModelMultipleChoiceField(queryset=ConsoleServerPort.objects.all(), widget=forms.MultipleHiddenInput)
  1143. #
  1144. # Power ports
  1145. #
  1146. class PowerPortForm(BootstrapMixin, forms.ModelForm):
  1147. class Meta:
  1148. model = PowerPort
  1149. fields = ['device', 'name']
  1150. widgets = {
  1151. 'device': forms.HiddenInput(),
  1152. }
  1153. class PowerPortCreateForm(ComponentForm):
  1154. name_pattern = ExpandableNameField(label='Name')
  1155. class PowerConnectionCSVForm(forms.ModelForm):
  1156. pdu = FlexibleModelChoiceField(
  1157. queryset=Device.objects.filter(device_type__is_pdu=True),
  1158. to_field_name='name',
  1159. help_text='PDU name or ID',
  1160. error_messages={
  1161. 'invalid_choice': 'PDU not found.',
  1162. }
  1163. )
  1164. power_outlet = forms.CharField(
  1165. help_text='Power outlet name'
  1166. )
  1167. device = FlexibleModelChoiceField(
  1168. queryset=Device.objects.all(),
  1169. to_field_name='name',
  1170. help_text='Device name or ID',
  1171. error_messages={
  1172. 'invalid_choice': 'Device not found',
  1173. }
  1174. )
  1175. power_port = forms.CharField(
  1176. help_text='Power port name'
  1177. )
  1178. connection_status = CSVChoiceField(
  1179. choices=CONNECTION_STATUS_CHOICES,
  1180. help_text='Connection status'
  1181. )
  1182. class Meta:
  1183. model = PowerPort
  1184. fields = ['pdu', 'power_outlet', 'device', 'power_port', 'connection_status']
  1185. def clean_power_port(self):
  1186. power_port_name = self.cleaned_data.get('power_port')
  1187. if not self.cleaned_data.get('device') or not power_port_name:
  1188. return None
  1189. try:
  1190. # Retrieve power port by name
  1191. powerport = PowerPort.objects.get(
  1192. device=self.cleaned_data['device'], name=power_port_name
  1193. )
  1194. # Check if the power port is already connected
  1195. if powerport.power_outlet is not None:
  1196. raise forms.ValidationError("{} {} is already connected".format(
  1197. self.cleaned_data['device'], power_port_name
  1198. ))
  1199. except PowerPort.DoesNotExist:
  1200. raise forms.ValidationError("Invalid power port ({} {})".format(
  1201. self.cleaned_data['device'], power_port_name
  1202. ))
  1203. self.instance = powerport
  1204. return powerport
  1205. def clean_power_outlet(self):
  1206. power_outlet_name = self.cleaned_data.get('power_outlet')
  1207. if not self.cleaned_data.get('pdu') or not power_outlet_name:
  1208. return None
  1209. try:
  1210. # Retrieve power outlet by name
  1211. power_outlet = PowerOutlet.objects.get(
  1212. device=self.cleaned_data['pdu'], name=power_outlet_name
  1213. )
  1214. # Check if the power outlet is already connected
  1215. if PowerPort.objects.filter(power_outlet=power_outlet).count():
  1216. raise forms.ValidationError("{} {} is already connected".format(
  1217. self.cleaned_data['pdu'], power_outlet_name
  1218. ))
  1219. except PowerOutlet.DoesNotExist:
  1220. raise forms.ValidationError("Invalid power outlet ({} {})".format(
  1221. self.cleaned_data['pdu'], power_outlet_name
  1222. ))
  1223. return power_outlet
  1224. class PowerPortConnectionForm(BootstrapMixin, ChainedFieldsMixin, forms.ModelForm):
  1225. site = forms.ModelChoiceField(
  1226. queryset=Site.objects.all(),
  1227. required=False,
  1228. widget=forms.Select(
  1229. attrs={'filter-for': 'rack'}
  1230. )
  1231. )
  1232. rack = ChainedModelChoiceField(
  1233. queryset=Rack.objects.all(),
  1234. chains=(
  1235. ('site', 'site'),
  1236. ),
  1237. label='Rack',
  1238. required=False,
  1239. widget=APISelect(
  1240. api_url='/api/dcim/racks/?site_id={{site}}',
  1241. attrs={'filter-for': 'pdu', 'nullable': 'true'}
  1242. )
  1243. )
  1244. pdu = ChainedModelChoiceField(
  1245. queryset=Device.objects.all(),
  1246. chains=(
  1247. ('site', 'site'),
  1248. ('rack', 'rack'),
  1249. ),
  1250. label='PDU',
  1251. required=False,
  1252. widget=APISelect(
  1253. api_url='/api/dcim/devices/?site_id={{site}}&rack_id={{rack}}&is_pdu=True',
  1254. display_field='display_name',
  1255. attrs={'filter-for': 'power_outlet'}
  1256. )
  1257. )
  1258. livesearch = forms.CharField(
  1259. required=False,
  1260. label='PDU',
  1261. widget=Livesearch(
  1262. query_key='q',
  1263. query_url='dcim-api:device-list',
  1264. field_to_update='pdu'
  1265. )
  1266. )
  1267. power_outlet = ChainedModelChoiceField(
  1268. queryset=PowerOutlet.objects.all(),
  1269. chains=(
  1270. ('device', 'pdu'),
  1271. ),
  1272. label='Outlet',
  1273. widget=APISelect(
  1274. api_url='/api/dcim/power-outlets/?device_id={{pdu}}',
  1275. disabled_indicator='connected_port'
  1276. )
  1277. )
  1278. class Meta:
  1279. model = PowerPort
  1280. fields = ['site', 'rack', 'pdu', 'livesearch', 'power_outlet', 'connection_status']
  1281. labels = {
  1282. 'power_outlet': 'Outlet',
  1283. 'connection_status': 'Status',
  1284. }
  1285. def __init__(self, *args, **kwargs):
  1286. super(PowerPortConnectionForm, self).__init__(*args, **kwargs)
  1287. if not self.instance.pk:
  1288. raise RuntimeError("PowerPortConnectionForm must be initialized with an existing PowerPort instance.")
  1289. #
  1290. # Power outlets
  1291. #
  1292. class PowerOutletForm(BootstrapMixin, forms.ModelForm):
  1293. class Meta:
  1294. model = PowerOutlet
  1295. fields = ['device', 'name']
  1296. widgets = {
  1297. 'device': forms.HiddenInput(),
  1298. }
  1299. class PowerOutletCreateForm(ComponentForm):
  1300. name_pattern = ExpandableNameField(label='Name')
  1301. class PowerOutletConnectionForm(BootstrapMixin, ChainedFieldsMixin, forms.Form):
  1302. site = forms.ModelChoiceField(
  1303. queryset=Site.objects.all(),
  1304. required=False,
  1305. widget=forms.Select(
  1306. attrs={'filter-for': 'rack'}
  1307. )
  1308. )
  1309. rack = ChainedModelChoiceField(
  1310. queryset=Rack.objects.all(),
  1311. chains=(
  1312. ('site', 'site'),
  1313. ),
  1314. label='Rack',
  1315. required=False,
  1316. widget=APISelect(
  1317. api_url='/api/dcim/racks/?site_id={{site}}',
  1318. attrs={'filter-for': 'device', 'nullable': 'true'}
  1319. )
  1320. )
  1321. device = ChainedModelChoiceField(
  1322. queryset=Device.objects.all(),
  1323. chains=(
  1324. ('site', 'site'),
  1325. ('rack', 'rack'),
  1326. ),
  1327. label='Device',
  1328. required=False,
  1329. widget=APISelect(
  1330. api_url='/api/dcim/devices/?site_id={{site}}&rack_id={{rack}}',
  1331. display_field='display_name',
  1332. attrs={'filter-for': 'port'}
  1333. )
  1334. )
  1335. livesearch = forms.CharField(
  1336. required=False,
  1337. label='Device',
  1338. widget=Livesearch(
  1339. query_key='q',
  1340. query_url='dcim-api:device-list',
  1341. field_to_update='device'
  1342. )
  1343. )
  1344. port = ChainedModelChoiceField(
  1345. queryset=PowerPort.objects.all(),
  1346. chains=(
  1347. ('device', 'device'),
  1348. ),
  1349. label='Port',
  1350. widget=APISelect(
  1351. api_url='/api/dcim/power-ports/?device_id={{device}}',
  1352. disabled_indicator='power_outlet'
  1353. )
  1354. )
  1355. connection_status = forms.BooleanField(
  1356. required=False,
  1357. initial=CONNECTION_STATUS_CONNECTED,
  1358. label='Status',
  1359. widget=forms.Select(
  1360. choices=CONNECTION_STATUS_CHOICES
  1361. )
  1362. )
  1363. class Meta:
  1364. fields = ['site', 'rack', 'device', 'livesearch', 'port', 'connection_status']
  1365. labels = {
  1366. 'connection_status': 'Status',
  1367. }
  1368. class PowerOutletBulkRenameForm(BulkRenameForm):
  1369. pk = forms.ModelMultipleChoiceField(queryset=PowerOutlet.objects.all(), widget=forms.MultipleHiddenInput)
  1370. class PowerOutletBulkDisconnectForm(ConfirmationForm):
  1371. pk = forms.ModelMultipleChoiceField(queryset=PowerOutlet.objects.all(), widget=forms.MultipleHiddenInput)
  1372. #
  1373. # Interfaces
  1374. #
  1375. class InterfaceForm(BootstrapMixin, forms.ModelForm, ChainedFieldsMixin):
  1376. site = forms.ModelChoiceField(
  1377. queryset=Site.objects.all(),
  1378. required=False,
  1379. label='VLAN site',
  1380. widget=forms.Select(
  1381. attrs={'filter-for': 'vlan_group', 'nullable': 'true'},
  1382. )
  1383. )
  1384. vlan_group = ChainedModelChoiceField(
  1385. queryset=VLANGroup.objects.all(),
  1386. chains=(
  1387. ('site', 'site'),
  1388. ),
  1389. required=False,
  1390. label='VLAN group',
  1391. widget=APISelect(
  1392. attrs={'filter-for': 'untagged_vlan tagged_vlans', 'nullable': 'true'},
  1393. api_url='/api/ipam/vlan-groups/?site_id={{site}}',
  1394. )
  1395. )
  1396. untagged_vlan = ChainedModelChoiceField(
  1397. queryset=VLAN.objects.all(),
  1398. chains=(
  1399. ('site', 'site'),
  1400. ('group', 'vlan_group'),
  1401. ),
  1402. required=False,
  1403. label='Untagged VLAN',
  1404. widget=APISelect(
  1405. api_url='/api/ipam/vlans/?site_id={{site}}&group_id={{vlan_group}}',
  1406. )
  1407. )
  1408. tagged_vlans = ChainedModelMultipleChoiceField(
  1409. queryset=VLAN.objects.all(),
  1410. chains=(
  1411. ('site', 'site'),
  1412. ('group', 'vlan_group'),
  1413. ),
  1414. required=False,
  1415. label='Tagged VLANs',
  1416. widget=APISelectMultiple(
  1417. api_url='/api/ipam/vlans/?site_id={{site}}&group_id={{vlan_group}}',
  1418. )
  1419. )
  1420. class Meta:
  1421. model = Interface
  1422. fields = [
  1423. 'device', 'name', 'form_factor', 'enabled', 'lag', 'mac_address', 'mtu', 'mgmt_only', 'description',
  1424. 'mode', 'site', 'vlan_group', 'untagged_vlan', 'tagged_vlans',
  1425. ]
  1426. widgets = {
  1427. 'device': forms.HiddenInput(),
  1428. }
  1429. def __init__(self, *args, **kwargs):
  1430. super(InterfaceForm, self).__init__(*args, **kwargs)
  1431. # Limit LAG choices to interfaces belonging to this device (or VC master)
  1432. if self.is_bound:
  1433. device = Device.objects.get(pk=self.data['device'])
  1434. self.fields['lag'].queryset = Interface.objects.order_naturally().filter(
  1435. device__in=[device, device.get_vc_master()], form_factor=IFACE_FF_LAG
  1436. )
  1437. else:
  1438. device = self.instance.device
  1439. self.fields['lag'].queryset = Interface.objects.order_naturally().filter(
  1440. device__in=[self.instance.device, self.instance.device.get_vc_master()], form_factor=IFACE_FF_LAG
  1441. )
  1442. # Limit the queryset for the site to only include the interface's device's site
  1443. if device and device.site:
  1444. self.fields['site'].queryset = Site.objects.filter(pk=device.site.id)
  1445. self.fields['site'].initial = None
  1446. else:
  1447. self.fields['site'].queryset = Site.objects.none()
  1448. self.fields['site'].initial = None
  1449. # Limit the initial vlan choices
  1450. if self.is_bound:
  1451. filter_dict = {
  1452. 'group_id': self.data.get('vlan_group') or None,
  1453. 'site_id': self.data.get('site') or None,
  1454. }
  1455. elif self.initial.get('untagged_vlan'):
  1456. filter_dict = {
  1457. 'group_id': self.instance.untagged_vlan.group,
  1458. 'site_id': self.instance.untagged_vlan.site,
  1459. }
  1460. elif self.initial.get('tagged_vlans'):
  1461. filter_dict = {
  1462. 'group_id': self.instance.tagged_vlans.first().group,
  1463. 'site_id': self.instance.tagged_vlans.first().site,
  1464. }
  1465. else:
  1466. filter_dict = {
  1467. 'group_id': None,
  1468. 'site_id': None,
  1469. }
  1470. self.fields['untagged_vlan'].queryset = VLAN.objects.filter(**filter_dict)
  1471. self.fields['tagged_vlans'].queryset = VLAN.objects.filter(**filter_dict)
  1472. def clean_tagged_vlans(self):
  1473. """
  1474. Because tagged_vlans is a many-to-many relationship, validation must be done in the form
  1475. """
  1476. if self.cleaned_data['mode'] == IFACE_MODE_ACCESS and self.cleaned_data['tagged_vlans']:
  1477. raise forms.ValidationError(
  1478. "An Access interface cannot have tagged VLANs."
  1479. )
  1480. if self.cleaned_data['mode'] == IFACE_MODE_TAGGED_ALL and self.cleaned_data['tagged_vlans']:
  1481. raise forms.ValidationError(
  1482. "Interface mode Tagged All implies all VLANs are tagged. "
  1483. "Do not select any tagged VLANs."
  1484. )
  1485. return self.cleaned_data['tagged_vlans']
  1486. class InterfaceCreateForm(ComponentForm, ChainedFieldsMixin):
  1487. name_pattern = ExpandableNameField(label='Name')
  1488. form_factor = forms.ChoiceField(choices=IFACE_FF_CHOICES)
  1489. enabled = forms.BooleanField(required=False)
  1490. lag = forms.ModelChoiceField(queryset=Interface.objects.all(), required=False, label='Parent LAG')
  1491. mtu = forms.IntegerField(required=False, min_value=1, max_value=32767, label='MTU')
  1492. mac_address = MACAddressFormField(required=False, label='MAC Address')
  1493. mgmt_only = forms.BooleanField(required=False, label='OOB Management')
  1494. description = forms.CharField(max_length=100, required=False)
  1495. mode = forms.ChoiceField(choices=add_blank_choice(IFACE_MODE_CHOICES), required=False)
  1496. site = forms.ModelChoiceField(
  1497. queryset=Site.objects.all(),
  1498. required=False,
  1499. label='VLAN Site',
  1500. widget=forms.Select(
  1501. attrs={'filter-for': 'vlan_group', 'nullable': 'true'},
  1502. )
  1503. )
  1504. vlan_group = ChainedModelChoiceField(
  1505. queryset=VLANGroup.objects.all(),
  1506. chains=(
  1507. ('site', 'site'),
  1508. ),
  1509. required=False,
  1510. label='VLAN group',
  1511. widget=APISelect(
  1512. attrs={'filter-for': 'untagged_vlan tagged_vlans', 'nullable': 'true'},
  1513. api_url='/api/ipam/vlan-groups/?site_id={{site}}',
  1514. )
  1515. )
  1516. untagged_vlan = ChainedModelChoiceField(
  1517. queryset=VLAN.objects.all(),
  1518. chains=(
  1519. ('site', 'site'),
  1520. ('group', 'vlan_group'),
  1521. ),
  1522. required=False,
  1523. label='Untagged VLAN',
  1524. widget=APISelect(
  1525. api_url='/api/ipam/vlans/?site_id={{site}}&group_id={{vlan_group}}',
  1526. )
  1527. )
  1528. tagged_vlans = ChainedModelMultipleChoiceField(
  1529. queryset=VLAN.objects.all(),
  1530. chains=(
  1531. ('site', 'site'),
  1532. ('group', 'vlan_group'),
  1533. ),
  1534. required=False,
  1535. label='Tagged VLANs',
  1536. widget=APISelectMultiple(
  1537. api_url='/api/ipam/vlans/?site_id={{site}}&group_id={{vlan_group}}',
  1538. )
  1539. )
  1540. def __init__(self, *args, **kwargs):
  1541. # Set interfaces enabled by default
  1542. kwargs['initial'] = kwargs.get('initial', {}).copy()
  1543. kwargs['initial'].update({'enabled': True})
  1544. super(InterfaceCreateForm, self).__init__(*args, **kwargs)
  1545. # Limit LAG choices to interfaces belonging to this device (or its VC master)
  1546. if self.parent is not None:
  1547. self.fields['lag'].queryset = Interface.objects.order_naturally().filter(
  1548. device__in=[self.parent, self.parent.get_vc_master()], form_factor=IFACE_FF_LAG
  1549. )
  1550. else:
  1551. self.fields['lag'].queryset = Interface.objects.none()
  1552. # Limit the queryset for the site to only include the interface's device's site
  1553. if self.parent is not None and self.parent.site:
  1554. self.fields['site'].queryset = Site.objects.filter(pk=self.parent.site.id)
  1555. self.fields['site'].initial = None
  1556. else:
  1557. self.fields['site'].queryset = Site.objects.none()
  1558. self.fields['site'].initial = None
  1559. # Limit the initial vlan choices
  1560. if self.is_bound:
  1561. filter_dict = {
  1562. 'group_id': self.data.get('vlan_group') or None,
  1563. 'site_id': self.data.get('site') or None,
  1564. }
  1565. elif self.initial.get('untagged_vlan'):
  1566. filter_dict = {
  1567. 'group_id': self.untagged_vlan.group,
  1568. 'site_id': self.untagged_vlan.site,
  1569. }
  1570. elif self.initial.get('tagged_vlans'):
  1571. filter_dict = {
  1572. 'group_id': self.tagged_vlans.first().group,
  1573. 'site_id': self.tagged_vlans.first().site,
  1574. }
  1575. else:
  1576. filter_dict = {
  1577. 'group_id': None,
  1578. 'site_id': None,
  1579. }
  1580. self.fields['untagged_vlan'].queryset = VLAN.objects.filter(**filter_dict)
  1581. self.fields['tagged_vlans'].queryset = VLAN.objects.filter(**filter_dict)
  1582. class InterfaceBulkEditForm(BootstrapMixin, BulkEditForm, ChainedFieldsMixin):
  1583. pk = forms.ModelMultipleChoiceField(queryset=Interface.objects.all(), widget=forms.MultipleHiddenInput)
  1584. device = forms.ModelChoiceField(queryset=Device.objects.all(), widget=forms.HiddenInput)
  1585. form_factor = forms.ChoiceField(choices=add_blank_choice(IFACE_FF_CHOICES), required=False)
  1586. enabled = forms.NullBooleanField(required=False, widget=BulkEditNullBooleanSelect)
  1587. lag = forms.ModelChoiceField(queryset=Interface.objects.all(), required=False, label='Parent LAG')
  1588. mtu = forms.IntegerField(required=False, min_value=1, max_value=32767, label='MTU')
  1589. mgmt_only = forms.NullBooleanField(required=False, widget=BulkEditNullBooleanSelect, label='Management only')
  1590. description = forms.CharField(max_length=100, required=False)
  1591. mode = forms.ChoiceField(choices=add_blank_choice(IFACE_MODE_CHOICES), required=False)
  1592. site = forms.ModelChoiceField(
  1593. queryset=Site.objects.all(),
  1594. required=False,
  1595. label='VLAN Site',
  1596. widget=forms.Select(
  1597. attrs={'filter-for': 'vlan_group', 'nullable': 'true'},
  1598. )
  1599. )
  1600. vlan_group = ChainedModelChoiceField(
  1601. queryset=VLANGroup.objects.all(),
  1602. chains=(
  1603. ('site', 'site'),
  1604. ),
  1605. required=False,
  1606. label='VLAN group',
  1607. widget=APISelect(
  1608. attrs={'filter-for': 'untagged_vlan tagged_vlans', 'nullable': 'true'},
  1609. api_url='/api/ipam/vlan-groups/?site_id={{site}}',
  1610. )
  1611. )
  1612. untagged_vlan = ChainedModelChoiceField(
  1613. queryset=VLAN.objects.all(),
  1614. chains=(
  1615. ('site', 'site'),
  1616. ('group', 'vlan_group'),
  1617. ),
  1618. required=False,
  1619. label='Untagged VLAN',
  1620. widget=APISelect(
  1621. api_url='/api/ipam/vlans/?site_id={{site}}&group_id={{vlan_group}}',
  1622. )
  1623. )
  1624. tagged_vlans = ChainedModelMultipleChoiceField(
  1625. queryset=VLAN.objects.all(),
  1626. chains=(
  1627. ('site', 'site'),
  1628. ('group', 'vlan_group'),
  1629. ),
  1630. required=False,
  1631. label='Tagged VLANs',
  1632. widget=APISelectMultiple(
  1633. api_url='/api/ipam/vlans/?site_id={{site}}&group_id={{vlan_group}}',
  1634. )
  1635. )
  1636. class Meta:
  1637. nullable_fields = ['lag', 'mtu', 'description', 'untagged_vlan', 'tagged_vlans']
  1638. def __init__(self, *args, **kwargs):
  1639. super(InterfaceBulkEditForm, self).__init__(*args, **kwargs)
  1640. # Limit LAG choices to interfaces which belong to the parent device (or VC master)
  1641. device = None
  1642. if self.initial.get('device'):
  1643. try:
  1644. device = Device.objects.get(pk=self.initial.get('device'))
  1645. except Device.DoesNotExist:
  1646. pass
  1647. else:
  1648. try:
  1649. device = Device.objects.get(pk=self.data.get('device'))
  1650. except Device.DoesNotExist:
  1651. pass
  1652. if device is not None:
  1653. interface_ordering = device.device_type.interface_ordering
  1654. self.fields['lag'].queryset = Interface.objects.order_naturally(method=interface_ordering).filter(
  1655. device__in=[device, device.get_vc_master()], form_factor=IFACE_FF_LAG
  1656. )
  1657. else:
  1658. self.fields['lag'].choices = []
  1659. # Limit the queryset for the site to only include the interface's device's site
  1660. if device and device.site:
  1661. self.fields['site'].queryset = Site.objects.filter(pk=device.site.id)
  1662. self.fields['site'].initial = None
  1663. else:
  1664. self.fields['site'].queryset = Site.objects.none()
  1665. self.fields['site'].initial = None
  1666. if self.is_bound:
  1667. filter_dict = {
  1668. 'group_id': self.data.get('vlan_group') or None,
  1669. 'site_id': self.data.get('site') or None,
  1670. }
  1671. else:
  1672. filter_dict = {
  1673. 'group_id': None,
  1674. 'site_id': None,
  1675. }
  1676. self.fields['untagged_vlan'].queryset = VLAN.objects.filter(**filter_dict)
  1677. self.fields['tagged_vlans'].queryset = VLAN.objects.filter(**filter_dict)
  1678. class InterfaceBulkRenameForm(BulkRenameForm):
  1679. pk = forms.ModelMultipleChoiceField(queryset=Interface.objects.all(), widget=forms.MultipleHiddenInput)
  1680. class InterfaceBulkDisconnectForm(ConfirmationForm):
  1681. pk = forms.ModelMultipleChoiceField(queryset=Interface.objects.all(), widget=forms.MultipleHiddenInput)
  1682. #
  1683. # Interface connections
  1684. #
  1685. class InterfaceConnectionForm(BootstrapMixin, ChainedFieldsMixin, forms.ModelForm):
  1686. interface_a = forms.ChoiceField(
  1687. choices=[],
  1688. widget=SelectWithDisabled,
  1689. label='Interface'
  1690. )
  1691. site_b = forms.ModelChoiceField(
  1692. queryset=Site.objects.all(),
  1693. label='Site',
  1694. required=False,
  1695. widget=forms.Select(
  1696. attrs={'filter-for': 'rack_b'}
  1697. )
  1698. )
  1699. rack_b = ChainedModelChoiceField(
  1700. queryset=Rack.objects.all(),
  1701. chains=(
  1702. ('site', 'site_b'),
  1703. ),
  1704. label='Rack',
  1705. required=False,
  1706. widget=APISelect(
  1707. api_url='/api/dcim/racks/?site_id={{site_b}}',
  1708. attrs={'filter-for': 'device_b', 'nullable': 'true'}
  1709. )
  1710. )
  1711. device_b = ChainedModelChoiceField(
  1712. queryset=Device.objects.all(),
  1713. chains=(
  1714. ('site', 'site_b'),
  1715. ('rack', 'rack_b'),
  1716. ),
  1717. label='Device',
  1718. required=False,
  1719. widget=APISelect(
  1720. api_url='/api/dcim/devices/?site_id={{site_b}}&rack_id={{rack_b}}',
  1721. display_field='display_name',
  1722. attrs={'filter-for': 'interface_b'}
  1723. )
  1724. )
  1725. livesearch = forms.CharField(
  1726. required=False,
  1727. label='Device',
  1728. widget=Livesearch(
  1729. query_key='q',
  1730. query_url='dcim-api:device-list',
  1731. field_to_update='device_b'
  1732. )
  1733. )
  1734. interface_b = ChainedModelChoiceField(
  1735. queryset=Interface.objects.connectable().select_related(
  1736. 'circuit_termination', 'connected_as_a', 'connected_as_b'
  1737. ),
  1738. chains=(
  1739. ('device', 'device_b'),
  1740. ),
  1741. label='Interface',
  1742. widget=APISelect(
  1743. api_url='/api/dcim/interfaces/?device_id={{device_b}}&type=physical',
  1744. disabled_indicator='is_connected'
  1745. )
  1746. )
  1747. class Meta:
  1748. model = InterfaceConnection
  1749. fields = ['interface_a', 'site_b', 'rack_b', 'device_b', 'interface_b', 'livesearch', 'connection_status']
  1750. def __init__(self, device_a, *args, **kwargs):
  1751. super(InterfaceConnectionForm, self).__init__(*args, **kwargs)
  1752. # Initialize interface A choices
  1753. device_a_interfaces = Interface.objects.connectable().order_naturally().filter(device=device_a).select_related(
  1754. 'circuit_termination', 'connected_as_a', 'connected_as_b'
  1755. )
  1756. self.fields['interface_a'].choices = [
  1757. (iface.id, {'label': iface.name, 'disabled': iface.is_connected}) for iface in device_a_interfaces
  1758. ]
  1759. # Mark connected interfaces as disabled
  1760. if self.data.get('device_b'):
  1761. self.fields['interface_b'].choices = [
  1762. (iface.id, {'label': iface.name, 'disabled': iface.is_connected}) for iface in self.fields['interface_b'].queryset
  1763. ]
  1764. class InterfaceConnectionCSVForm(forms.ModelForm):
  1765. device_a = FlexibleModelChoiceField(
  1766. queryset=Device.objects.all(),
  1767. to_field_name='name',
  1768. help_text='Name or ID of device A',
  1769. error_messages={'invalid_choice': 'Device A not found.'}
  1770. )
  1771. interface_a = forms.CharField(
  1772. help_text='Name of interface A'
  1773. )
  1774. device_b = FlexibleModelChoiceField(
  1775. queryset=Device.objects.all(),
  1776. to_field_name='name',
  1777. help_text='Name or ID of device B',
  1778. error_messages={'invalid_choice': 'Device B not found.'}
  1779. )
  1780. interface_b = forms.CharField(
  1781. help_text='Name of interface B'
  1782. )
  1783. connection_status = CSVChoiceField(
  1784. choices=CONNECTION_STATUS_CHOICES,
  1785. help_text='Connection status'
  1786. )
  1787. class Meta:
  1788. model = InterfaceConnection
  1789. fields = InterfaceConnection.csv_headers
  1790. def clean_interface_a(self):
  1791. interface_name = self.cleaned_data.get('interface_a')
  1792. if not interface_name:
  1793. return None
  1794. try:
  1795. # Retrieve interface by name
  1796. interface = Interface.objects.get(
  1797. device=self.cleaned_data['device_a'], name=interface_name
  1798. )
  1799. # Check for an existing connection to this interface
  1800. if InterfaceConnection.objects.filter(Q(interface_a=interface) | Q(interface_b=interface)).count():
  1801. raise forms.ValidationError("{} {} is already connected".format(
  1802. self.cleaned_data['device_a'], interface_name
  1803. ))
  1804. except Interface.DoesNotExist:
  1805. raise forms.ValidationError("Invalid interface ({} {})".format(
  1806. self.cleaned_data['device_a'], interface_name
  1807. ))
  1808. return interface
  1809. def clean_interface_b(self):
  1810. interface_name = self.cleaned_data.get('interface_b')
  1811. if not interface_name:
  1812. return None
  1813. try:
  1814. # Retrieve interface by name
  1815. interface = Interface.objects.get(
  1816. device=self.cleaned_data['device_b'], name=interface_name
  1817. )
  1818. # Check for an existing connection to this interface
  1819. if InterfaceConnection.objects.filter(Q(interface_a=interface) | Q(interface_b=interface)).count():
  1820. raise forms.ValidationError("{} {} is already connected".format(
  1821. self.cleaned_data['device_b'], interface_name
  1822. ))
  1823. except Interface.DoesNotExist:
  1824. raise forms.ValidationError("Invalid interface ({} {})".format(
  1825. self.cleaned_data['device_b'], interface_name
  1826. ))
  1827. return interface
  1828. #
  1829. # Device bays
  1830. #
  1831. class DeviceBayForm(BootstrapMixin, forms.ModelForm):
  1832. class Meta:
  1833. model = DeviceBay
  1834. fields = ['device', 'name']
  1835. widgets = {
  1836. 'device': forms.HiddenInput(),
  1837. }
  1838. class DeviceBayCreateForm(ComponentForm):
  1839. name_pattern = ExpandableNameField(label='Name')
  1840. class PopulateDeviceBayForm(BootstrapMixin, forms.Form):
  1841. installed_device = forms.ModelChoiceField(
  1842. queryset=Device.objects.all(),
  1843. label='Child Device',
  1844. help_text="Child devices must first be created and assigned to the site/rack of the parent device."
  1845. )
  1846. def __init__(self, device_bay, *args, **kwargs):
  1847. super(PopulateDeviceBayForm, self).__init__(*args, **kwargs)
  1848. self.fields['installed_device'].queryset = Device.objects.filter(
  1849. site=device_bay.device.site,
  1850. rack=device_bay.device.rack,
  1851. parent_bay__isnull=True,
  1852. device_type__u_height=0,
  1853. device_type__subdevice_role=SUBDEVICE_ROLE_CHILD
  1854. ).exclude(pk=device_bay.device.pk)
  1855. class DeviceBayBulkRenameForm(BulkRenameForm):
  1856. pk = forms.ModelMultipleChoiceField(queryset=DeviceBay.objects.all(), widget=forms.MultipleHiddenInput)
  1857. #
  1858. # Connections
  1859. #
  1860. class ConsoleConnectionFilterForm(BootstrapMixin, forms.Form):
  1861. site = forms.ModelChoiceField(required=False, queryset=Site.objects.all(), to_field_name='slug')
  1862. device = forms.CharField(required=False, label='Device name')
  1863. class PowerConnectionFilterForm(BootstrapMixin, forms.Form):
  1864. site = forms.ModelChoiceField(required=False, queryset=Site.objects.all(), to_field_name='slug')
  1865. device = forms.CharField(required=False, label='Device name')
  1866. class InterfaceConnectionFilterForm(BootstrapMixin, forms.Form):
  1867. site = forms.ModelChoiceField(required=False, queryset=Site.objects.all(), to_field_name='slug')
  1868. device = forms.CharField(required=False, label='Device name')
  1869. #
  1870. # Inventory items
  1871. #
  1872. class InventoryItemForm(BootstrapMixin, forms.ModelForm):
  1873. class Meta:
  1874. model = InventoryItem
  1875. fields = ['name', 'manufacturer', 'part_id', 'serial', 'asset_tag', 'description']
  1876. class InventoryItemCSVForm(forms.ModelForm):
  1877. device = FlexibleModelChoiceField(
  1878. queryset=Device.objects.all(),
  1879. to_field_name='name',
  1880. help_text='Device name or ID',
  1881. error_messages={
  1882. 'invalid_choice': 'Device not found.',
  1883. }
  1884. )
  1885. manufacturer = forms.ModelChoiceField(
  1886. queryset=Manufacturer.objects.all(),
  1887. to_field_name='name',
  1888. required=False,
  1889. help_text='Manufacturer name',
  1890. error_messages={
  1891. 'invalid_choice': 'Invalid manufacturer.',
  1892. }
  1893. )
  1894. class Meta:
  1895. model = InventoryItem
  1896. fields = InventoryItem.csv_headers
  1897. class InventoryItemBulkEditForm(BootstrapMixin, BulkEditForm):
  1898. pk = forms.ModelMultipleChoiceField(queryset=InventoryItem.objects.all(), widget=forms.MultipleHiddenInput)
  1899. manufacturer = forms.ModelChoiceField(queryset=Manufacturer.objects.all(), required=False)
  1900. part_id = forms.CharField(max_length=50, required=False, label='Part ID')
  1901. description = forms.CharField(max_length=100, required=False)
  1902. class Meta:
  1903. nullable_fields = ['manufacturer', 'part_id', 'description']
  1904. class InventoryItemFilterForm(BootstrapMixin, forms.Form):
  1905. model = InventoryItem
  1906. q = forms.CharField(required=False, label='Search')
  1907. manufacturer = FilterChoiceField(
  1908. queryset=Manufacturer.objects.annotate(filter_count=Count('inventory_items')),
  1909. to_field_name='slug',
  1910. null_label='-- None --'
  1911. )
  1912. #
  1913. # Virtual chassis
  1914. #
  1915. class DeviceSelectionForm(forms.Form):
  1916. pk = forms.ModelMultipleChoiceField(queryset=Device.objects.all(), widget=forms.MultipleHiddenInput)
  1917. class VirtualChassisForm(BootstrapMixin, forms.ModelForm):
  1918. class Meta:
  1919. model = VirtualChassis
  1920. fields = ['master', 'domain']
  1921. class VCMemberSelectForm(BootstrapMixin, ChainedFieldsMixin, forms.Form):
  1922. site = forms.ModelChoiceField(
  1923. queryset=Site.objects.all(),
  1924. label='Site',
  1925. required=False,
  1926. widget=forms.Select(
  1927. attrs={'filter-for': 'rack'}
  1928. )
  1929. )
  1930. rack = ChainedModelChoiceField(
  1931. queryset=Rack.objects.all(),
  1932. chains=(
  1933. ('site', 'site'),
  1934. ),
  1935. label='Rack',
  1936. required=False,
  1937. widget=APISelect(
  1938. api_url='/api/dcim/racks/?site_id={{site}}',
  1939. attrs={'filter-for': 'device', 'nullable': 'true'}
  1940. )
  1941. )
  1942. device = ChainedModelChoiceField(
  1943. queryset=Device.objects.all(),
  1944. chains=(
  1945. ('site', 'site'),
  1946. ('rack', 'rack'),
  1947. ),
  1948. label='Device',
  1949. widget=APISelect(
  1950. api_url='/api/dcim/devices/?site_id={{site}}&rack_id={{rack}}',
  1951. display_field='display_name'
  1952. )
  1953. )
  1954. def clean_device(self):
  1955. device = self.cleaned_data['device']
  1956. if device.virtual_chassis is not None:
  1957. raise forms.ValidationError("Device {} is already assigned to a virtual chassis.".format(device))
  1958. class DeviceVCMembershipForm(forms.ModelForm):
  1959. class Meta:
  1960. model = Device
  1961. fields = ['vc_position', 'vc_priority']
  1962. labels = {
  1963. 'vc_position': 'Position',
  1964. 'vc_priority': 'Priority',
  1965. }
  1966. def __init__(self, *args, **kwargs):
  1967. super(DeviceVCMembershipForm, self).__init__(*args, **kwargs)
  1968. # Require VC position when assigning a member
  1969. self.fields['vc_position'].required = True
  1970. def clean_vc_position(self):
  1971. vc_position = self.cleaned_data['vc_position']
  1972. if Device.objects.filter(virtual_chassis=self.instance.virtual_chassis, vc_position=vc_position).exists():
  1973. raise forms.ValidationError("A virtual chassis member already exists in this position.")
  1974. return vc_position