postgres version

server/auth.py

@@ -0,0 +1,34 @@
+from functools import wraps
+from flask import request, Response
+
+import reloader
+
+
+def authenticate():
+    return Response(
+        'Could not verify your access level for that URL. '
+        'You have to login with proper credentials', 401,
+        {'WWW-Authenticate': 'Basic realm="Login Required"'})
+
+
+def auth_required(f):
+    @wraps(f)
+    def decorated(*args, **kwargs):
+        auth = request.cookies.get('password')
+        config = reloader.get_config()
+        if auth is None or auth != config['SERVER_PASSWORD']:
+            return authenticate()
+        return f(*args, **kwargs)
+    return decorated
+
+
+def api_auth_required(f):
+    @wraps(f)
+    def decorated(*args, **kwargs):
+        config = reloader.get_config()
+        if config['ENABLE_API_AUTH']:
+            if request.headers.get('X-Token', '') != config['API_TOKEN']:
+                return Response('Provided token is invalid.', 403)
+        return f(*args, **kwargs)
+    return decorated
+