dnmgmt/domains/views.py

from django.shortcuts import render
from django.urls import reverse_lazy
from django.contrib.auth.decorators import login_required
from django.views.generic import DetailView, CreateView, UpdateView
from django.contrib.auth.mixins import LoginRequiredMixin

from whoisdb.generic import MntGenericMixin, DeleteCheckView

from .models import Domain, Nameserver
from .forms import DomainForm, NameserverForm


@login_required
def overview(request):
	mnts = request.user.maintainer_set.all()

	# get all domains and nameservers
	domains = Domain.objects.filter(mnt_by__in=mnts)
	nameservers = Nameserver.objects.filter(mnt_by__in=mnts)

	return render(request, "domains/overview.html", {"domains": domains, "nameservers": nameservers})


class DomainCreate(LoginRequiredMixin, CreateView):
	template_name = "domains/obj_create.html"
	form_class = DomainForm

	def get_form_kwargs(self, *args, **kwargs):
		kwargs = super(DomainCreate, self).get_form_kwargs(*args, **kwargs)
		kwargs["user"] = self.request.user

		return kwargs


class DomainDetail(LoginRequiredMixin, DetailView):
	model = Domain
	slug_field = "name"
	slug_url_kwarg = "domain"
	context_object_name = "domain"


class DomainEdit(MntGenericMixin, LoginRequiredMixin, UpdateView):
	model = Domain
	form_class = DomainForm
	slug_field = "name"
	slug_url_kwarg = "domain"
	template_name = "domains/obj_edit.html"

	def get_form_kwargs(self, *args, **kwargs):
		kwargs = super(DomainEdit, self).get_form_kwargs(*args, **kwargs)
		kwargs["user"] = self.request.user
		return kwargs


class DomainDelete(MntGenericMixin, LoginRequiredMixin, DeleteCheckView):
	template_name = "domains/obj_delete.html"
	model = Domain
	slug_field = "name"
	slug_url_kwarg = "domain"
	success_url = reverse_lazy("domains:overview")


class NameserverCreate(LoginRequiredMixin, CreateView):
	template_name = "domains/obj_create.html"
	form_class = NameserverForm

	def get_form_kwargs(self, *args, **kwargs):
		kwargs = super(NameserverCreate, self).get_form_kwargs(*args, **kwargs)
		kwargs["user"] = self.request.user

		return kwargs


class NameserverDetail(LoginRequiredMixin, DetailView):
	model = Nameserver
	slug_field = "name"
	slug_url_kwarg = "domain"
	context_object_name = "nameserver"


class NameserverEdit(MntGenericMixin, LoginRequiredMixin, UpdateView):
	model = Nameserver
	form_class = NameserverForm
	slug_field = "name"
	slug_url_kwarg = "domain"
	template_name = "domains/obj_edit.html"

	def get_form_kwargs(self, *args, **kwargs):
		kwargs = super(NameserverEdit, self).get_form_kwargs(*args, **kwargs)
		kwargs["user"] = self.request.user
		return kwargs


class NameserverDelete(MntGenericMixin, LoginRequiredMixin, DeleteCheckView):
	template_name = "domains/obj_delete.html"
	model = Nameserver
	slug_field = "name"
	slug_url_kwarg = "domain"
	success_url = reverse_lazy("domains:overview")