Added basic authentication support for tinderbox reports

1 files changed, 33 insertions, 0 deletions

diff --git a/grumpy/utils.py b/grumpy/utils.py
new file mode 100644
index 0000000..a70fd95
--- /dev/null
+++ b/grumpy/utils.py
@@ -0,0 +1,33 @@
+# -*- coding: utf-8 -*-
+"""
+    grumpy.utils
+    ~~~~~~~~~~~~
+
+    Miscellaneous utils for authentication and pagination.
+
+    :copyright: (c) 2010 Priit Laes
+"""
+from flask import request, Response
+from functools import wraps
+
+from . import app
+
+def authenticate():
+    """Sends a 401 response that enables basic HTTP auth"""
+    return Response('Could not verify your access level for that URL.\n'
+                    'You have to login with proper credentials', 401,
+                    {'WWW-Authenticate': 'Basic realm="Login required"'})
+
+def check_auth(username, password):
+    """Checks username password against ones stored in configuration."""
+    return username == app.config['TINDERBOX_USER'] and \
+        password == app.config['TINDERBOX_PASS']
+
+def requires_auth_basic(f):
+    @wraps(f)
+    def decorated(*args, **kwargs):
+        auth = request.authorization
+        if not auth or not check_auth(auth.username, auth.password):
+            return authenticate()
+        return f(*args, **kwargs)
+    return decorated