mirror of
https://github.com/yogeshojha/rengine.git
synced 2026-09-17 23:37:47 +02:00
38 lines
1.4 KiB
Python
38 lines
1.4 KiB
Python
from django.shortcuts import render, get_object_or_404
|
|
from scanEngine.models import EngineType
|
|
from scanEngine.forms import AddEngineForm
|
|
from django.contrib import messages
|
|
from django import http
|
|
from django.urls import reverse
|
|
|
|
def index(request):
|
|
engine_type = EngineType.objects.all().order_by('id')
|
|
context = {'scan_engine_nav_active': 'true',
|
|
'engine_type': engine_type,
|
|
}
|
|
return render(request, 'scanEngine/index.html', context)
|
|
|
|
def add_engine(request):
|
|
form = AddEngineForm()
|
|
if request.method == "POST":
|
|
form = AddEngineForm(request.POST)
|
|
if form.is_valid():
|
|
form.save()
|
|
messages.add_message(request, messages.INFO, 'Scan Engine Added successfully')
|
|
return http.HttpResponseRedirect(reverse('scan_engine_index'))
|
|
context = {'scan_engine_nav_active': 'true',
|
|
'form': form}
|
|
return render(request, 'scanEngine/add_engine.html', context)
|
|
|
|
|
|
def delete_engine(request, id):
|
|
obj = get_object_or_404(EngineType, id=id)
|
|
if request.method == "POST":
|
|
obj.delete()
|
|
responseData = {'status': 'true'}
|
|
messages.add_message(request, messages.INFO, 'Engine successfully deleted!')
|
|
else:
|
|
responseData = {'status': 'false'}
|
|
messages.add_message(request, messages.INFO, 'Oops! Engine could not be deleted!')
|
|
return http.JsonResponse(responseData)
|