dojo/development_environment/views.py - master - Art86/DefectDojo-DevSecOps

dojo/development_environment/views.py
98 строк4 KB
kiblik
Ruff: Add and fix Q000 (#10095)
13 июл 2024, 01:18
13 июл 2024, 01:18375bb4b
# #dev envs
import logging

from django.contrib import messages
from django.contrib.auth.decorators import login_required
from django.db.models.deletion import RestrictedError
from django.http import HttpResponseRedirect
from django.shortcuts import get_object_or_404, render
from django.urls import reverse

from dojo.authorization.authorization import user_has_configuration_permission_or_403
from dojo.authorization.authorization_decorators import user_is_configuration_authorized
from dojo.filters import DevelopmentEnvironmentFilter
from dojo.forms import Delete_Dev_EnvironmentForm, Development_EnvironmentForm
from dojo.models import Development_Environment
from dojo.utils import add_breadcrumb, get_page_items

logger = logging.getLogger(__name__)


@login_required
def dev_env(request):
    initial_queryset = Development_Environment.objects.all().order_by("name")
    name_words = [de.name for de in
                  initial_queryset]
    devs = DevelopmentEnvironmentFilter(request.GET, queryset=initial_queryset)
    dev_page = get_page_items(request, devs.qs, 25)
    add_breadcrumb(title="Environment List", top_level=True, request=request)
    return render(request, "dojo/dev_env.html", {
        "name": "Environment",
        "metric": False,
        "user": request.user,
        "devs": dev_page,
        "dts": devs,
        "name_words": name_words})


@user_is_configuration_authorized("dojo.add_development_environment")
def add_dev_env(request):
    form = Development_EnvironmentForm()
    if request.method == "POST":
        form = Development_EnvironmentForm(request.POST)
        if form.is_valid():
            form.save()
            messages.add_message(request,
                                 messages.SUCCESS,
                                 "Environment added successfully.",
                                 extra_tags="alert-success")
            return HttpResponseRedirect(reverse("dev_env"))
    add_breadcrumb(title="Add Environment", top_level=False, request=request)
    return render(request, "dojo/new_dev_env.html", {
        "name": "Add Environment",
        "metric": False,
        "user": request.user,
        "form": form,
    })


@user_is_configuration_authorized("dojo.change_development_environment")
def edit_dev_env(request, deid):
    de = get_object_or_404(Development_Environment, pk=deid)
    form1 = Development_EnvironmentForm(instance=de)
    form2 = Delete_Dev_EnvironmentForm(instance=de)
    if request.method == "POST" and request.POST.get("edit_dev_env"):
        form1 = Development_EnvironmentForm(request.POST, instance=de)
        if form1.is_valid():
            de = form1.save()
            messages.add_message(
                request,
                messages.SUCCESS,
                "Environment updated successfully.",
                extra_tags="alert-success")
            return HttpResponseRedirect(reverse("dev_env"))
    if request.method == "POST" and request.POST.get("delete_dev_env"):
        user_has_configuration_permission_or_403(request.user, "dojo.delete_development_environment")
        form2 = Delete_Dev_EnvironmentForm(request.POST, instance=de)
        if form2.is_valid():
            try:
                de.delete()
                messages.add_message(
                    request,
                    messages.SUCCESS,
                    "Environment deleted successfully.",
                    extra_tags="alert-success")
            except RestrictedError as err:
                messages.add_message(request,
                                        messages.WARNING,
                                        f"Environment cannot be deleted: {err}",
                                        extra_tags="alert-warning")
            return HttpResponseRedirect(reverse("dev_env"))

    add_breadcrumb(title="Edit Environment", top_level=False, request=request)
    return render(request, "dojo/edit_dev_env.html", {
        "name": "Edit Environment",
        "metric": False,
        "user": request.user,
        "form1": form1,
        "de": de})