Commit 383d23bb authored by Okhin's avatar Okhin

Let's return an error when attachment not found

parent 8ac35254
......@@ -284,9 +284,12 @@ class ViolationSearchView(SearchView):
def get_attach(request,id):
f = get_object_or_404(Attachment, pk=id)
wrapper = FileWrapper(f.storage)
response=HttpResponse(wrapper, content_type=f.type)
response['Content-Disposition'] = 'attachment; filename="%s"' % f.name
response['Content-Length'] = f.storage.size
try:
response=HttpResponse(wrapper, content_type=f.type)
response['Content-Disposition'] = 'attachment; filename="%s"' % f.name
response['Content-Length'] = f.storage.size
except:
response=HttpResponseNotFound('<h1>Document not found.</h1>')
return response
class ViolationCsvList(ListView):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment