forms.py 54 KB

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