361 lines
11 KiB
Python
361 lines
11 KiB
Python
from django.shortcuts import render, get_object_or_404
|
|
from django.db.models import Q
|
|
from django.contrib.auth.decorators import login_required
|
|
from django.http import HttpResponseRedirect, Http404
|
|
from django.urls import reverse, reverse_lazy
|
|
from django.views.generic import DetailView, CreateView, UpdateView
|
|
from django.contrib.auth.mixins import LoginRequiredMixin
|
|
from django.contrib import messages
|
|
from formtools.wizard.views import SessionWizardView
|
|
|
|
from .models import Maintainer, Contact, InetNum, ASBlock, ASNumber
|
|
from .forms import MntForm, MntInitialForm, ContactForm, ContactInitialForm, InetNumForm, ASBlockForm, ASNumberForm
|
|
from .generic import DeleteCheckView, MntGenericMixin
|
|
from .helpers import findInDatabase
|
|
|
|
|
|
@login_required
|
|
def createObjectOverview(request):
|
|
mnts = request.user.maintainer_set.filter().all()
|
|
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})
|
|
|
|
|
|
@login_required
|
|
def dbDashboard(request):
|
|
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
|
|
|
|
contacts = Contact.objects.filter(mnt_by__in=mnts)
|
|
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()
|
|
mntForm = contactForm = None
|
|
|
|
if mnts.count() == 0:
|
|
mntForm = contactForm = None
|
|
if request.method == "POST":
|
|
mntForm = MntInitialForm(user=request.user, data=request.POST, prefix="mnt")
|
|
contactForm = ContactInitialForm(user=request.user, data=request.POST, prefix="contact")
|
|
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()
|
|
contact.type = Contact.TYPE_PERSON
|
|
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:
|
|
mntForm = MntInitialForm(user=request.user, prefix="mnt", initial={'handle': 'AUTO', 'description': 'Primary maintainer of %s' % request.user.username})
|
|
contactForm = ContactInitialForm(user=request.user, initial={'handle': 'AUTO', 'name': request.user.username.capitalize()}, prefix='contact')
|
|
|
|
return render(request, "whoisdb/overview.html", {"mnts": mnts, "contacts": contacts, "mntForm": mntForm, "contactForm": contactForm, "netblocks": netblocks, "asblocks": asblocks, "asnumbers": asnumbers, 'hasDelegations': hasDelegations})
|
|
|
|
|
|
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})
|
|
|
|
|
|
def searchObject(request):
|
|
results = None
|
|
term = None
|
|
if request.method == "POST":
|
|
term = request.POST.get("q", None)
|
|
if term:
|
|
results = findInDatabase(term)
|
|
return render(request, "whoisdb/search.html", {"results": results, "term": term})
|
|
|
|
|
|
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)
|
|
|
|
|
|
class MaintainerEdit(LoginRequiredMixin, UpdateView):
|
|
template_name = "whoisdb/obj_edit.html"
|
|
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)
|
|
|
|
|
|
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)
|
|
|
|
|
|
class MaintainerDetail(LoginRequiredMixin, DetailView):
|
|
template_name = "whoisdb/handle_show.html"
|
|
model = Maintainer
|
|
slug_field = "handle"
|
|
slug_url_kwarg = "handle"
|
|
#context_object_name = "mnt"
|
|
|
|
|
|
class ContactDetail(DetailView):
|
|
model = Contact
|
|
slug_field = "handle"
|
|
slug_url_kwarg = "handle"
|
|
context_object_name = "contact"
|
|
|
|
|
|
class ContactEdit(MntGenericMixin, LoginRequiredMixin, UpdateView):
|
|
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
|
|
|
|
|
|
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
|
|
|
|
|
|
class ContactDelete(MntGenericMixin, LoginRequiredMixin, DeleteCheckView):
|
|
template_name = "whoisdb/obj_delete.html"
|
|
model = Contact
|
|
slug_field = "handle"
|
|
slug_url_kwarg = "handle"
|
|
success_url = reverse_lazy("whoisdb:dashboard")
|
|
|
|
|
|
# InetNum
|
|
class InetNumCreate(LoginRequiredMixin, CreateView):
|
|
template_name = "whoisdb/obj_create.html"
|
|
form_class = InetNumForm
|
|
|
|
def get_form_kwargs(self, *args, **kwargs):
|
|
print("NOOOOOOOOOOOOOOOOOOOOOOOOOOOOT", args, kwargs)
|
|
kwargs = super(InetNumCreate, self).get_form_kwargs(*args, **kwargs)
|
|
kwargs["user"] = self.request.user
|
|
kwargs["initial"] = {
|
|
"handle": "AUTO",
|
|
}
|
|
|
|
return kwargs
|
|
|
|
|
|
class InetNumDetail(DetailView):
|
|
model = InetNum
|
|
slug_field = "handle"
|
|
slug_url_kwarg = "handle"
|
|
context_object_name = "inetnum"
|
|
|
|
|
|
class InetNumEdit(MntGenericMixin, LoginRequiredMixin, UpdateView):
|
|
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())}
|
|
|
|
return kwargs
|
|
|
|
|
|
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",
|
|
}
|
|
|
|
return kwargs
|
|
|
|
|
|
class ASBlockDetail(DetailView):
|
|
model = ASBlock
|
|
slug_field = "handle"
|
|
slug_url_kwarg = "handle"
|
|
context_object_name = "asblock"
|
|
|
|
|
|
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
|
|
|
|
|
|
class ASBlockDelete(MntGenericMixin, LoginRequiredMixin, DeleteCheckView):
|
|
template_name = "whoisdb/obj_delete.html"
|
|
model = ASBlock
|
|
slug_field = "handle"
|
|
slug_url_kwarg = "handle"
|
|
success_url = reverse_lazy("whoisdb:dashboard")
|
|
|
|
|
|
# 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
|
|
slug_field = "handle"
|
|
slug_url_kwarg = "handle"
|
|
success_url = reverse_lazy("whoisdb:dashboard")
|
|
|
|
|
|
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
|