def upload_threatmodel(request, eid): eng = Engagement.objects.get(id=eid) add_breadcrumb( parent=eng, title="Upload a threat model", top_level=False, request=request) if request.method == 'POST': form = UploadThreatForm(request.POST, request.FILES) if form.is_valid(): handle_uploaded_threat(request.FILES['file'], eng) eng.progress = 'other' eng.threat_model = True eng.save() messages.add_message( request, messages.SUCCESS, 'Threat model saved.', extra_tags='alert-success') return HttpResponseRedirect( reverse('view_engagement', args=(eid, ))) else: form = UploadThreatForm() product_tab = Product_Tab(eng.product.id, title="Upload Threat Model", tab="engagements") return render(request, 'dojo/up_threat.html', { 'form': form, 'product_tab': product_tab, 'eng': eng, })