From 57b9f09127e84051109cdcf3557ad6d069f5121f Mon Sep 17 00:00:00 2001 From: Zambom Date: Tue, 24 Jan 2017 20:24:57 -0200 Subject: [PATCH] Adding file path inexist handler --- file_link/views.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/file_link/views.py b/file_link/views.py index aab43a7..f15e2de 100644 --- a/file_link/views.py +++ b/file_link/views.py @@ -1,5 +1,5 @@ from django.shortcuts import get_object_or_404, redirect, render -from django.http import HttpResponse +from django.http import HttpResponse, Http404 from os import path from django.views import generic from django.contrib import messages @@ -35,6 +35,9 @@ class DownloadFile(LoginRequiredMixin, generic.DetailView): slug = self.kwargs.get('slug', '') file_link = get_object_or_404(FileLink, slug = slug) + if not path.exists(file_link.file_content.path): + raise Http404() + response = HttpResponse(open(file_link.file_content.path, 'rb').read()) response['Content-Type'] = 'application/force-download' response['Pragma'] = 'public' -- libgit2 0.21.2