From 24f197935e9b4b11ce57bf3cb8b7ddfd0a3db8ba Mon Sep 17 00:00:00 2001 From: gi0baro Date: Mon, 15 Dec 2014 23:09:35 +0100 Subject: [PATCH] Handling new pydal Exceptions in response.download --- gluon/globals.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/gluon/globals.py b/gluon/globals.py index f872864a..419578f7 100644 --- a/gluon/globals.py +++ b/gluon/globals.py @@ -597,6 +597,7 @@ class Response(Storage): Downloads from http://..../download/filename """ + from pydal.exceptions import NotAuthorizedException, NotFoundException current.session.forget(current.response) @@ -613,6 +614,10 @@ class Response(Storage): raise HTTP(404) try: (filename, stream) = field.retrieve(name, nameonly=True) + except NotAuthorizedException: + raise HTTP(403) + except NotFoundException: + raise HTTP(404) except IOError: raise HTTP(404) headers = self.headers