Better error gestion with correct status code

1 files changed, 10 insertions, 2 deletions

diff --git a/pasteme.py b/pasteme.py
index 744b07c..f155502 100755
--- a/pasteme.py
+++ b/pasteme.py
@@ -25,7 +25,7 @@ def route_paste_post():
         pid = identigen.generate(content)
     except AttributeError as e:
         print(e)
-        return bottle.template('internal_error')
+        bottle.abort(500)
     path = pathbase / pid
     with path.open(mode='wb') as fd:
             fd.write(content.encode('utf8'))
@@ -46,7 +46,7 @@ def route_paste_get(pid, pformat='colored'):
             content = fd.read()
     except IOError:
         # use this template for all file based exception
-        return bottle.template('not_found')
+        bottle.abort(404)
     if pformat == 'colored':
         try:
             lexer = pygments.lexers.guess_lexer(content)
@@ -56,6 +56,14 @@ def route_paste_get(pid, pformat='colored'):
         return bottle.template('paste', content=content)
     return content
 
+@bottle.error(404)
+def error404(error):
+    return bottle.template("not_found")
+
+@bottle.error(500)
+def error500(error):
+    return bottle.template("internal_error")
+
 if __name__ == '__main__':
     print('I: Starting application with development server')
     bottle.run(host='0.0.0.0', port=8080, debug=True, reloader=True)