mattstibbs · March 14, 2017 00:12 · Mar 14, 2017
diff --git a/auth.py b/auth.py
@@ -0,0 +1,29 @@
+import config as config
+from flask import Response, request
+from functools import wraps
+
+
+def check_auth(username, password):
+    """This function is called to check if a username /
+    password combination is valid.
+    """
+    result = (username == config.API_USER and password == config.API_PASS)
+    return result
+
+
+def authenticate():
+    """Sends a 401 response that enables basic 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 requires_auth(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