require OpenID for register and authorize

examples/demoprovider/provider.py

@@ -3,6 +3,7 @@
 from sqlalchemy.orm.exc import NoResultFound
 from models import ResourceOwner, Client, Nonce, Callback
 from models import RequestToken, AccessToken
+from utils import require_openid
 
 
 class ExampleProvider(OAuthProvider):
@@ -15,6 +16,7 @@ def enforce_ssl(self):
     def realms(self):
         return [u"secret", u"trolling"]
 
+    @require_openid
     def authorize(self):
         if request.method == u"POST":
             token = request.form.get("oauth_token")
@@ -24,6 +26,7 @@ def authorize(self):
             token = request.args.get(u"oauth_token")
             return render_template(u"authorize.html", token=token)
 
+    @require_openid
     def register(self):
         if request.method == u'POST':
             client_key = self.generate_client_key()

examples/demoprovider/utils.py

@@ -0,0 +1,14 @@
+from functools import wraps
+from flask import g, url_for, request, redirect
+
+
+def require_openid(f):
+    """Require user to be logged in."""
+    @wraps(f)
+    def decorator(*args, **kwargs):
+        if g.user is None:
+            next_url = url_for("login") + "?next=" + request.url
+            return redirect(next_url)
+        else:
+            return f(*args, **kwargs)
+    return decorator