From ffda3fac7e80f6a86cf89b3d124eef41793da950 Mon Sep 17 00:00:00 2001 From: pictuga Date: Fri, 28 Aug 2015 19:15:40 +0200 Subject: [PATCH] Improve file detection in web server --- morss/morss.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/morss/morss.py b/morss/morss.py index cbe0b6c..170c1a3 100644 --- a/morss/morss.py +++ b/morss/morss.py @@ -621,12 +621,15 @@ def cgi_wrapper(environ, start_response): if url == '': url = 'index.html' - if os.path.isfile(url): + try: + body = open(path, 'rb').read() + headers['status'] = '200 OK' headers['content-type'] = files[url] start_response(headers['status'], list(headers.items())) - return open(url, 'rb').read() - else: + return body + + except IOError: headers['status'] = '404 Not found' start_response(headers['status'], list(headers.items())) return ''