Esempio n. 1
0
def mktemplate(request, fid):
    finding = get_object_or_404(Finding, id=fid)
    templates = Finding_Template.objects.filter(title=finding.title)
    if len(templates) > 0:
        messages.add_message(
            request,
            messages.ERROR,
            'A finding template with that title already exists.',
            extra_tags='alert-danger')
    else:
        template = Finding_Template(
            title=finding.title,
            cwe=finding.cwe,
            severity=finding.severity,
            description=finding.description,
            mitigation=finding.mitigation,
            impact=finding.impact,
            references=finding.references,
            numerical_severity=finding.numerical_severity)
        template.save()
        template.tags = finding.tags
        messages.add_message(
            request,
            messages.SUCCESS,
            mark_safe(
                'Finding template added successfully. You may edit it <a href="%s">here</a>.'
                % reverse('edit_template', args=(template.id, ))),
            extra_tags='alert-success')
    return HttpResponseRedirect(reverse('view_finding', args=(finding.id, )))
Esempio n. 2
0
def mktemplate(request, fid):
    finding = get_object_or_404(Finding, id=fid)
    templates = Finding_Template.objects.filter(title=finding.title)
    if len(templates) > 0:
        messages.add_message(request,
                             messages.ERROR,
                             'A finding template with that title already exists.',
                             extra_tags='alert-danger')
    else:
        template = Finding_Template(title=finding.title,
                                    cwe=finding.cwe,
                                    severity=finding.severity,
                                    description=finding.description,
                                    mitigation=finding.mitigation,
                                    impact=finding.impact,
                                    references=finding.references,
                                    numerical_severity=finding.numerical_severity)
        template.save()
        template.tags = finding.tags
        messages.add_message(request,
                             messages.SUCCESS,
                             mark_safe('Finding template added successfully. You may edit it <a href="%s">here</a>.' %
                                       reverse('edit_template',
                                               args=(template.id,))),
                             extra_tags='alert-success')
    return HttpResponseRedirect(reverse('view_finding', args=(finding.id,)))