Add support for email confirmation

1 files changed, 40 insertions, 0 deletions

diff --git a/app/decorators.py b/app/decorators.py
new file mode 100644
index 0000000..d5b8821
--- /dev/null
+++ b/app/decorators.py
@@ -0,0 +1,40 @@
+# -*- coding: utf-8 -*-
+
+from functools import wraps
+
+from flask import current_app, request, redirect, flash, url_for
+from flask_login import current_user
+from flask_login.config import EXEMPT_METHODS
+
+
+def admin_required(func):
+    @wraps(func)
+    def decorated_view(*args, **kwargs):
+        if request.method in EXEMPT_METHODS:
+            return func(*args, **kwargs)
+        elif current_app.config.get("LOGIN_DISABLED"):
+            return func(*args, **kwargs)
+        elif not current_user.is_authenticated:
+            return current_app.login_manager.unauthorized()
+        elif not current_user.is_admin:
+            return current_app.login_manager.unauthorized()
+        return func(*args, **kwargs)
+
+    return decorated_view
+
+
+def login_required(func):
+    @wraps(func)
+    def decorated_view(*args, **kwargs):
+        if request.method in EXEMPT_METHODS:
+            return func(*args, **kwargs)
+        elif current_app.config.get("LOGIN_DISABLED"):
+            return func(*args, **kwargs)
+        elif not current_user.is_authenticated:
+            return current_app.login_manager.unauthorized()
+        elif not current_user.is_confirmed:
+            flash("hello world")
+            return redirect(url_for("auth.not_confirmed"))
+        return func(*args, **kwargs)
+
+    return decorated_view