dnmgmt/whoisdb/generic.py

from django.views.generic import DeleteView
from django.http import HttpResponseRedirect
from django.contrib import messages
from django.db.models import Q

class DeleteCheckView(DeleteView):
	""" Check if object actually can be deleted. Provide reasons to template
	    if not.
	"""
	def delete(self, request, *args, **kwargs):
		self.object = self.get_object()
		success_url = self.get_success_url()

		reasons = self.object.getNoDeleteReasons()
		if reasons:
			# do not delete, do what get does...
			return self.get(request, *args, **kwargs)
		else:
			self.object.delete()
			messages.info(request, "Object %s has been deleted" % self.object.handle)
			return HttpResponseRedirect(success_url)

	def get_context_data(self, **kwargs):
		if 'reasons' not in kwargs:
			kwargs['reasons'] = self.object.getNoDeleteReasons()
		return super(DeleteCheckView, self).get_context_data(**kwargs)


class MntGenericMixin(object):
	def get_queryset(self):
		mnts = self.request.user.maintainer_set.all()

		q = Q(mnt_by__in=mnts)
		if hasattr(self.model, "mnt_lower"):
			q |= Q(mnt_lower__in=mnts)

		return self.model.objects.filter(q).distinct()

	def get_form_kwargs(self, *args, **kwargs):
		kwargs = super(MntGenericMixin, self).get_form_kwargs(*args, **kwargs)

		if hasattr(self.model, "mnt_lower"):
			mnts = self.request.user.maintainer_set.all()
			if not any(mnt in self.object.mnt_by.all() for mnt in mnts):
				# we are in mnt_lower
				kwargs["lower"] = True

		return kwargs