dnmgmt/whoisdb/views.py

361 lines
11 KiB
Python
Raw Normal View History

2017-03-20 00:16:33 +01:00
from django.shortcuts import render, get_object_or_404
2017-02-28 19:44:15 +01:00
from django.db.models import Q
2017-02-23 22:28:06 +01:00
from django.contrib.auth.decorators import login_required
2017-03-20 00:16:33 +01:00
from django.http import HttpResponseRedirect, Http404
2017-02-28 19:44:15 +01:00
from django.urls import reverse, reverse_lazy
from django.views.generic import DetailView, CreateView, UpdateView
from django.contrib.auth.mixins import LoginRequiredMixin
2017-03-25 04:11:37 +01:00
from django.contrib import messages
from formtools.wizard.views import SessionWizardView
2017-02-23 22:28:06 +01:00
2017-02-28 19:44:15 +01:00
from .models import Maintainer, Contact, InetNum, ASBlock, ASNumber
from .forms import MntForm, MntInitialForm, ContactForm, ContactInitialForm, InetNumForm, ASBlockForm, ASNumberForm
from .generic import DeleteCheckView, MntGenericMixin
2017-03-27 12:18:02 +02:00
from .helpers import findInDatabase
2017-02-24 03:43:59 +01:00
2017-03-22 04:06:24 +01:00
2017-03-20 00:16:33 +01:00
@login_required
def createObjectOverview(request):
2017-03-27 01:09:40 +02:00
mnts = request.user.maintainer_set.filter().all()
2017-03-20 00:16:33 +01:00
netblocks = InetNum.objects.filter(Q(mnt_by__in=mnts) | Q(mnt_lower__in=mnts)).distinct()
asblocks = ASBlock.objects.filter(Q(mnt_by__in=mnts) | Q(mnt_lower__in=mnts)).distinct()
return render(request, "whoisdb/create_overview.html", {"netblocks": netblocks, "asblocks": asblocks})
2017-03-01 03:21:03 +01:00
2017-03-22 04:06:24 +01:00
2017-02-23 22:28:06 +01:00
@login_required
def dbDashboard(request):
2017-03-27 01:09:40 +02:00
mnts = request.user.maintainer_set.filter(rir=False, lir=False).all()
if request.GET.get("delegated", None):
mnts = request.user.maintainer_set.all()
hasDelegations = request.user.maintainer_set.filter(Q(rir=True) | Q(lir=True)).count() > 0
2017-02-28 19:44:15 +01:00
contacts = Contact.objects.filter(mnt_by__in=mnts)
2017-03-01 03:21:03 +01:00
netblocks = InetNum.objects.filter(Q(mnt_by__in=mnts) | Q(mnt_lower__in=mnts)).distinct()
asblocks = ASBlock.objects.filter(Q(mnt_by__in=mnts) | Q(mnt_lower__in=mnts)).distinct()
asnumbers = ASNumber.objects.filter(Q(mnt_by__in=mnts) | Q(mnt_lower__in=mnts)).distinct()
2017-02-23 22:28:06 +01:00
mntForm = contactForm = None
2017-02-24 03:43:59 +01:00
if mnts.count() == 0:
mntForm = contactForm = None
if request.method == "POST":
2017-02-28 19:44:15 +01:00
mntForm = MntInitialForm(user=request.user, data=request.POST, prefix="mnt")
2017-03-22 04:06:24 +01:00
contactForm = ContactInitialForm(user=request.user, data=request.POST, prefix="contact")
2017-02-24 03:43:59 +01:00
if mntForm.is_valid() and contactForm.is_valid():
mnt = mntForm.save(commit=False)
mnt.handleAuto(request.user.username)
mnt.save()
contact = contactForm.save(commit=False)
contact.handleAuto()
2017-02-28 19:44:15 +01:00
contact.type = Contact.TYPE_PERSON
2017-02-24 03:43:59 +01:00
contact.save()
contact.mnt_by.add(mnt.id)
contact.save()
mnt.auth.add(request.user.id)
mnt.admin_c.add(contact.id)
mnt.save()
return HttpResponseRedirect(reverse("whoisdb:dashboard"))
else:
2017-02-28 19:44:15 +01:00
mntForm = MntInitialForm(user=request.user, prefix="mnt", initial={'handle': 'AUTO', 'description': 'Primary maintainer of %s' % request.user.username})
2017-03-22 04:06:24 +01:00
contactForm = ContactInitialForm(user=request.user, initial={'handle': 'AUTO', 'name': request.user.username.capitalize()}, prefix='contact')
2017-02-28 19:44:15 +01:00
2017-03-27 01:09:40 +02:00
return render(request, "whoisdb/overview.html", {"mnts": mnts, "contacts": contacts, "mntForm": mntForm, "contactForm": contactForm, "netblocks": netblocks, "asblocks": asblocks, "asnumbers": asnumbers, 'hasDelegations': hasDelegations})
2017-02-28 19:44:15 +01:00
2017-03-01 03:21:03 +01:00
2017-03-20 00:16:33 +01:00
def showHandle(request, handle):
# a) find handle
models = [Contact, Maintainer, ASBlock, ASNumber, InetNum]
obj = None
for model in models:
if handle.endswith(model.handleSuffix):
obj = get_object_or_404(model, handle=handle)
break
if not obj:
raise Http404("Handle object not found")
return render(request, "whoisdb/handle_show.html", {"object": obj})
2017-03-27 12:18:02 +02:00
def searchObject(request):
2017-03-27 13:29:32 +02:00
results = None
term = None
2017-03-27 12:18:02 +02:00
if request.method == "POST":
2017-03-27 13:29:32 +02:00
term = request.POST.get("q", None)
if term:
results = findInDatabase(term)
return render(request, "whoisdb/search.html", {"results": results, "term": term})
2017-03-27 12:18:02 +02:00
2017-02-28 19:44:15 +01:00
class MaintainerCreate(LoginRequiredMixin, CreateView):
template_name = "whoisdb/obj_create.html"
form_class = MntForm
def get_form_kwargs(self, *args, **kwargs):
kwargs = super(MaintainerCreate, self).get_form_kwargs(*args, **kwargs)
kwargs["user"] = self.request.user
kwargs["initial"] = {
"handle": "AUTO",
}
return kwargs
def form_valid(self, form):
self.object = form.save(commit=False)
self.object.handleAuto(self.request.user.username)
self.object.save()
self.object.auth.add(self.request.user)
self.object.save()
return super(MaintainerCreate, self).form_valid(form)
2017-03-01 03:21:03 +01:00
2017-02-28 19:44:15 +01:00
class MaintainerEdit(LoginRequiredMixin, UpdateView):
2017-03-27 02:40:01 +02:00
template_name = "whoisdb/obj_edit.html"
2017-02-28 19:44:15 +01:00
model = Maintainer
form_class = MntForm
slug_field = "handle"
slug_url_kwarg = "handle"
def get_form_kwargs(self, *args, **kwargs):
kwargs = super(MaintainerEdit, self).get_form_kwargs(*args, **kwargs)
kwargs["user"] = self.request.user
return kwargs
def get_queryset(self):
return self.model.objects.filter(auth=self.request.user)
2017-03-01 03:21:03 +01:00
2017-02-28 19:44:15 +01:00
class MaintainerDelete(LoginRequiredMixin, DeleteCheckView):
template_name = "whoisdb/obj_delete.html"
model = Maintainer
slug_field = "handle"
slug_url_kwarg = "handle"
success_url = reverse_lazy("whoisdb:dashboard")
def get_queryset(self):
return self.model.objects.filter(auth=self.request.user)
2017-03-01 03:21:03 +01:00
2017-02-28 19:44:15 +01:00
class MaintainerDetail(LoginRequiredMixin, DetailView):
2017-03-20 00:16:33 +01:00
template_name = "whoisdb/handle_show.html"
2017-02-28 19:44:15 +01:00
model = Maintainer
slug_field = "handle"
slug_url_kwarg = "handle"
2017-03-20 00:16:33 +01:00
#context_object_name = "mnt"
2017-02-28 19:44:15 +01:00
2017-03-01 03:21:03 +01:00
2017-02-28 19:44:15 +01:00
class ContactDetail(DetailView):
model = Contact
slug_field = "handle"
slug_url_kwarg = "handle"
context_object_name = "contact"
2017-03-01 03:21:03 +01:00
class ContactEdit(MntGenericMixin, LoginRequiredMixin, UpdateView):
2017-02-28 19:44:15 +01:00
template_name = "whoisdb/obj_edit.html"
model = Contact
form_class = ContactForm
slug_field = "handle"
slug_url_kwarg = "handle"
def get_form_kwargs(self, *args, **kwargs):
kwargs = super(ContactEdit, self).get_form_kwargs(*args, **kwargs)
kwargs["user"] = self.request.user
return kwargs
2017-03-01 03:21:03 +01:00
2017-02-28 19:44:15 +01:00
class ContactCreate(LoginRequiredMixin, CreateView):
template_name = "whoisdb/obj_create.html"
form_class = ContactForm
def get_form_kwargs(self, *args, **kwargs):
kwargs = super(ContactCreate, self).get_form_kwargs(*args, **kwargs)
kwargs["user"] = self.request.user
kwargs["initial"] = {
"handle": "AUTO",
"type": Contact.TYPE_PERSON
}
return kwargs
2017-03-01 03:21:03 +01:00
class ContactDelete(MntGenericMixin, LoginRequiredMixin, DeleteCheckView):
2017-02-28 19:44:15 +01:00
template_name = "whoisdb/obj_delete.html"
model = Contact
slug_field = "handle"
slug_url_kwarg = "handle"
success_url = reverse_lazy("whoisdb:dashboard")
2017-03-01 03:21:03 +01:00
2017-02-28 19:44:15 +01:00
# InetNum
class InetNumCreate(LoginRequiredMixin, CreateView):
template_name = "whoisdb/obj_create.html"
form_class = InetNumForm
2017-02-23 22:28:06 +01:00
2017-02-28 19:44:15 +01:00
def get_form_kwargs(self, *args, **kwargs):
2017-03-14 20:18:57 +01:00
print("NOOOOOOOOOOOOOOOOOOOOOOOOOOOOT", args, kwargs)
2017-02-28 19:44:15 +01:00
kwargs = super(InetNumCreate, self).get_form_kwargs(*args, **kwargs)
kwargs["user"] = self.request.user
kwargs["initial"] = {
"handle": "AUTO",
}
2017-03-01 03:21:03 +01:00
2017-02-28 19:44:15 +01:00
return kwargs
2017-02-24 03:43:59 +01:00
2017-03-01 03:21:03 +01:00
class InetNumDetail(DetailView):
model = InetNum
slug_field = "handle"
slug_url_kwarg = "handle"
context_object_name = "inetnum"
class InetNumEdit(MntGenericMixin, LoginRequiredMixin, UpdateView):
2017-03-01 03:21:03 +01:00
template_name = "whoisdb/obj_edit.html"
model = InetNum
form_class = InetNumForm
slug_field = "handle"
slug_url_kwarg = "handle"
def get_form_kwargs(self, *args, **kwargs):
kwargs = super(InetNumEdit, self).get_form_kwargs(*args, **kwargs)
kwargs["user"] = self.request.user
kwargs["initial"] = {'prefix': str(self.object.getNetwork())}
2017-03-01 03:21:03 +01:00
return kwargs
2017-03-01 03:21:03 +01:00
class InetNumDelete(MntGenericMixin, LoginRequiredMixin, DeleteCheckView):
template_name = "whoisdb/obj_delete.html"
model = InetNum
slug_field = "handle"
slug_url_kwarg = "handle"
success_url = reverse_lazy("whoisdb:dashboard")
# asblock
class ASBlockCreate(LoginRequiredMixin, CreateView):
template_name = "whoisdb/obj_create.html"
form_class = ASBlockForm
def get_form_kwargs(self, *args, **kwargs):
kwargs = super(ASBlockCreate, self).get_form_kwargs(*args, **kwargs)
kwargs["user"] = self.request.user
kwargs["initial"] = {
"handle": "AUTO",
}
2017-03-01 03:21:03 +01:00
return kwargs
class ASBlockDetail(DetailView):
model = ASBlock
slug_field = "handle"
slug_url_kwarg = "handle"
context_object_name = "asblock"
2017-03-22 04:06:24 +01:00
class ASBlockEdit(MntGenericMixin, LoginRequiredMixin, UpdateView):
template_name = "whoisdb/obj_edit.html"
model = ASBlock
form_class = ASBlockForm
slug_field = "handle"
slug_url_kwarg = "handle"
def get_form_kwargs(self, *args, **kwargs):
kwargs = super(ASBlockEdit, self).get_form_kwargs(*args, **kwargs)
kwargs["user"] = self.request.user
#kwargs["initial"] = {'prefix': str(self.object.getNetwork())}
return kwargs
2017-03-01 03:21:03 +01:00
class ASBlockDelete(MntGenericMixin, LoginRequiredMixin, DeleteCheckView):
2017-03-01 03:21:03 +01:00
template_name = "whoisdb/obj_delete.html"
model = ASBlock
slug_field = "handle"
slug_url_kwarg = "handle"
success_url = reverse_lazy("whoisdb:dashboard")
2017-03-22 04:06:24 +01:00
# asnumber
class ASNumberCreate(LoginRequiredMixin, CreateView):
template_name = "whoisdb/obj_create.html"
form_class = ASNumberForm
def get_form_kwargs(self, *args, **kwargs):
kwargs = super(ASNumberCreate, self).get_form_kwargs(*args, **kwargs)
kwargs["user"] = self.request.user
kwargs["initial"] = {
"handle": "AUTO",
}
return kwargs
class ASNumberDetail(DetailView):
model = ASNumber
slug_field = "handle"
slug_url_kwarg = "handle"
context_object_name = "asnumber"
class ASNumberEdit(MntGenericMixin, LoginRequiredMixin, UpdateView):
template_name = "whoisdb/obj_edit.html"
model = ASNumber
form_class = ASNumberForm
slug_field = "handle"
slug_url_kwarg = "handle"
def get_form_kwargs(self, *args, **kwargs):
kwargs = super(ASNumberEdit, self).get_form_kwargs(*args, **kwargs)
kwargs["user"] = self.request.user
return kwargs
class ASNumberDelete(MntGenericMixin, LoginRequiredMixin, DeleteCheckView):
template_name = "whoisdb/obj_delete.html"
model = ASNumber
2017-03-01 03:21:03 +01:00
slug_field = "handle"
slug_url_kwarg = "handle"
success_url = reverse_lazy("whoisdb:dashboard")
2017-03-25 04:11:37 +01:00
class ASAndSubnetWizard(LoginRequiredMixin, SessionWizardView):
form_list = [ASNumberForm, InetNumForm]
template_name = "whoisdb/wizard.html"
def get_form_initial(self, step):
return {"handle": "AUTO"}
def done(self, form_list, **kwargs):
fl = list(form_list)
asNum = fl[0].save()
net = fl[1].save()
messages.info(self.request, "The following objects have been created: AS %s %s, Subnet %s %s" % (asNum.handle, asNum.number, net.handle, net.getNetwork()))
return HttpResponseRedirect(reverse("whoisdb:dashboard"))
def get_form_kwargs(self, *args, **kwargs):
kwargs = super(ASAndSubnetWizard, self).get_form_kwargs(*args, **kwargs)
kwargs["user"] = self.request.user
return kwargs
def get_context_data(self, *args, **kwargs):
d = super(ASAndSubnetWizard, self).get_context_data(*args, **kwargs)
step = d["wizard"]["steps"].step1
if step == 1:
d["message"] = "Create an AS object"
elif step == 2:
d["message"] = "Create a Subnet"
return d