mirror of
https://github.com/yogeshojha/rengine.git
synced 2026-09-26 19:54:53 +02:00
37 lines
1.4 KiB
Python
37 lines
1.4 KiB
Python
from django.shortcuts import render, get_object_or_404
|
|
from django import http
|
|
from .models import Domain
|
|
from django.contrib import messages
|
|
from .forms import AddDomainForm, RawDomainForm
|
|
from django.utils import timezone
|
|
from django.urls import reverse
|
|
|
|
def index(request):
|
|
# TODO bring default target page
|
|
return render(request, 'target/index.html')
|
|
|
|
def add_target_form(request):
|
|
form = RawDomainForm()
|
|
if request.method == "POST":
|
|
form = RawDomainForm(request.POST)
|
|
if form.is_valid():
|
|
Domain.objects.create(**form.cleaned_data, insert_date=timezone.now())
|
|
messages.add_message(request, messages.INFO, 'Target domain ' + form.cleaned_data['domain_name'] + ' added successfully')
|
|
return http.HttpResponseRedirect(reverse('add_target_form'))
|
|
context = {"add_target_li": "active", "target_data_active": "true", 'form': form}
|
|
return render(request, 'target/add.html', context)
|
|
|
|
def list_target(request):
|
|
domains = Domain.objects
|
|
context = {'list_target_li': 'active', 'target_data_active': 'true', 'domains': domains}
|
|
return render(request, 'target/list.html', context)
|
|
|
|
def delete_domain(request, id):
|
|
obj = get_object_or_404(Domain, id=id)
|
|
if request.method == "POST":
|
|
obj.delete()
|
|
responseData = {'status': 'true'}
|
|
else:
|
|
responseData = {'status': 'false'}
|
|
return http.JsonResponse(responseData)
|