views.py 33 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872
  1. from __future__ import unicode_literals
  2. from collections import OrderedDict
  3. from copy import deepcopy
  4. from django.conf import settings
  5. from django.contrib import messages
  6. from django.contrib.contenttypes.models import ContentType
  7. from django.core.exceptions import ValidationError
  8. from django.db import transaction, IntegrityError
  9. from django.db.models import ProtectedError
  10. from django.forms import CharField, Form, ModelMultipleChoiceField, MultipleHiddenInput, Textarea, TypedChoiceField
  11. from django.shortcuts import get_object_or_404, redirect, render
  12. from django.template import TemplateSyntaxError
  13. from django.urls import reverse
  14. from django.utils.html import escape
  15. from django.utils.http import is_safe_url
  16. from django.utils.safestring import mark_safe
  17. from django.views.generic import View
  18. from django_tables2 import RequestConfig
  19. from extras.models import CustomField, CustomFieldValue, ExportTemplate, UserAction
  20. from utilities.utils import queryset_to_csv
  21. from utilities.forms import BootstrapMixin, CSVDataField
  22. from .error_handlers import handle_protectederror
  23. from .forms import ConfirmationForm
  24. from .paginator import EnhancedPaginator
  25. class CustomFieldQueryset:
  26. """
  27. Annotate custom fields on objects within a QuerySet.
  28. """
  29. def __init__(self, queryset, custom_fields):
  30. self.queryset = queryset
  31. self.custom_fields = custom_fields
  32. def __iter__(self):
  33. for obj in self.queryset:
  34. values_dict = {cfv.field_id: cfv.value for cfv in obj.custom_field_values.all()}
  35. obj.custom_fields = OrderedDict([(field, values_dict.get(field.pk)) for field in self.custom_fields])
  36. yield obj
  37. class GetReturnURLMixin(object):
  38. """
  39. Provides logic for determining where a user should be redirected after processing a form.
  40. """
  41. default_return_url = None
  42. def get_return_url(self, request, obj):
  43. query_param = request.GET.get('return_url')
  44. if query_param and is_safe_url(url=query_param, host=request.get_host()):
  45. return query_param
  46. elif obj.pk and hasattr(obj, 'get_absolute_url'):
  47. return obj.get_absolute_url()
  48. elif self.default_return_url is not None:
  49. return reverse(self.default_return_url)
  50. return reverse('home')
  51. class ObjectListView(View):
  52. """
  53. List a series of objects.
  54. queryset: The queryset of objects to display
  55. filter: A django-filter FilterSet that is applied to the queryset
  56. filter_form: The form used to render filter options
  57. table: The django-tables2 Table used to render the objects list
  58. template_name: The name of the template
  59. """
  60. queryset = None
  61. filter = None
  62. filter_form = None
  63. table = None
  64. template_name = None
  65. def get(self, request):
  66. model = self.queryset.model
  67. content_type = ContentType.objects.get_for_model(model)
  68. if self.filter:
  69. self.queryset = self.filter(request.GET, self.queryset).qs
  70. # If this type of object has one or more custom fields, prefetch any relevant custom field values
  71. custom_fields = CustomField.objects.filter(obj_type=ContentType.objects.get_for_model(model))\
  72. .prefetch_related('choices')
  73. if custom_fields:
  74. self.queryset = self.queryset.prefetch_related('custom_field_values')
  75. # Check for export template rendering
  76. if request.GET.get('export'):
  77. et = get_object_or_404(ExportTemplate, content_type=content_type, name=request.GET.get('export'))
  78. queryset = CustomFieldQueryset(self.queryset, custom_fields) if custom_fields else self.queryset
  79. try:
  80. return et.render_to_response(queryset)
  81. except TemplateSyntaxError:
  82. messages.error(
  83. request,
  84. "There was an error rendering the selected export template ({}).".format(et.name)
  85. )
  86. # Fall back to built-in CSV export if no template was specified
  87. elif 'export' in request.GET and hasattr(model, 'to_csv'):
  88. return queryset_to_csv(self.queryset)
  89. # Provide a hook to tweak the queryset based on the request immediately prior to rendering the object list
  90. self.queryset = self.alter_queryset(request)
  91. # Compile user model permissions for access from within the template
  92. perm_base_name = '{}.{{}}_{}'.format(model._meta.app_label, model._meta.model_name)
  93. permissions = {p: request.user.has_perm(perm_base_name.format(p)) for p in ['add', 'change', 'delete']}
  94. # Construct the table based on the user's permissions
  95. table = self.table(self.queryset)
  96. if 'pk' in table.base_columns and (permissions['change'] or permissions['delete']):
  97. table.columns.show('pk')
  98. # Apply the request context
  99. paginate = {
  100. 'klass': EnhancedPaginator,
  101. 'per_page': request.GET.get('per_page', settings.PAGINATE_COUNT)
  102. }
  103. RequestConfig(request, paginate).configure(table)
  104. context = {
  105. 'content_type': content_type,
  106. 'table': table,
  107. 'permissions': permissions,
  108. 'filter_form': self.filter_form(request.GET, label_suffix='') if self.filter_form else None,
  109. }
  110. context.update(self.extra_context())
  111. return render(request, self.template_name, context)
  112. def alter_queryset(self, request):
  113. # .all() is necessary to avoid caching queries
  114. return self.queryset.all()
  115. def extra_context(self):
  116. return {}
  117. class ObjectEditView(GetReturnURLMixin, View):
  118. """
  119. Create or edit a single object.
  120. model: The model of the object being edited
  121. model_form: The form used to create or edit the object
  122. template_name: The name of the template
  123. default_return_url: The name of the URL used to display a list of this object type
  124. """
  125. model = None
  126. model_form = None
  127. template_name = 'utilities/obj_edit.html'
  128. def get_object(self, kwargs):
  129. # Look up object by slug or PK. Return None if neither was provided.
  130. if 'slug' in kwargs:
  131. return get_object_or_404(self.model, slug=kwargs['slug'])
  132. elif 'pk' in kwargs:
  133. return get_object_or_404(self.model, pk=kwargs['pk'])
  134. return self.model()
  135. def alter_obj(self, obj, request, url_args, url_kwargs):
  136. # Allow views to add extra info to an object before it is processed. For example, a parent object can be defined
  137. # given some parameter from the request URL.
  138. return obj
  139. def get(self, request, *args, **kwargs):
  140. obj = self.get_object(kwargs)
  141. obj = self.alter_obj(obj, request, args, kwargs)
  142. # Parse initial data manually to avoid setting field values as lists
  143. initial_data = {k: request.GET[k] for k in request.GET}
  144. form = self.model_form(instance=obj, initial=initial_data)
  145. return render(request, self.template_name, {
  146. 'obj': obj,
  147. 'obj_type': self.model._meta.verbose_name,
  148. 'form': form,
  149. 'return_url': self.get_return_url(request, obj),
  150. })
  151. def post(self, request, *args, **kwargs):
  152. obj = self.get_object(kwargs)
  153. obj = self.alter_obj(obj, request, args, kwargs)
  154. form = self.model_form(request.POST, request.FILES, instance=obj)
  155. if form.is_valid():
  156. obj_created = not form.instance.pk
  157. obj = form.save()
  158. msg = 'Created ' if obj_created else 'Modified '
  159. msg += self.model._meta.verbose_name
  160. if hasattr(obj, 'get_absolute_url'):
  161. msg = '{} <a href="{}">{}</a>'.format(msg, obj.get_absolute_url(), escape(obj))
  162. else:
  163. msg = '{} {}'.format(msg, escape(obj))
  164. messages.success(request, mark_safe(msg))
  165. if obj_created:
  166. UserAction.objects.log_create(request.user, obj, msg)
  167. else:
  168. UserAction.objects.log_edit(request.user, obj, msg)
  169. if '_addanother' in request.POST:
  170. return redirect(request.get_full_path())
  171. return_url = form.cleaned_data.get('return_url')
  172. if return_url is not None and is_safe_url(url=return_url, host=request.get_host()):
  173. return redirect(return_url)
  174. else:
  175. return redirect(self.get_return_url(request, obj))
  176. return render(request, self.template_name, {
  177. 'obj': obj,
  178. 'obj_type': self.model._meta.verbose_name,
  179. 'form': form,
  180. 'return_url': self.get_return_url(request, obj),
  181. })
  182. class ObjectDeleteView(GetReturnURLMixin, View):
  183. """
  184. Delete a single object.
  185. model: The model of the object being deleted
  186. template_name: The name of the template
  187. default_return_url: Name of the URL to which the user is redirected after deleting the object
  188. """
  189. model = None
  190. template_name = 'utilities/obj_delete.html'
  191. def get_object(self, kwargs):
  192. # Look up object by slug if one has been provided. Otherwise, use PK.
  193. if 'slug' in kwargs:
  194. return get_object_or_404(self.model, slug=kwargs['slug'])
  195. else:
  196. return get_object_or_404(self.model, pk=kwargs['pk'])
  197. def get(self, request, **kwargs):
  198. obj = self.get_object(kwargs)
  199. form = ConfirmationForm(initial=request.GET)
  200. return render(request, self.template_name, {
  201. 'obj': obj,
  202. 'form': form,
  203. 'obj_type': self.model._meta.verbose_name,
  204. 'return_url': self.get_return_url(request, obj),
  205. })
  206. def post(self, request, **kwargs):
  207. obj = self.get_object(kwargs)
  208. form = ConfirmationForm(request.POST)
  209. if form.is_valid():
  210. try:
  211. obj.delete()
  212. except ProtectedError as e:
  213. handle_protectederror(obj, request, e)
  214. return redirect(obj.get_absolute_url())
  215. msg = 'Deleted {} {}'.format(self.model._meta.verbose_name, obj)
  216. messages.success(request, msg)
  217. UserAction.objects.log_delete(request.user, obj, msg)
  218. return_url = form.cleaned_data.get('return_url')
  219. if return_url is not None and is_safe_url(url=return_url, host=request.get_host()):
  220. return redirect(return_url)
  221. else:
  222. return redirect(self.get_return_url(request, obj))
  223. return render(request, self.template_name, {
  224. 'obj': obj,
  225. 'form': form,
  226. 'obj_type': self.model._meta.verbose_name,
  227. 'return_url': self.get_return_url(request, obj),
  228. })
  229. class BulkCreateView(View):
  230. """
  231. Create new objects in bulk.
  232. form: Form class which provides the `pattern` field
  233. model_form: The ModelForm used to create individual objects
  234. template_name: The name of the template
  235. default_return_url: Name of the URL to which the user is redirected after creating the objects
  236. """
  237. form = None
  238. model_form = None
  239. pattern_target = ''
  240. template_name = None
  241. default_return_url = 'home'
  242. def get(self, request):
  243. # Set initial values for visible form fields from query args
  244. initial = {}
  245. for field in getattr(self.model_form._meta, 'fields', []):
  246. if request.GET.get(field):
  247. initial[field] = request.GET[field]
  248. form = self.form()
  249. model_form = self.model_form(initial=initial)
  250. return render(request, self.template_name, {
  251. 'obj_type': self.model_form._meta.model._meta.verbose_name,
  252. 'form': form,
  253. 'model_form': model_form,
  254. 'return_url': reverse(self.default_return_url),
  255. })
  256. def post(self, request):
  257. model = self.model_form._meta.model
  258. form = self.form(request.POST)
  259. model_form = self.model_form(request.POST)
  260. if form.is_valid():
  261. pattern = form.cleaned_data['pattern']
  262. new_objs = []
  263. try:
  264. with transaction.atomic():
  265. # Create objects from the expanded. Abort the transaction on the first validation error.
  266. for value in pattern:
  267. # Reinstantiate the model form each time to avoid overwriting the same instance. Use a mutable
  268. # copy of the POST QueryDict so that we can update the target field value.
  269. model_form = self.model_form(request.POST.copy())
  270. model_form.data[self.pattern_target] = value
  271. # Validate each new object independently.
  272. if model_form.is_valid():
  273. obj = model_form.save()
  274. new_objs.append(obj)
  275. else:
  276. # Copy any errors on the pattern target field to the pattern form.
  277. errors = model_form.errors.as_data()
  278. if errors.get(self.pattern_target):
  279. form.add_error('pattern', errors[self.pattern_target])
  280. # Raise an IntegrityError to break the for loop and abort the transaction.
  281. raise IntegrityError()
  282. # If we make it to this point, validation has succeeded on all new objects.
  283. msg = "Added {} {}".format(len(new_objs), model._meta.verbose_name_plural)
  284. messages.success(request, msg)
  285. UserAction.objects.log_bulk_create(request.user, ContentType.objects.get_for_model(model), msg)
  286. if '_addanother' in request.POST:
  287. return redirect(request.path)
  288. return redirect(self.default_return_url)
  289. except IntegrityError:
  290. pass
  291. return render(request, self.template_name, {
  292. 'form': form,
  293. 'model_form': model_form,
  294. 'obj_type': model._meta.verbose_name,
  295. 'return_url': reverse(self.default_return_url),
  296. })
  297. class BulkImportView(View):
  298. """
  299. Import objects in bulk (CSV format).
  300. model_form: The form used to create each imported object
  301. table: The django-tables2 Table used to render the list of imported objects
  302. template_name: The name of the template
  303. default_return_url: The name of the URL to use for the cancel button
  304. widget_attrs: A dict of attributes to apply to the import widget (e.g. to require a session key)
  305. """
  306. model_form = None
  307. table = None
  308. default_return_url = None
  309. template_name = 'utilities/obj_import.html'
  310. widget_attrs = {}
  311. def _import_form(self, *args, **kwargs):
  312. fields = self.model_form().fields.keys()
  313. required_fields = [name for name, field in self.model_form().fields.items() if field.required]
  314. class ImportForm(BootstrapMixin, Form):
  315. csv = CSVDataField(fields=fields, required_fields=required_fields, widget=Textarea(attrs=self.widget_attrs))
  316. return ImportForm(*args, **kwargs)
  317. def _save_obj(self, obj_form):
  318. """
  319. Provide a hook to modify the object immediately before saving it (e.g. to encrypt secret data).
  320. """
  321. return obj_form.save()
  322. def get(self, request):
  323. return render(request, self.template_name, {
  324. 'form': self._import_form(),
  325. 'fields': self.model_form().fields,
  326. 'obj_type': self.model_form._meta.model._meta.verbose_name,
  327. 'return_url': self.default_return_url,
  328. })
  329. def post(self, request):
  330. new_objs = []
  331. form = self._import_form(request.POST)
  332. if form.is_valid():
  333. try:
  334. # Iterate through CSV data and bind each row to a new model form instance.
  335. with transaction.atomic():
  336. for row, data in enumerate(form.cleaned_data['csv'], start=1):
  337. obj_form = self.model_form(data)
  338. if obj_form.is_valid():
  339. obj = self._save_obj(obj_form)
  340. new_objs.append(obj)
  341. else:
  342. for field, err in obj_form.errors.items():
  343. form.add_error('csv', "Row {} {}: {}".format(row, field, err[0]))
  344. raise ValidationError("")
  345. # Compile a table containing the imported objects
  346. obj_table = self.table(new_objs)
  347. if new_objs:
  348. msg = 'Imported {} {}'.format(len(new_objs), new_objs[0]._meta.verbose_name_plural)
  349. messages.success(request, msg)
  350. UserAction.objects.log_import(request.user, ContentType.objects.get_for_model(new_objs[0]), msg)
  351. return render(request, "import_success.html", {
  352. 'table': obj_table,
  353. 'return_url': self.default_return_url,
  354. })
  355. except ValidationError:
  356. pass
  357. return render(request, self.template_name, {
  358. 'form': form,
  359. 'fields': self.model_form().fields,
  360. 'obj_type': self.model_form._meta.model._meta.verbose_name,
  361. 'return_url': self.default_return_url,
  362. })
  363. class BulkEditView(View):
  364. """
  365. Edit objects in bulk.
  366. cls: The model of the objects being edited
  367. parent_cls: The model of the parent object (if any)
  368. queryset: Custom queryset to use when retrieving objects (e.g. to select related objects)
  369. filter: FilterSet to apply when deleting by QuerySet
  370. table: The table used to display devices being edited
  371. form: The form class used to edit objects in bulk
  372. template_name: The name of the template
  373. default_return_url: Name of the URL to which the user is redirected after editing the objects (can be overridden by
  374. POSTing return_url)
  375. """
  376. cls = None
  377. parent_cls = None
  378. queryset = None
  379. filter = None
  380. table = None
  381. form = None
  382. template_name = 'utilities/obj_bulk_edit.html'
  383. default_return_url = 'home'
  384. def get(self, request):
  385. return redirect(self.default_return_url)
  386. def post(self, request, **kwargs):
  387. # Attempt to derive parent object if a parent class has been given
  388. if self.parent_cls:
  389. parent_obj = get_object_or_404(self.parent_cls, **kwargs)
  390. else:
  391. parent_obj = None
  392. # Determine URL to redirect users upon modification of objects
  393. posted_return_url = request.POST.get('return_url')
  394. if posted_return_url and is_safe_url(url=posted_return_url, host=request.get_host()):
  395. return_url = posted_return_url
  396. elif parent_obj:
  397. return_url = parent_obj.get_absolute_url()
  398. else:
  399. return_url = reverse(self.default_return_url)
  400. # Are we editing *all* objects in the queryset or just a selected subset?
  401. if request.POST.get('_all') and self.filter is not None:
  402. pk_list = [obj.pk for obj in self.filter(request.GET, self.cls.objects.only('pk')).qs]
  403. else:
  404. pk_list = [int(pk) for pk in request.POST.getlist('pk')]
  405. if '_apply' in request.POST:
  406. form = self.form(self.cls, request.POST)
  407. if form.is_valid():
  408. custom_fields = form.custom_fields if hasattr(form, 'custom_fields') else []
  409. standard_fields = [field for field in form.fields if field not in custom_fields and field != 'pk']
  410. # Update standard fields. If a field is listed in _nullify, delete its value.
  411. nullified_fields = request.POST.getlist('_nullify')
  412. fields_to_update = {}
  413. for field in standard_fields:
  414. if field in form.nullable_fields and field in nullified_fields:
  415. if isinstance(form.fields[field], CharField):
  416. fields_to_update[field] = ''
  417. else:
  418. fields_to_update[field] = None
  419. elif form.cleaned_data[field] not in (None, ''):
  420. fields_to_update[field] = form.cleaned_data[field]
  421. updated_count = self.cls.objects.filter(pk__in=pk_list).update(**fields_to_update)
  422. # Update custom fields for objects
  423. if custom_fields:
  424. objs_updated = self.update_custom_fields(pk_list, form, custom_fields, nullified_fields)
  425. if objs_updated and not updated_count:
  426. updated_count = objs_updated
  427. if updated_count:
  428. msg = 'Updated {} {}'.format(updated_count, self.cls._meta.verbose_name_plural)
  429. messages.success(self.request, msg)
  430. UserAction.objects.log_bulk_edit(request.user, ContentType.objects.get_for_model(self.cls), msg)
  431. return redirect(return_url)
  432. else:
  433. initial_data = request.POST.copy()
  434. initial_data['pk'] = pk_list
  435. form = self.form(self.cls, initial=initial_data)
  436. # Retrieve objects being edited
  437. queryset = self.queryset or self.cls.objects.all()
  438. table = self.table(queryset.filter(pk__in=pk_list), orderable=False)
  439. if not table.rows:
  440. messages.warning(request, "No {} were selected.".format(self.cls._meta.verbose_name_plural))
  441. return redirect(return_url)
  442. return render(request, self.template_name, {
  443. 'form': form,
  444. 'table': table,
  445. 'obj_type_plural': self.cls._meta.verbose_name_plural,
  446. 'return_url': return_url,
  447. })
  448. def update_custom_fields(self, pk_list, form, fields, nullified_fields):
  449. obj_type = ContentType.objects.get_for_model(self.cls)
  450. objs_updated = False
  451. for name in fields:
  452. field = form.fields[name].model
  453. # Setting the field to null
  454. if name in form.nullable_fields and name in nullified_fields:
  455. # Delete all CustomFieldValues for instances of this field belonging to the selected objects.
  456. CustomFieldValue.objects.filter(field=field, obj_type=obj_type, obj_id__in=pk_list).delete()
  457. objs_updated = True
  458. # Updating the value of the field
  459. elif form.cleaned_data[name] not in [None, '']:
  460. # Check for zero value (bulk editing)
  461. if isinstance(form.fields[name], TypedChoiceField) and form.cleaned_data[name] == 0:
  462. serialized_value = field.serialize_value(None)
  463. else:
  464. serialized_value = field.serialize_value(form.cleaned_data[name])
  465. # Gather any pre-existing CustomFieldValues for the objects being edited.
  466. existing_cfvs = CustomFieldValue.objects.filter(field=field, obj_type=obj_type, obj_id__in=pk_list)
  467. # Determine which objects have an existing CFV to update and which need a new CFV created.
  468. update_list = [cfv['obj_id'] for cfv in existing_cfvs.values()]
  469. create_list = list(set(pk_list) - set(update_list))
  470. # Creating/updating CFVs
  471. if serialized_value:
  472. existing_cfvs.update(serialized_value=serialized_value)
  473. CustomFieldValue.objects.bulk_create([
  474. CustomFieldValue(field=field, obj_type=obj_type, obj_id=pk, serialized_value=serialized_value)
  475. for pk in create_list
  476. ])
  477. # Deleting CFVs
  478. else:
  479. existing_cfvs.delete()
  480. objs_updated = True
  481. return len(pk_list) if objs_updated else 0
  482. class BulkDeleteView(View):
  483. """
  484. Delete objects in bulk.
  485. cls: The model of the objects being deleted
  486. parent_cls: The model of the parent object (if any)
  487. queryset: Custom queryset to use when retrieving objects (e.g. to select related objects)
  488. filter: FilterSet to apply when deleting by QuerySet
  489. table: The table used to display devices being deleted
  490. form: The form class used to delete objects in bulk
  491. template_name: The name of the template
  492. default_return_url: Name of the URL to which the user is redirected after deleting the objects (can be overriden by
  493. POSTing return_url)
  494. """
  495. cls = None
  496. parent_cls = None
  497. queryset = None
  498. filter = None
  499. table = None
  500. form = None
  501. template_name = 'utilities/obj_bulk_delete.html'
  502. default_return_url = 'home'
  503. def get(self, request):
  504. return redirect(self.default_return_url)
  505. def post(self, request, **kwargs):
  506. # Attempt to derive parent object if a parent class has been given
  507. if self.parent_cls:
  508. parent_obj = get_object_or_404(self.parent_cls, **kwargs)
  509. else:
  510. parent_obj = None
  511. # Determine URL to redirect users upon deletion of objects
  512. posted_return_url = request.POST.get('return_url')
  513. if posted_return_url and is_safe_url(url=posted_return_url, host=request.get_host()):
  514. return_url = posted_return_url
  515. elif parent_obj:
  516. return_url = parent_obj.get_absolute_url()
  517. else:
  518. return_url = reverse(self.default_return_url)
  519. # Are we deleting *all* objects in the queryset or just a selected subset?
  520. if request.POST.get('_all') and self.filter is not None:
  521. pk_list = [obj.pk for obj in self.filter(request.GET, self.cls.objects.only('pk')).qs]
  522. else:
  523. pk_list = [int(pk) for pk in request.POST.getlist('pk')]
  524. form_cls = self.get_form()
  525. if '_confirm' in request.POST:
  526. form = form_cls(request.POST)
  527. if form.is_valid():
  528. # Delete objects
  529. queryset = self.cls.objects.filter(pk__in=pk_list)
  530. try:
  531. deleted_count = queryset.delete()[1][self.cls._meta.label]
  532. except ProtectedError as e:
  533. handle_protectederror(list(queryset), request, e)
  534. return redirect(return_url)
  535. msg = 'Deleted {} {}'.format(deleted_count, self.cls._meta.verbose_name_plural)
  536. messages.success(request, msg)
  537. UserAction.objects.log_bulk_delete(request.user, ContentType.objects.get_for_model(self.cls), msg)
  538. return redirect(return_url)
  539. else:
  540. form = form_cls(initial={'pk': pk_list, 'return_url': return_url})
  541. # Retrieve objects being deleted
  542. queryset = self.queryset or self.cls.objects.all()
  543. table = self.table(queryset.filter(pk__in=pk_list), orderable=False)
  544. if not table.rows:
  545. messages.warning(request, "No {} were selected for deletion.".format(self.cls._meta.verbose_name_plural))
  546. return redirect(return_url)
  547. return render(request, self.template_name, {
  548. 'form': form,
  549. 'parent_obj': parent_obj,
  550. 'obj_type_plural': self.cls._meta.verbose_name_plural,
  551. 'table': table,
  552. 'return_url': return_url,
  553. })
  554. def get_form(self):
  555. """
  556. Provide a standard bulk delete form if none has been specified for the view
  557. """
  558. class BulkDeleteForm(ConfirmationForm):
  559. pk = ModelMultipleChoiceField(queryset=self.cls.objects.all(), widget=MultipleHiddenInput)
  560. if self.form:
  561. return self.form
  562. return BulkDeleteForm
  563. #
  564. # Device/VirtualMachine components
  565. #
  566. class ComponentCreateView(View):
  567. """
  568. Add one or more components (e.g. interfaces, console ports, etc.) to a Device or VirtualMachine.
  569. """
  570. parent_model = None
  571. parent_field = None
  572. model = None
  573. form = None
  574. model_form = None
  575. template_name = None
  576. def get(self, request, pk):
  577. parent = get_object_or_404(self.parent_model, pk=pk)
  578. form = self.form(parent, initial=request.GET)
  579. return render(request, self.template_name, {
  580. 'parent': parent,
  581. 'component_type': self.model._meta.verbose_name,
  582. 'form': form,
  583. 'return_url': parent.get_absolute_url(),
  584. })
  585. def post(self, request, pk):
  586. parent = get_object_or_404(self.parent_model, pk=pk)
  587. form = self.form(parent, request.POST)
  588. if form.is_valid():
  589. new_components = []
  590. data = deepcopy(form.cleaned_data)
  591. for name in form.cleaned_data['name_pattern']:
  592. component_data = {
  593. self.parent_field: parent.pk,
  594. 'name': name,
  595. }
  596. # Replace objects with their primary key to keep component_form.clean() happy
  597. for k, v in data.items():
  598. if hasattr(v, 'pk'):
  599. component_data[k] = v.pk
  600. else:
  601. component_data[k] = v
  602. component_form = self.model_form(component_data)
  603. if component_form.is_valid():
  604. new_components.append(component_form.save(commit=False))
  605. else:
  606. for field, errors in component_form.errors.as_data().items():
  607. # Assign errors on the child form's name field to name_pattern on the parent form
  608. if field == 'name':
  609. field = 'name_pattern'
  610. for e in errors:
  611. form.add_error(field, '{}: {}'.format(name, ', '.join(e)))
  612. if not form.errors:
  613. self.model.objects.bulk_create(new_components)
  614. messages.success(request, "Added {} {} to {}.".format(
  615. len(new_components), self.model._meta.verbose_name_plural, parent
  616. ))
  617. if '_addanother' in request.POST:
  618. return redirect(request.path)
  619. else:
  620. return redirect(parent.get_absolute_url())
  621. return render(request, self.template_name, {
  622. 'parent': parent,
  623. 'component_type': self.model._meta.verbose_name,
  624. 'form': form,
  625. 'return_url': parent.get_absolute_url(),
  626. })
  627. class ComponentEditView(ObjectEditView):
  628. parent_field = None
  629. def get_return_url(self, request, obj):
  630. return getattr(obj, self.parent_field).get_absolute_url()
  631. class ComponentDeleteView(ObjectDeleteView):
  632. parent_field = None
  633. def get_return_url(self, request, obj):
  634. return getattr(obj, self.parent_field).get_absolute_url()
  635. class BulkComponentCreateView(View):
  636. """
  637. Add one or more components (e.g. interfaces, console ports, etc.) to a set of Devices or VirtualMachines.
  638. """
  639. parent_model = None
  640. parent_field = None
  641. form = None
  642. model = None
  643. model_form = None
  644. filter = None
  645. table = None
  646. template_name = 'utilities/obj_bulk_add_component.html'
  647. default_return_url = 'home'
  648. def post(self, request):
  649. # Are we editing *all* objects in the queryset or just a selected subset?
  650. if request.POST.get('_all') and self.filter is not None:
  651. pk_list = [obj.pk for obj in self.filter(request.GET, self.model.objects.only('pk')).qs]
  652. else:
  653. pk_list = [int(pk) for pk in request.POST.getlist('pk')]
  654. # Determine URL to redirect users upon modification of objects
  655. posted_return_url = request.POST.get('return_url')
  656. if posted_return_url and is_safe_url(url=posted_return_url, host=request.get_host()):
  657. return_url = posted_return_url
  658. else:
  659. return_url = reverse(self.default_return_url)
  660. selected_objects = self.parent_model.objects.filter(pk__in=pk_list)
  661. if not selected_objects:
  662. messages.warning(request, "No {} were selected.".format(self.parent_model._meta.verbose_name_plural))
  663. return redirect(return_url)
  664. table = self.table(selected_objects)
  665. if '_create' in request.POST:
  666. form = self.form(request.POST)
  667. if form.is_valid():
  668. new_components = []
  669. data = deepcopy(form.cleaned_data)
  670. for obj in data['pk']:
  671. names = data['name_pattern']
  672. for name in names:
  673. component_data = {
  674. self.parent_field: obj.pk,
  675. 'name': name,
  676. }
  677. component_data.update(data)
  678. component_form = self.model_form(component_data)
  679. if component_form.is_valid():
  680. new_components.append(component_form.save(commit=False))
  681. else:
  682. for field, errors in component_form.errors.as_data().items():
  683. for e in errors:
  684. form.add_error(field, '{} {}: {}'.format(obj, name, ', '.join(e)))
  685. if not form.errors:
  686. self.model.objects.bulk_create(new_components)
  687. messages.success(request, "Added {} {} to {} {}.".format(
  688. len(new_components),
  689. self.model._meta.verbose_name_plural,
  690. len(form.cleaned_data['pk']),
  691. self.parent_model._meta.verbose_name_plural
  692. ))
  693. return redirect(return_url)
  694. else:
  695. form = self.form(initial={'pk': pk_list})
  696. return render(request, self.template_name, {
  697. 'form': form,
  698. 'component_name': self.model._meta.verbose_name_plural,
  699. 'table': table,
  700. 'return_url': reverse(self.default_return_url),
  701. })