forms.py 55 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323
  1. import re
  2. from django import forms
  3. from django.db.models import Count, Q
  4. from extras.forms import CustomFieldForm
  5. from ipam.models import IPAddress
  6. from tenancy.forms import bulkedit_tenant_choices
  7. from tenancy.models import Tenant
  8. from utilities.forms import (
  9. APISelect, add_blank_choice, BootstrapMixin, BulkImportForm, CommentField, CSVDataField, ExpandableNameField,
  10. FlexibleModelChoiceField, Livesearch, SelectWithDisabled, SmallTextarea, SlugField,
  11. )
  12. from .models import (
  13. DeviceBay, DeviceBayTemplate, CONNECTION_STATUS_CHOICES, CONNECTION_STATUS_PLANNED, CONNECTION_STATUS_CONNECTED,
  14. ConsolePort, ConsolePortTemplate, ConsoleServerPort, ConsoleServerPortTemplate, Device, DeviceRole, DeviceType,
  15. Interface, IFACE_FF_VIRTUAL, InterfaceConnection, InterfaceTemplate, Manufacturer, Module, Platform, PowerOutlet,
  16. PowerOutletTemplate, PowerPort, PowerPortTemplate, RACK_TYPE_CHOICES, RACK_WIDTH_CHOICES, Rack, RackGroup, RackRole,
  17. Site, STATUS_CHOICES, SUBDEVICE_ROLE_CHILD
  18. )
  19. FORM_STATUS_CHOICES = [
  20. ['', '---------'],
  21. ]
  22. FORM_STATUS_CHOICES += STATUS_CHOICES
  23. DEVICE_BY_PK_RE = '{\d+\}'
  24. def get_device_by_name_or_pk(name):
  25. """
  26. Attempt to retrieve a device by either its name or primary key ('{pk}').
  27. """
  28. if re.match(DEVICE_BY_PK_RE, name):
  29. pk = name.strip('{}')
  30. device = Device.objects.get(pk=pk)
  31. else:
  32. device = Device.objects.get(name=name)
  33. return device
  34. def bulkedit_platform_choices():
  35. choices = [
  36. (None, '---------'),
  37. (0, 'None'),
  38. ]
  39. choices += [(p.pk, p.name) for p in Platform.objects.all()]
  40. return choices
  41. def bulkedit_rackgroup_choices():
  42. """
  43. Include an option to remove the currently assigned group from a rack.
  44. """
  45. choices = [
  46. (None, '---------'),
  47. (0, 'None'),
  48. ]
  49. choices += [(r.pk, r) for r in RackGroup.objects.all()]
  50. return choices
  51. def bulkedit_rackrole_choices():
  52. """
  53. Include an option to remove the currently assigned role from a rack.
  54. """
  55. choices = [
  56. (None, '---------'),
  57. (0, 'None'),
  58. ]
  59. choices += [(r.pk, r.name) for r in RackRole.objects.all()]
  60. return choices
  61. #
  62. # Sites
  63. #
  64. class SiteForm(BootstrapMixin, CustomFieldForm):
  65. slug = SlugField()
  66. comments = CommentField()
  67. class Meta:
  68. model = Site
  69. fields = ['name', 'slug', 'tenant', 'facility', 'asn', 'physical_address', 'shipping_address', 'comments']
  70. widgets = {
  71. 'physical_address': SmallTextarea(attrs={'rows': 3}),
  72. 'shipping_address': SmallTextarea(attrs={'rows': 3}),
  73. }
  74. help_texts = {
  75. 'name': "Full name of the site",
  76. 'facility': "Data center provider and facility (e.g. Equinix NY7)",
  77. 'asn': "BGP autonomous system number",
  78. 'physical_address': "Physical location of the building (e.g. for GPS)",
  79. 'shipping_address': "If different from the physical address"
  80. }
  81. class SiteFromCSVForm(forms.ModelForm):
  82. tenant = forms.ModelChoiceField(Tenant.objects.all(), to_field_name='name', required=False,
  83. error_messages={'invalid_choice': 'Tenant not found.'})
  84. class Meta:
  85. model = Site
  86. fields = ['name', 'slug', 'tenant', 'facility', 'asn']
  87. class SiteImportForm(BulkImportForm, BootstrapMixin):
  88. csv = CSVDataField(csv_form=SiteFromCSVForm)
  89. class SiteBulkEditForm(forms.Form, BootstrapMixin):
  90. pk = forms.ModelMultipleChoiceField(queryset=Site.objects.all(), widget=forms.MultipleHiddenInput)
  91. tenant = forms.TypedChoiceField(choices=bulkedit_tenant_choices, coerce=int, required=False, label='Tenant')
  92. def site_tenant_choices():
  93. tenant_choices = Tenant.objects.annotate(site_count=Count('sites'))
  94. return [(t.slug, u'{} ({})'.format(t.name, t.site_count)) for t in tenant_choices]
  95. class SiteFilterForm(forms.Form, BootstrapMixin):
  96. tenant = forms.MultipleChoiceField(required=False, choices=site_tenant_choices,
  97. widget=forms.SelectMultiple(attrs={'size': 8}))
  98. #
  99. # Rack groups
  100. #
  101. class RackGroupForm(forms.ModelForm, BootstrapMixin):
  102. slug = SlugField()
  103. class Meta:
  104. model = RackGroup
  105. fields = ['site', 'name', 'slug']
  106. def rackgroup_site_choices():
  107. site_choices = Site.objects.annotate(rack_count=Count('rack_groups'))
  108. return [(s.slug, u'{} ({})'.format(s.name, s.rack_count)) for s in site_choices]
  109. class RackGroupFilterForm(forms.Form, BootstrapMixin):
  110. site = forms.MultipleChoiceField(required=False, choices=rackgroup_site_choices,
  111. widget=forms.SelectMultiple(attrs={'size': 8}))
  112. #
  113. # Rack roles
  114. #
  115. class RackRoleForm(forms.ModelForm, BootstrapMixin):
  116. slug = SlugField()
  117. class Meta:
  118. model = RackRole
  119. fields = ['name', 'slug', 'color']
  120. #
  121. # Racks
  122. #
  123. class RackForm(BootstrapMixin, CustomFieldForm):
  124. group = forms.ModelChoiceField(queryset=RackGroup.objects.all(), required=False, label='Group', widget=APISelect(
  125. api_url='/api/dcim/rack-groups/?site_id={{site}}',
  126. ))
  127. comments = CommentField()
  128. class Meta:
  129. model = Rack
  130. fields = ['site', 'group', 'name', 'facility_id', 'tenant', 'role', 'type', 'width', 'u_height', 'comments']
  131. help_texts = {
  132. 'site': "The site at which the rack exists",
  133. 'name': "Organizational rack name",
  134. 'facility_id': "The unique rack ID assigned by the facility",
  135. 'u_height': "Height in rack units",
  136. }
  137. widgets = {
  138. 'site': forms.Select(attrs={'filter-for': 'group'}),
  139. }
  140. def __init__(self, *args, **kwargs):
  141. super(RackForm, self).__init__(*args, **kwargs)
  142. # Limit rack group choices
  143. if self.is_bound and self.data.get('site'):
  144. self.fields['group'].queryset = RackGroup.objects.filter(site__pk=self.data['site'])
  145. elif self.initial.get('site'):
  146. self.fields['group'].queryset = RackGroup.objects.filter(site=self.initial['site'])
  147. else:
  148. self.fields['group'].choices = []
  149. class RackFromCSVForm(forms.ModelForm):
  150. site = forms.ModelChoiceField(queryset=Site.objects.all(), to_field_name='name',
  151. error_messages={'invalid_choice': 'Site not found.'})
  152. group_name = forms.CharField(required=False)
  153. tenant = forms.ModelChoiceField(Tenant.objects.all(), to_field_name='name', required=False,
  154. error_messages={'invalid_choice': 'Tenant not found.'})
  155. role = forms.ModelChoiceField(RackRole.objects.all(), to_field_name='name', required=False,
  156. error_messages={'invalid_choice': 'Role not found.'})
  157. type = forms.CharField(required=False)
  158. class Meta:
  159. model = Rack
  160. fields = ['site', 'group_name', 'name', 'facility_id', 'tenant', 'role', 'type', 'width', 'u_height']
  161. def clean(self):
  162. site = self.cleaned_data.get('site')
  163. group = self.cleaned_data.get('group_name')
  164. # Validate rack group
  165. if site and group:
  166. try:
  167. self.instance.group = RackGroup.objects.get(site=site, name=group)
  168. except RackGroup.DoesNotExist:
  169. self.add_error('group_name', "Invalid rack group ({})".format(group))
  170. def clean_type(self):
  171. rack_type = self.cleaned_data['type']
  172. if not rack_type:
  173. return None
  174. try:
  175. choices = {v.lower(): k for k, v in RACK_TYPE_CHOICES}
  176. return choices[rack_type.lower()]
  177. except KeyError:
  178. raise forms.ValidationError('Invalid rack type ({}). Valid choices are: {}.'.format(
  179. rack_type,
  180. ', '.join({v: k for k, v in RACK_TYPE_CHOICES}),
  181. ))
  182. class RackImportForm(BulkImportForm, BootstrapMixin):
  183. csv = CSVDataField(csv_form=RackFromCSVForm)
  184. class RackBulkEditForm(forms.Form, BootstrapMixin):
  185. pk = forms.ModelMultipleChoiceField(queryset=Rack.objects.all(), widget=forms.MultipleHiddenInput)
  186. site = forms.ModelChoiceField(queryset=Site.objects.all(), required=False, label='Site')
  187. group = forms.TypedChoiceField(choices=bulkedit_rackgroup_choices, coerce=int, required=False, label='Group')
  188. tenant = forms.TypedChoiceField(choices=bulkedit_tenant_choices, coerce=int, required=False, label='Tenant')
  189. role = forms.TypedChoiceField(choices=bulkedit_rackrole_choices, coerce=int, required=False, label='Role')
  190. type = forms.ChoiceField(choices=add_blank_choice(RACK_TYPE_CHOICES), required=False, label='Type')
  191. width = forms.ChoiceField(choices=add_blank_choice(RACK_WIDTH_CHOICES), required=False, label='Width')
  192. u_height = forms.IntegerField(required=False, label='Height (U)')
  193. comments = CommentField()
  194. def rack_site_choices():
  195. site_choices = Site.objects.annotate(rack_count=Count('racks'))
  196. return [(s.slug, u'{} ({})'.format(s.name, s.rack_count)) for s in site_choices]
  197. def rack_group_choices():
  198. group_choices = RackGroup.objects.select_related('site').annotate(rack_count=Count('racks'))
  199. return [(g.pk, u'{} ({})'.format(g, g.rack_count)) for g in group_choices]
  200. def rack_tenant_choices():
  201. tenant_choices = Tenant.objects.annotate(rack_count=Count('racks'))
  202. return [(t.slug, u'{} ({})'.format(t.name, t.rack_count)) for t in tenant_choices]
  203. def rack_role_choices():
  204. role_choices = RackRole.objects.annotate(rack_count=Count('racks'))
  205. return [(r.slug, u'{} ({})'.format(r.name, r.rack_count)) for r in role_choices]
  206. class RackFilterForm(forms.Form, BootstrapMixin):
  207. site = forms.MultipleChoiceField(required=False, choices=rack_site_choices,
  208. widget=forms.SelectMultiple(attrs={'size': 8}))
  209. group_id = forms.MultipleChoiceField(required=False, choices=rack_group_choices, label='Rack Group',
  210. widget=forms.SelectMultiple(attrs={'size': 8}))
  211. tenant = forms.MultipleChoiceField(required=False, choices=rack_tenant_choices,
  212. widget=forms.SelectMultiple(attrs={'size': 8}))
  213. role = forms.MultipleChoiceField(required=False, choices=rack_role_choices,
  214. widget=forms.SelectMultiple(attrs={'size': 8}))
  215. #
  216. # Manufacturers
  217. #
  218. class ManufacturerForm(forms.ModelForm, BootstrapMixin):
  219. slug = SlugField()
  220. class Meta:
  221. model = Manufacturer
  222. fields = ['name', 'slug']
  223. #
  224. # Device types
  225. #
  226. class DeviceTypeForm(forms.ModelForm, BootstrapMixin):
  227. slug = SlugField(slug_source='model')
  228. class Meta:
  229. model = DeviceType
  230. fields = ['manufacturer', 'model', 'slug', 'part_number', 'u_height', 'is_full_depth', 'is_console_server',
  231. 'is_pdu', 'is_network_device', 'subdevice_role']
  232. class DeviceTypeBulkEditForm(forms.Form, BootstrapMixin):
  233. pk = forms.ModelMultipleChoiceField(queryset=DeviceType.objects.all(), widget=forms.MultipleHiddenInput)
  234. manufacturer = forms.ModelChoiceField(queryset=Manufacturer.objects.all(), required=False)
  235. u_height = forms.IntegerField(min_value=1, required=False)
  236. def devicetype_manufacturer_choices():
  237. manufacturer_choices = Manufacturer.objects.annotate(devicetype_count=Count('device_types'))
  238. return [(m.slug, u'{} ({})'.format(m.name, m.devicetype_count)) for m in manufacturer_choices]
  239. class DeviceTypeFilterForm(forms.Form, BootstrapMixin):
  240. manufacturer = forms.MultipleChoiceField(required=False, choices=devicetype_manufacturer_choices,
  241. widget=forms.SelectMultiple(attrs={'size': 8}))
  242. #
  243. # Device component templates
  244. #
  245. class ConsolePortTemplateForm(forms.ModelForm, BootstrapMixin):
  246. name_pattern = ExpandableNameField(label='Name')
  247. class Meta:
  248. model = ConsolePortTemplate
  249. fields = ['name_pattern']
  250. class ConsoleServerPortTemplateForm(forms.ModelForm, BootstrapMixin):
  251. name_pattern = ExpandableNameField(label='Name')
  252. class Meta:
  253. model = ConsoleServerPortTemplate
  254. fields = ['name_pattern']
  255. class PowerPortTemplateForm(forms.ModelForm, BootstrapMixin):
  256. name_pattern = ExpandableNameField(label='Name')
  257. class Meta:
  258. model = PowerPortTemplate
  259. fields = ['name_pattern']
  260. class PowerOutletTemplateForm(forms.ModelForm, BootstrapMixin):
  261. name_pattern = ExpandableNameField(label='Name')
  262. class Meta:
  263. model = PowerOutletTemplate
  264. fields = ['name_pattern']
  265. class InterfaceTemplateForm(forms.ModelForm, BootstrapMixin):
  266. name_pattern = ExpandableNameField(label='Name')
  267. class Meta:
  268. model = InterfaceTemplate
  269. fields = ['name_pattern', 'form_factor', 'mgmt_only']
  270. class DeviceBayTemplateForm(forms.ModelForm, BootstrapMixin):
  271. name_pattern = ExpandableNameField(label='Name')
  272. class Meta:
  273. model = DeviceBayTemplate
  274. fields = ['name_pattern']
  275. #
  276. # Device roles
  277. #
  278. class DeviceRoleForm(forms.ModelForm, BootstrapMixin):
  279. slug = SlugField()
  280. class Meta:
  281. model = DeviceRole
  282. fields = ['name', 'slug', 'color']
  283. #
  284. # Platforms
  285. #
  286. class PlatformForm(forms.ModelForm, BootstrapMixin):
  287. slug = SlugField()
  288. class Meta:
  289. model = Platform
  290. fields = ['name', 'slug']
  291. #
  292. # Devices
  293. #
  294. class DeviceForm(BootstrapMixin, CustomFieldForm):
  295. site = forms.ModelChoiceField(queryset=Site.objects.all(), widget=forms.Select(attrs={'filter-for': 'rack'}))
  296. rack = forms.ModelChoiceField(queryset=Rack.objects.all(), widget=APISelect(
  297. api_url='/api/dcim/racks/?site_id={{site}}',
  298. display_field='display_name',
  299. attrs={'filter-for': 'position'}
  300. ))
  301. position = forms.TypedChoiceField(required=False, empty_value=None,
  302. help_text="For multi-U devices, this is the lowest occupied rack unit.",
  303. widget=APISelect(api_url='/api/dcim/racks/{{rack}}/rack-units/?face={{face}}',
  304. disabled_indicator='device'))
  305. manufacturer = forms.ModelChoiceField(queryset=Manufacturer.objects.all(),
  306. widget=forms.Select(attrs={'filter-for': 'device_type'}))
  307. device_type = forms.ModelChoiceField(queryset=DeviceType.objects.all(), label='Device type', widget=APISelect(
  308. api_url='/api/dcim/device-types/?manufacturer_id={{manufacturer}}',
  309. display_field='model'
  310. ))
  311. comments = CommentField()
  312. class Meta:
  313. model = Device
  314. fields = ['name', 'device_role', 'tenant', 'device_type', 'serial', 'asset_tag', 'site', 'rack', 'position',
  315. 'face', 'status', 'platform', 'primary_ip4', 'primary_ip6', 'comments']
  316. help_texts = {
  317. 'device_role': "The function this device serves",
  318. 'serial': "Chassis serial number",
  319. }
  320. widgets = {
  321. 'face': forms.Select(attrs={'filter-for': 'position'}),
  322. 'manufacturer': forms.Select(attrs={'filter-for': 'device_type'}),
  323. }
  324. def __init__(self, *args, **kwargs):
  325. super(DeviceForm, self).__init__(*args, **kwargs)
  326. if self.instance.pk:
  327. # Initialize helper selections
  328. self.initial['site'] = self.instance.rack.site
  329. self.initial['manufacturer'] = self.instance.device_type.manufacturer
  330. # Compile list of choices for primary IPv4 and IPv6 addresses
  331. for family in [4, 6]:
  332. ip_choices = []
  333. interface_ips = IPAddress.objects.filter(family=family, interface__device=self.instance)
  334. ip_choices += [(ip.id, u'{} ({})'.format(ip.address, ip.interface)) for ip in interface_ips]
  335. nat_ips = IPAddress.objects.filter(family=family, nat_inside__interface__device=self.instance)\
  336. .select_related('nat_inside__interface')
  337. ip_choices += [(ip.id, u'{} ({} NAT)'.format(ip.address, ip.nat_inside.interface)) for ip in nat_ips]
  338. self.fields['primary_ip{}'.format(family)].choices = [(None, '---------')] + ip_choices
  339. else:
  340. # An object that doesn't exist yet can't have any IPs assigned to it
  341. self.fields['primary_ip4'].choices = []
  342. self.fields['primary_ip4'].widget.attrs['readonly'] = True
  343. self.fields['primary_ip6'].choices = []
  344. self.fields['primary_ip6'].widget.attrs['readonly'] = True
  345. # Limit rack choices
  346. if self.is_bound and self.data.get('site'):
  347. self.fields['rack'].queryset = Rack.objects.filter(site__pk=self.data['site'])
  348. elif self.initial.get('site'):
  349. self.fields['rack'].queryset = Rack.objects.filter(site=self.initial['site'])
  350. else:
  351. self.fields['rack'].choices = []
  352. # Rack position
  353. pk = self.instance.pk if self.instance.pk else None
  354. try:
  355. if self.is_bound and self.data.get('rack') and str(self.data.get('face')):
  356. position_choices = Rack.objects.get(pk=self.data['rack'])\
  357. .get_rack_units(face=self.data.get('face'), exclude=pk)
  358. elif self.initial.get('rack') and str(self.initial.get('face')):
  359. position_choices = Rack.objects.get(pk=self.initial['rack'])\
  360. .get_rack_units(face=self.initial.get('face'), exclude=pk)
  361. else:
  362. position_choices = []
  363. except Rack.DoesNotExist:
  364. position_choices = []
  365. self.fields['position'].choices = [('', '---------')] + [
  366. (p['id'], {
  367. 'label': p['name'],
  368. 'disabled': bool(p['device'] and p['id'] != self.initial.get('position')),
  369. }) for p in position_choices
  370. ]
  371. # Limit device_type choices
  372. if self.is_bound:
  373. self.fields['device_type'].queryset = DeviceType.objects.filter(manufacturer__pk=self.data['manufacturer'])\
  374. .select_related('manufacturer')
  375. elif self.initial.get('manufacturer'):
  376. self.fields['device_type'].queryset = DeviceType.objects.filter(manufacturer=self.initial['manufacturer'])\
  377. .select_related('manufacturer')
  378. else:
  379. self.fields['device_type'].choices = []
  380. # Disable rack assignment if this is a child device installed in a parent device
  381. if pk and self.instance.device_type.is_child_device and hasattr(self.instance, 'parent_bay'):
  382. self.fields['site'].disabled = True
  383. self.fields['rack'].disabled = True
  384. self.initial['site'] = self.instance.parent_bay.device.rack.site_id
  385. self.initial['rack'] = self.instance.parent_bay.device.rack_id
  386. class BaseDeviceFromCSVForm(forms.ModelForm):
  387. device_role = forms.ModelChoiceField(queryset=DeviceRole.objects.all(), to_field_name='name',
  388. error_messages={'invalid_choice': 'Invalid device role.'})
  389. tenant = forms.ModelChoiceField(Tenant.objects.all(), to_field_name='name', required=False,
  390. error_messages={'invalid_choice': 'Tenant not found.'})
  391. manufacturer = forms.ModelChoiceField(queryset=Manufacturer.objects.all(), to_field_name='name',
  392. error_messages={'invalid_choice': 'Invalid manufacturer.'})
  393. model_name = forms.CharField()
  394. platform = forms.ModelChoiceField(queryset=Platform.objects.all(), required=False, to_field_name='name',
  395. error_messages={'invalid_choice': 'Invalid platform.'})
  396. class Meta:
  397. fields = []
  398. model = Device
  399. def clean(self):
  400. manufacturer = self.cleaned_data.get('manufacturer')
  401. model_name = self.cleaned_data.get('model_name')
  402. # Validate device type
  403. if manufacturer and model_name:
  404. try:
  405. self.instance.device_type = DeviceType.objects.get(manufacturer=manufacturer, model=model_name)
  406. except DeviceType.DoesNotExist:
  407. self.add_error('model_name', "Invalid device type ({} {})".format(manufacturer, model_name))
  408. class DeviceFromCSVForm(BaseDeviceFromCSVForm):
  409. site = forms.ModelChoiceField(queryset=Site.objects.all(), to_field_name='name', error_messages={
  410. 'invalid_choice': 'Invalid site name.',
  411. })
  412. rack_name = forms.CharField()
  413. face = forms.CharField(required=False)
  414. class Meta(BaseDeviceFromCSVForm.Meta):
  415. fields = ['name', 'device_role', 'tenant', 'manufacturer', 'model_name', 'platform', 'serial', 'asset_tag',
  416. 'site', 'rack_name', 'position', 'face']
  417. def clean(self):
  418. super(DeviceFromCSVForm, self).clean()
  419. site = self.cleaned_data.get('site')
  420. rack_name = self.cleaned_data.get('rack_name')
  421. # Validate rack
  422. if site and rack_name:
  423. try:
  424. self.instance.rack = Rack.objects.get(site=site, name=rack_name)
  425. except Rack.DoesNotExist:
  426. self.add_error('rack_name', "Invalid rack ({})".format(rack_name))
  427. def clean_face(self):
  428. face = self.cleaned_data['face']
  429. if not face:
  430. return None
  431. try:
  432. return {
  433. 'front': 0,
  434. 'rear': 1,
  435. }[face.lower()]
  436. except KeyError:
  437. raise forms.ValidationError('Invalid rack face ({}); must be "front" or "rear".'.format(face))
  438. class ChildDeviceFromCSVForm(BaseDeviceFromCSVForm):
  439. parent = FlexibleModelChoiceField(queryset=Device.objects.all(), to_field_name='name', required=False,
  440. error_messages={'invalid_choice': 'Parent device not found.'})
  441. device_bay_name = forms.CharField(required=False)
  442. class Meta(BaseDeviceFromCSVForm.Meta):
  443. fields = ['name', 'device_role', 'tenant', 'manufacturer', 'model_name', 'platform', 'serial', 'asset_tag',
  444. 'parent', 'device_bay_name']
  445. def clean(self):
  446. super(ChildDeviceFromCSVForm, self).clean()
  447. parent = self.cleaned_data.get('parent')
  448. device_bay_name = self.cleaned_data.get('device_bay_name')
  449. # Validate device bay
  450. if parent and device_bay_name:
  451. try:
  452. device_bay = DeviceBay.objects.get(device=parent, name=device_bay_name)
  453. if device_bay.installed_device:
  454. self.add_error('device_bay_name',
  455. "Device bay ({} {}) is already occupied".format(parent, device_bay_name))
  456. else:
  457. self.instance.parent_bay = device_bay
  458. except DeviceBay.DoesNotExist:
  459. self.add_error('device_bay_name', "Parent device/bay ({} {}) not found".format(parent, device_bay_name))
  460. class DeviceImportForm(BulkImportForm, BootstrapMixin):
  461. csv = CSVDataField(csv_form=DeviceFromCSVForm)
  462. class ChildDeviceImportForm(BulkImportForm, BootstrapMixin):
  463. csv = CSVDataField(csv_form=ChildDeviceFromCSVForm)
  464. class DeviceBulkEditForm(forms.Form, BootstrapMixin):
  465. pk = forms.ModelMultipleChoiceField(queryset=Device.objects.all(), widget=forms.MultipleHiddenInput)
  466. device_type = forms.ModelChoiceField(queryset=DeviceType.objects.all(), required=False, label='Type')
  467. device_role = forms.ModelChoiceField(queryset=DeviceRole.objects.all(), required=False, label='Role')
  468. tenant = forms.TypedChoiceField(choices=bulkedit_tenant_choices, coerce=int, required=False, label='Tenant')
  469. platform = forms.TypedChoiceField(choices=bulkedit_platform_choices, coerce=int, required=False,
  470. label='Platform')
  471. status = forms.ChoiceField(choices=FORM_STATUS_CHOICES, required=False, initial='', label='Status')
  472. serial = forms.CharField(max_length=50, required=False, label='Serial Number')
  473. def device_site_choices():
  474. site_choices = Site.objects.annotate(device_count=Count('racks__devices'))
  475. return [(s.slug, u'{} ({})'.format(s.name, s.device_count)) for s in site_choices]
  476. def device_rack_group_choices():
  477. group_choices = RackGroup.objects.select_related('site').annotate(device_count=Count('racks__devices'))
  478. return [(g.pk, u'{} ({})'.format(g, g.device_count)) for g in group_choices]
  479. def device_role_choices():
  480. role_choices = DeviceRole.objects.annotate(device_count=Count('devices'))
  481. return [(r.slug, u'{} ({})'.format(r.name, r.device_count)) for r in role_choices]
  482. def device_tenant_choices():
  483. tenant_choices = Tenant.objects.annotate(device_count=Count('devices'))
  484. return [(t.slug, u'{} ({})'.format(t.name, t.device_count)) for t in tenant_choices]
  485. def device_type_choices():
  486. type_choices = DeviceType.objects.select_related('manufacturer').annotate(device_count=Count('instances'))
  487. return [(t.pk, u'{} ({})'.format(t, t.device_count)) for t in type_choices]
  488. def device_platform_choices():
  489. platform_choices = Platform.objects.annotate(device_count=Count('devices'))
  490. return [(p.slug, u'{} ({})'.format(p.name, p.device_count)) for p in platform_choices]
  491. class DeviceFilterForm(forms.Form, BootstrapMixin):
  492. site = forms.MultipleChoiceField(required=False, choices=device_site_choices,
  493. widget=forms.SelectMultiple(attrs={'size': 8}))
  494. rack_group_id = forms.MultipleChoiceField(required=False, choices=device_rack_group_choices, label='Rack Group',
  495. widget=forms.SelectMultiple(attrs={'size': 8}))
  496. role = forms.MultipleChoiceField(required=False, choices=device_role_choices,
  497. widget=forms.SelectMultiple(attrs={'size': 8}))
  498. tenant = forms.MultipleChoiceField(required=False, choices=device_tenant_choices,
  499. widget=forms.SelectMultiple(attrs={'size': 8}))
  500. device_type_id = forms.MultipleChoiceField(required=False, choices=device_type_choices, label='Type',
  501. widget=forms.SelectMultiple(attrs={'size': 8}))
  502. platform = forms.MultipleChoiceField(required=False, choices=device_platform_choices)
  503. status = forms.NullBooleanField(required=False, widget=forms.Select(choices=FORM_STATUS_CHOICES))
  504. #
  505. # Console ports
  506. #
  507. class ConsolePortForm(forms.ModelForm, BootstrapMixin):
  508. class Meta:
  509. model = ConsolePort
  510. fields = ['device', 'name']
  511. widgets = {
  512. 'device': forms.HiddenInput(),
  513. }
  514. class ConsolePortCreateForm(forms.Form, BootstrapMixin):
  515. name_pattern = ExpandableNameField(label='Name')
  516. class ConsoleConnectionCSVForm(forms.Form):
  517. console_server = FlexibleModelChoiceField(queryset=Device.objects.filter(device_type__is_console_server=True),
  518. to_field_name='name',
  519. error_messages={'invalid_choice': 'Console server not found'})
  520. cs_port = forms.CharField()
  521. device = FlexibleModelChoiceField(queryset=Device.objects.all(), to_field_name='name',
  522. error_messages={'invalid_choice': 'Device not found'})
  523. console_port = forms.CharField()
  524. status = forms.ChoiceField(choices=[('planned', 'Planned'), ('connected', 'Connected')])
  525. def clean(self):
  526. # Validate console server port
  527. if self.cleaned_data.get('console_server'):
  528. try:
  529. cs_port = ConsoleServerPort.objects.get(device=self.cleaned_data['console_server'],
  530. name=self.cleaned_data['cs_port'])
  531. if ConsolePort.objects.filter(cs_port=cs_port):
  532. raise forms.ValidationError("Console server port is already occupied (by {} {})"
  533. .format(cs_port.connected_console.device, cs_port.connected_console))
  534. except ConsoleServerPort.DoesNotExist:
  535. raise forms.ValidationError("Invalid console server port ({} {})"
  536. .format(self.cleaned_data['console_server'], self.cleaned_data['cs_port']))
  537. # Validate console port
  538. if self.cleaned_data.get('device'):
  539. try:
  540. console_port = ConsolePort.objects.get(device=self.cleaned_data['device'],
  541. name=self.cleaned_data['console_port'])
  542. if console_port.cs_port:
  543. raise forms.ValidationError("Console port is already connected (to {} {})"
  544. .format(console_port.cs_port.device, console_port.cs_port))
  545. except ConsolePort.DoesNotExist:
  546. raise forms.ValidationError("Invalid console port ({} {})"
  547. .format(self.cleaned_data['device'], self.cleaned_data['console_port']))
  548. class ConsoleConnectionImportForm(BulkImportForm, BootstrapMixin):
  549. csv = CSVDataField(csv_form=ConsoleConnectionCSVForm)
  550. def clean(self):
  551. records = self.cleaned_data.get('csv')
  552. if not records:
  553. return
  554. connection_list = []
  555. for i, record in enumerate(records, start=1):
  556. form = self.fields['csv'].csv_form(data=record)
  557. if form.is_valid():
  558. console_port = ConsolePort.objects.get(device=form.cleaned_data['device'],
  559. name=form.cleaned_data['console_port'])
  560. console_port.cs_port = ConsoleServerPort.objects.get(device=form.cleaned_data['console_server'],
  561. name=form.cleaned_data['cs_port'])
  562. if form.cleaned_data['status'] == 'planned':
  563. console_port.connection_status = CONNECTION_STATUS_PLANNED
  564. else:
  565. console_port.connection_status = CONNECTION_STATUS_CONNECTED
  566. connection_list.append(console_port)
  567. else:
  568. for field, errors in form.errors.items():
  569. for e in errors:
  570. self.add_error('csv', "Record {} {}: {}".format(i, field, e))
  571. self.cleaned_data['csv'] = connection_list
  572. class ConsolePortConnectionForm(forms.ModelForm, BootstrapMixin):
  573. rack = forms.ModelChoiceField(queryset=Rack.objects.all(), label='Rack', required=False,
  574. widget=forms.Select(attrs={'filter-for': 'console_server'}))
  575. console_server = forms.ModelChoiceField(queryset=Device.objects.all(), label='Console Server', required=False,
  576. widget=APISelect(api_url='/api/dcim/devices/?rack_id={{rack}}&is_console_server=True',
  577. attrs={'filter-for': 'cs_port'}))
  578. livesearch = forms.CharField(required=False, label='Console Server', widget=Livesearch(
  579. query_key='q', query_url='dcim-api:device_list', field_to_update='console_server')
  580. )
  581. cs_port = forms.ModelChoiceField(queryset=ConsoleServerPort.objects.all(), label='Port',
  582. widget=APISelect(api_url='/api/dcim/devices/{{console_server}}/console-server-ports/',
  583. disabled_indicator='connected_console'))
  584. class Meta:
  585. model = ConsolePort
  586. fields = ['rack', 'console_server', 'livesearch', 'cs_port', 'connection_status']
  587. labels = {
  588. 'cs_port': 'Port',
  589. 'connection_status': 'Status',
  590. }
  591. def __init__(self, *args, **kwargs):
  592. super(ConsolePortConnectionForm, self).__init__(*args, **kwargs)
  593. if not self.instance.pk:
  594. raise RuntimeError("ConsolePortConnectionForm must be initialized with an existing ConsolePort instance.")
  595. self.fields['rack'].queryset = Rack.objects.filter(site=self.instance.device.rack.site)
  596. self.fields['cs_port'].required = True
  597. self.fields['connection_status'].choices = CONNECTION_STATUS_CHOICES
  598. # Initialize console server choices
  599. if self.is_bound and self.data.get('rack'):
  600. self.fields['console_server'].queryset = Device.objects.filter(rack=self.data['rack'], device_type__is_console_server=True)
  601. elif self.initial.get('rack'):
  602. self.fields['console_server'].queryset = Device.objects.filter(rack=self.initial['rack'], device_type__is_console_server=True)
  603. else:
  604. self.fields['console_server'].choices = []
  605. # Initialize CS port choices
  606. if self.is_bound:
  607. self.fields['cs_port'].queryset = ConsoleServerPort.objects.filter(device__pk=self.data['console_server'])
  608. elif self.initial.get('console_server', None):
  609. self.fields['cs_port'].queryset = ConsoleServerPort.objects.filter(device__pk=self.initial['console_server'])
  610. else:
  611. self.fields['cs_port'].choices = []
  612. #
  613. # Console server ports
  614. #
  615. class ConsoleServerPortForm(forms.ModelForm, BootstrapMixin):
  616. class Meta:
  617. model = ConsoleServerPort
  618. fields = ['device', 'name']
  619. widgets = {
  620. 'device': forms.HiddenInput(),
  621. }
  622. class ConsoleServerPortCreateForm(forms.Form, BootstrapMixin):
  623. name_pattern = ExpandableNameField(label='Name')
  624. class ConsoleServerPortConnectionForm(forms.Form, BootstrapMixin):
  625. rack = forms.ModelChoiceField(queryset=Rack.objects.all(), label='Rack', required=False,
  626. widget=forms.Select(attrs={'filter-for': 'device'}))
  627. device = forms.ModelChoiceField(queryset=Device.objects.all(), label='Device', required=False,
  628. widget=APISelect(api_url='/api/dcim/devices/?rack_id={{rack}}',
  629. attrs={'filter-for': 'port'}))
  630. livesearch = forms.CharField(required=False, label='Device', widget=Livesearch(
  631. query_key='q', query_url='dcim-api:device_list', field_to_update='device')
  632. )
  633. port = forms.ModelChoiceField(queryset=ConsolePort.objects.all(), label='Port',
  634. widget=APISelect(api_url='/api/dcim/devices/{{device}}/console-ports/',
  635. disabled_indicator='cs_port'))
  636. connection_status = forms.BooleanField(required=False, initial=CONNECTION_STATUS_CONNECTED, label='Status',
  637. widget=forms.Select(choices=CONNECTION_STATUS_CHOICES))
  638. class Meta:
  639. fields = ['rack', 'device', 'livesearch', 'port', 'connection_status']
  640. labels = {
  641. 'connection_status': 'Status',
  642. }
  643. def __init__(self, consoleserverport, *args, **kwargs):
  644. super(ConsoleServerPortConnectionForm, self).__init__(*args, **kwargs)
  645. self.fields['rack'].queryset = Rack.objects.filter(site=consoleserverport.device.rack.site)
  646. # Initialize device choices
  647. if self.is_bound and self.data.get('rack'):
  648. self.fields['device'].queryset = Device.objects.filter(rack=self.data['rack'])
  649. elif self.initial.get('rack', None):
  650. self.fields['device'].queryset = Device.objects.filter(rack=self.initial['rack'])
  651. else:
  652. self.fields['device'].choices = []
  653. # Initialize port choices
  654. if self.is_bound:
  655. self.fields['port'].queryset = ConsolePort.objects.filter(device__pk=self.data['device'])
  656. elif self.initial.get('device', None):
  657. self.fields['port'].queryset = ConsolePort.objects.filter(device_pk=self.initial['device'])
  658. else:
  659. self.fields['port'].choices = []
  660. #
  661. # Power ports
  662. #
  663. class PowerPortForm(forms.ModelForm, BootstrapMixin):
  664. class Meta:
  665. model = PowerPort
  666. fields = ['device', 'name']
  667. widgets = {
  668. 'device': forms.HiddenInput(),
  669. }
  670. class PowerPortCreateForm(forms.Form, BootstrapMixin):
  671. name_pattern = ExpandableNameField(label='Name')
  672. class PowerConnectionCSVForm(forms.Form):
  673. pdu = FlexibleModelChoiceField(queryset=Device.objects.filter(device_type__is_pdu=True), to_field_name='name',
  674. error_messages={'invalid_choice': 'PDU not found.'})
  675. power_outlet = forms.CharField()
  676. device = FlexibleModelChoiceField(queryset=Device.objects.all(), to_field_name='name',
  677. error_messages={'invalid_choice': 'Device not found'})
  678. power_port = forms.CharField()
  679. status = forms.ChoiceField(choices=[('planned', 'Planned'), ('connected', 'Connected')])
  680. def clean(self):
  681. # Validate power outlet
  682. if self.cleaned_data.get('pdu'):
  683. try:
  684. power_outlet = PowerOutlet.objects.get(device=self.cleaned_data['pdu'],
  685. name=self.cleaned_data['power_outlet'])
  686. if PowerPort.objects.filter(power_outlet=power_outlet):
  687. raise forms.ValidationError("Power outlet is already occupied (by {} {})"
  688. .format(power_outlet.connected_port.device,
  689. power_outlet.connected_port))
  690. except PowerOutlet.DoesNotExist:
  691. raise forms.ValidationError("Invalid PDU port ({} {})"
  692. .format(self.cleaned_data['pdu'], self.cleaned_data['power_outlet']))
  693. # Validate power port
  694. if self.cleaned_data.get('device'):
  695. try:
  696. power_port = PowerPort.objects.get(device=self.cleaned_data['device'],
  697. name=self.cleaned_data['power_port'])
  698. if power_port.power_outlet:
  699. raise forms.ValidationError("Power port is already connected (to {} {})"
  700. .format(power_port.power_outlet.device, power_port.power_outlet))
  701. except PowerPort.DoesNotExist:
  702. raise forms.ValidationError("Invalid power port ({} {})"
  703. .format(self.cleaned_data['device'], self.cleaned_data['power_port']))
  704. class PowerConnectionImportForm(BulkImportForm, BootstrapMixin):
  705. csv = CSVDataField(csv_form=PowerConnectionCSVForm)
  706. def clean(self):
  707. records = self.cleaned_data.get('csv')
  708. if not records:
  709. return
  710. connection_list = []
  711. for i, record in enumerate(records, start=1):
  712. form = self.fields['csv'].csv_form(data=record)
  713. if form.is_valid():
  714. power_port = PowerPort.objects.get(device=form.cleaned_data['device'],
  715. name=form.cleaned_data['power_port'])
  716. power_port.power_outlet = PowerOutlet.objects.get(device=form.cleaned_data['pdu'],
  717. name=form.cleaned_data['power_outlet'])
  718. if form.cleaned_data['status'] == 'planned':
  719. power_port.connection_status = CONNECTION_STATUS_PLANNED
  720. else:
  721. power_port.connection_status = CONNECTION_STATUS_CONNECTED
  722. connection_list.append(power_port)
  723. else:
  724. for field, errors in form.errors.items():
  725. for e in errors:
  726. self.add_error('csv', "Record {} {}: {}".format(i, field, e))
  727. self.cleaned_data['csv'] = connection_list
  728. class PowerPortConnectionForm(forms.ModelForm, BootstrapMixin):
  729. rack = forms.ModelChoiceField(queryset=Rack.objects.all(), label='Rack', required=False,
  730. widget=forms.Select(attrs={'filter-for': 'pdu'}))
  731. pdu = forms.ModelChoiceField(queryset=Device.objects.all(), label='PDU', required=False,
  732. widget=APISelect(api_url='/api/dcim/devices/?rack_id={{rack}}&is_pdu=True',
  733. attrs={'filter-for': 'power_outlet'}))
  734. livesearch = forms.CharField(required=False, label='PDU', widget=Livesearch(
  735. query_key='q', query_url='dcim-api:device_list', field_to_update='pdu')
  736. )
  737. power_outlet = forms.ModelChoiceField(queryset=PowerOutlet.objects.all(), label='Outlet',
  738. widget=APISelect(api_url='/api/dcim/devices/{{pdu}}/power-outlets/',
  739. disabled_indicator='connected_port'))
  740. class Meta:
  741. model = PowerPort
  742. fields = ['rack', 'pdu', 'livesearch', 'power_outlet', 'connection_status']
  743. labels = {
  744. 'power_outlet': 'Outlet',
  745. 'connection_status': 'Status',
  746. }
  747. def __init__(self, *args, **kwargs):
  748. super(PowerPortConnectionForm, self).__init__(*args, **kwargs)
  749. if not self.instance.pk:
  750. raise RuntimeError("PowerPortConnectionForm must be initialized with an existing PowerPort instance.")
  751. self.fields['rack'].queryset = Rack.objects.filter(site=self.instance.device.rack.site)
  752. self.fields['power_outlet'].required = True
  753. self.fields['connection_status'].choices = CONNECTION_STATUS_CHOICES
  754. # Initialize PDU choices
  755. if self.is_bound and self.data.get('rack'):
  756. self.fields['pdu'].queryset = Device.objects.filter(rack=self.data['rack'], device_type__is_pdu=True)
  757. elif self.initial.get('rack', None):
  758. self.fields['pdu'].queryset = Device.objects.filter(rack=self.initial['rack'], device_type__is_pdu=True)
  759. else:
  760. self.fields['pdu'].choices = []
  761. # Initialize power outlet choices
  762. if self.is_bound:
  763. self.fields['power_outlet'].queryset = PowerOutlet.objects.filter(device__pk=self.data['pdu'])
  764. elif self.initial.get('pdu', None):
  765. self.fields['power_outlet'].queryset = PowerOutlet.objects.filter(device__pk=self.initial['pdu'])
  766. else:
  767. self.fields['power_outlet'].choices = []
  768. #
  769. # Power outlets
  770. #
  771. class PowerOutletForm(forms.ModelForm, BootstrapMixin):
  772. class Meta:
  773. model = PowerOutlet
  774. fields = ['device', 'name']
  775. widgets = {
  776. 'device': forms.HiddenInput(),
  777. }
  778. class PowerOutletCreateForm(forms.Form, BootstrapMixin):
  779. name_pattern = ExpandableNameField(label='Name')
  780. class PowerOutletConnectionForm(forms.Form, BootstrapMixin):
  781. rack = forms.ModelChoiceField(queryset=Rack.objects.all(), label='Rack', required=False,
  782. widget=forms.Select(attrs={'filter-for': 'device'}))
  783. device = forms.ModelChoiceField(queryset=Device.objects.all(), label='Device', required=False,
  784. widget=APISelect(api_url='/api/dcim/devices/?rack_id={{rack}}',
  785. attrs={'filter-for': 'port'}))
  786. livesearch = forms.CharField(required=False, label='Device', widget=Livesearch(
  787. query_key='q', query_url='dcim-api:device_list', field_to_update='device')
  788. )
  789. port = forms.ModelChoiceField(queryset=PowerPort.objects.all(), label='Port',
  790. widget=APISelect(api_url='/api/dcim/devices/{{device}}/power-ports/',
  791. disabled_indicator='power_outlet'))
  792. connection_status = forms.BooleanField(required=False, initial=CONNECTION_STATUS_CONNECTED, label='Status',
  793. widget=forms.Select(choices=CONNECTION_STATUS_CHOICES))
  794. class Meta:
  795. fields = ['rack', 'device', 'livesearch', 'port', 'connection_status']
  796. labels = {
  797. 'connection_status': 'Status',
  798. }
  799. def __init__(self, poweroutlet, *args, **kwargs):
  800. super(PowerOutletConnectionForm, self).__init__(*args, **kwargs)
  801. self.fields['rack'].queryset = Rack.objects.filter(site=poweroutlet.device.rack.site)
  802. # Initialize device choices
  803. if self.is_bound and self.data.get('rack'):
  804. self.fields['device'].queryset = Device.objects.filter(rack=self.data['rack'])
  805. elif self.initial.get('rack', None):
  806. self.fields['device'].queryset = Device.objects.filter(rack=self.initial['rack'])
  807. else:
  808. self.fields['device'].choices = []
  809. # Initialize port choices
  810. if self.is_bound:
  811. self.fields['port'].queryset = PowerPort.objects.filter(device__pk=self.data['device'])
  812. elif self.initial.get('device', None):
  813. self.fields['port'].queryset = PowerPort.objects.filter(device_pk=self.initial['device'])
  814. else:
  815. self.fields['port'].choices = []
  816. #
  817. # Interfaces
  818. #
  819. class InterfaceForm(forms.ModelForm, BootstrapMixin):
  820. class Meta:
  821. model = Interface
  822. fields = ['device', 'name', 'form_factor', 'mac_address', 'mgmt_only', 'description']
  823. widgets = {
  824. 'device': forms.HiddenInput(),
  825. }
  826. class InterfaceCreateForm(forms.ModelForm, BootstrapMixin):
  827. name_pattern = ExpandableNameField(label='Name')
  828. class Meta:
  829. model = Interface
  830. fields = ['name_pattern', 'form_factor', 'mac_address', 'mgmt_only', 'description']
  831. class InterfaceBulkCreateForm(InterfaceCreateForm, BootstrapMixin):
  832. pk = forms.ModelMultipleChoiceField(queryset=Device.objects.all(), widget=forms.MultipleHiddenInput)
  833. #
  834. # Interface connections
  835. #
  836. class InterfaceConnectionForm(forms.ModelForm, BootstrapMixin):
  837. interface_a = forms.ChoiceField(choices=[], widget=SelectWithDisabled, label='Interface')
  838. rack_b = forms.ModelChoiceField(queryset=Rack.objects.all(), label='Rack', required=False,
  839. widget=forms.Select(attrs={'filter-for': 'device_b'}))
  840. device_b = forms.ModelChoiceField(queryset=Device.objects.all(), label='Device', required=False,
  841. widget=APISelect(api_url='/api/dcim/devices/?rack_id={{rack_b}}',
  842. attrs={'filter-for': 'interface_b'}))
  843. livesearch = forms.CharField(required=False, label='Device', widget=Livesearch(
  844. query_key='q', query_url='dcim-api:device_list', field_to_update='device_b')
  845. )
  846. interface_b = forms.ModelChoiceField(queryset=Interface.objects.all(), label='Interface',
  847. widget=APISelect(api_url='/api/dcim/devices/{{device_b}}/interfaces/?type=physical',
  848. disabled_indicator='is_connected'))
  849. class Meta:
  850. model = InterfaceConnection
  851. fields = ['interface_a', 'rack_b', 'device_b', 'interface_b', 'livesearch', 'connection_status']
  852. def __init__(self, device_a, *args, **kwargs):
  853. super(InterfaceConnectionForm, self).__init__(*args, **kwargs)
  854. self.fields['rack_b'].queryset = Rack.objects.filter(site=device_a.rack.site)
  855. # Initialize interface A choices
  856. device_a_interfaces = Interface.objects.filter(device=device_a).exclude(form_factor=IFACE_FF_VIRTUAL) \
  857. .select_related('circuit', 'connected_as_a', 'connected_as_b')
  858. self.fields['interface_a'].choices = [
  859. (iface.id, {'label': iface.name, 'disabled': iface.is_connected}) for iface in device_a_interfaces
  860. ]
  861. # Initialize device_b choices if rack_b is set
  862. if self.is_bound and self.data.get('rack_b'):
  863. self.fields['device_b'].queryset = Device.objects.filter(rack__pk=self.data['rack_b'])
  864. elif self.initial.get('rack_b'):
  865. self.fields['device_b'].queryset = Device.objects.filter(rack=self.initial['rack_b'])
  866. else:
  867. self.fields['device_b'].choices = []
  868. # Initialize interface_b choices if device_b is set
  869. if self.is_bound:
  870. device_b_interfaces = Interface.objects.filter(device=self.data['device_b']) \
  871. .exclude(form_factor=IFACE_FF_VIRTUAL).select_related('circuit', 'connected_as_a', 'connected_as_b')
  872. elif self.initial.get('device_b'):
  873. device_b_interfaces = Interface.objects.filter(device=self.initial['device_b']) \
  874. .exclude(form_factor=IFACE_FF_VIRTUAL).select_related('circuit', 'connected_as_a', 'connected_as_b')
  875. else:
  876. device_b_interfaces = []
  877. self.fields['interface_b'].choices = [
  878. (iface.id, {'label': iface.name, 'disabled': iface.is_connected}) for iface in device_b_interfaces
  879. ]
  880. class InterfaceConnectionCSVForm(forms.Form):
  881. device_a = FlexibleModelChoiceField(queryset=Device.objects.all(), to_field_name='name',
  882. error_messages={'invalid_choice': 'Device A not found.'})
  883. interface_a = forms.CharField()
  884. device_b = FlexibleModelChoiceField(queryset=Device.objects.all(), to_field_name='name',
  885. error_messages={'invalid_choice': 'Device B not found.'})
  886. interface_b = forms.CharField()
  887. status = forms.ChoiceField(choices=[('planned', 'Planned'), ('connected', 'Connected')])
  888. def clean(self):
  889. # Validate interface A
  890. if self.cleaned_data.get('device_a'):
  891. try:
  892. interface_a = Interface.objects.get(device=self.cleaned_data['device_a'],
  893. name=self.cleaned_data['interface_a'])
  894. except Interface.DoesNotExist:
  895. raise forms.ValidationError("Invalid interface ({} {})"
  896. .format(self.cleaned_data['device_a'], self.cleaned_data['interface_a']))
  897. try:
  898. InterfaceConnection.objects.get(Q(interface_a=interface_a) | Q(interface_b=interface_a))
  899. raise forms.ValidationError("{} {} is already connected"
  900. .format(self.cleaned_data['device_a'], self.cleaned_data['interface_a']))
  901. except InterfaceConnection.DoesNotExist:
  902. pass
  903. # Validate interface B
  904. if self.cleaned_data.get('device_b'):
  905. try:
  906. interface_b = Interface.objects.get(device=self.cleaned_data['device_b'],
  907. name=self.cleaned_data['interface_b'])
  908. except Interface.DoesNotExist:
  909. raise forms.ValidationError("Invalid interface ({} {})"
  910. .format(self.cleaned_data['device_b'], self.cleaned_data['interface_b']))
  911. try:
  912. InterfaceConnection.objects.get(Q(interface_a=interface_b) | Q(interface_b=interface_b))
  913. raise forms.ValidationError("{} {} is already connected"
  914. .format(self.cleaned_data['device_b'], self.cleaned_data['interface_b']))
  915. except InterfaceConnection.DoesNotExist:
  916. pass
  917. class InterfaceConnectionImportForm(BulkImportForm, BootstrapMixin):
  918. csv = CSVDataField(csv_form=InterfaceConnectionCSVForm)
  919. def clean(self):
  920. records = self.cleaned_data.get('csv')
  921. if not records:
  922. return
  923. connection_list = []
  924. occupied_interfaces = []
  925. for i, record in enumerate(records, start=1):
  926. form = self.fields['csv'].csv_form(data=record)
  927. if form.is_valid():
  928. interface_a = Interface.objects.get(device=form.cleaned_data['device_a'],
  929. name=form.cleaned_data['interface_a'])
  930. if interface_a in occupied_interfaces:
  931. raise forms.ValidationError("{} {} found in multiple connections"
  932. .format(interface_a.device.name, interface_a.name))
  933. interface_b = Interface.objects.get(device=form.cleaned_data['device_b'],
  934. name=form.cleaned_data['interface_b'])
  935. if interface_b in occupied_interfaces:
  936. raise forms.ValidationError("{} {} found in multiple connections"
  937. .format(interface_b.device.name, interface_b.name))
  938. connection = InterfaceConnection(interface_a=interface_a, interface_b=interface_b)
  939. if form.cleaned_data['status'] == 'planned':
  940. connection.connection_status = CONNECTION_STATUS_PLANNED
  941. else:
  942. connection.connection_status = CONNECTION_STATUS_CONNECTED
  943. connection_list.append(connection)
  944. occupied_interfaces.append(interface_a)
  945. occupied_interfaces.append(interface_b)
  946. else:
  947. for field, errors in form.errors.items():
  948. for e in errors:
  949. self.add_error('csv', "Record {} {}: {}".format(i, field, e))
  950. self.cleaned_data['csv'] = connection_list
  951. class InterfaceConnectionDeletionForm(forms.Form, BootstrapMixin):
  952. confirm = forms.BooleanField(required=True)
  953. # Used for HTTP redirect upon successful deletion
  954. device = forms.ModelChoiceField(queryset=Device.objects.all(), widget=forms.HiddenInput(), required=False)
  955. #
  956. # Device bays
  957. #
  958. class DeviceBayForm(forms.ModelForm, BootstrapMixin):
  959. class Meta:
  960. model = DeviceBay
  961. fields = ['device', 'name']
  962. widgets = {
  963. 'device': forms.HiddenInput(),
  964. }
  965. class DeviceBayCreateForm(forms.Form, BootstrapMixin):
  966. name_pattern = ExpandableNameField(label='Name')
  967. class PopulateDeviceBayForm(forms.Form, BootstrapMixin):
  968. installed_device = forms.ModelChoiceField(queryset=Device.objects.all(), label='Child Device',
  969. help_text="Child devices must first be created within the rack occupied "
  970. "by the parent device. Then they can be assigned to a bay.")
  971. def __init__(self, device_bay, *args, **kwargs):
  972. super(PopulateDeviceBayForm, self).__init__(*args, **kwargs)
  973. children_queryset = Device.objects.filter(rack=device_bay.device.rack,
  974. parent_bay__isnull=True,
  975. device_type__u_height=0,
  976. device_type__subdevice_role=SUBDEVICE_ROLE_CHILD)\
  977. .exclude(pk=device_bay.device.pk)
  978. self.fields['installed_device'].queryset = children_queryset
  979. #
  980. # Connections
  981. #
  982. class ConsoleConnectionFilterForm(forms.Form, BootstrapMixin):
  983. site = forms.ModelChoiceField(required=False, queryset=Site.objects.all(), to_field_name='slug')
  984. class PowerConnectionFilterForm(forms.Form, BootstrapMixin):
  985. site = forms.ModelChoiceField(required=False, queryset=Site.objects.all(), to_field_name='slug')
  986. class InterfaceConnectionFilterForm(forms.Form, BootstrapMixin):
  987. site = forms.ModelChoiceField(required=False, queryset=Site.objects.all(), to_field_name='slug')
  988. #
  989. # IP addresses
  990. #
  991. class IPAddressForm(forms.ModelForm, BootstrapMixin):
  992. set_as_primary = forms.BooleanField(label='Set as primary IP for device', required=False)
  993. class Meta:
  994. model = IPAddress
  995. fields = ['address', 'vrf', 'interface', 'set_as_primary']
  996. help_texts = {
  997. 'address': 'IPv4 or IPv6 address (with mask)'
  998. }
  999. def __init__(self, device, *args, **kwargs):
  1000. super(IPAddressForm, self).__init__(*args, **kwargs)
  1001. self.fields['vrf'].empty_label = 'Global'
  1002. self.fields['interface'].queryset = device.interfaces.all()
  1003. self.fields['interface'].required = True
  1004. # If this device does not have any IP addresses assigned, default to setting the first IP as its primary
  1005. if not IPAddress.objects.filter(interface__device=device).count():
  1006. self.fields['set_as_primary'].initial = True
  1007. #
  1008. # Interfaces
  1009. #
  1010. class ModuleForm(forms.ModelForm, BootstrapMixin):
  1011. class Meta:
  1012. model = Module
  1013. fields = ['name', 'manufacturer', 'part_id', 'serial']