Branch:dec - adding return statement to all the decorator func

flask_oauthlib/provider/oauth1.py

@@ -151,6 +151,7 @@ def limit_client_request():
                 track_request(client)
         """
         self._before_request_funcs.append(f)
+        return f
 
     def after_request(self, f):
         """Register functions to be invoked after accessing the resource.
@@ -165,6 +166,7 @@ def valid_after_request(valid, oauth):
                 return valid, oauth
         """
         self._after_request_funcs.append(f)
+        return f
 
     def clientgetter(self, f):
         """Register a function as the client getter.
@@ -190,6 +192,7 @@ def get_client(client_key):
                 return client
         """
         self._clientgetter = f
+        return f
 
     def tokengetter(self, f):
         """Register a function as the access token getter.
@@ -210,6 +213,7 @@ def get_access_token(client_key, token):
                 return AccessToken.get(client_key=client_key, token=token)
         """
         self._tokengetter = f
+        return f
 
     def tokensetter(self, f):
         """Register a function as the access token setter.
@@ -243,6 +247,7 @@ def save_access_token(token, request):
             - request_token: Requst token for exchanging this access token
         """
         self._tokensetter = f
+        return f
 
     def grantgetter(self, f):
         """Register a function as the request token getter.
@@ -263,6 +268,7 @@ def get_request_token(token):
                 return RequestToken.get(token=token)
         """
         self._grantgetter = f
+        return f
 
     def grantsetter(self, f):
         """Register a function as the request token setter.
@@ -281,6 +287,7 @@ def save_request_token(token, request):
                 return data.save()
         """
         self._grantsetter = f
+        return f
 
     def noncegetter(self, f):
         """Register a function as the nonce and timestamp getter.
@@ -301,6 +308,7 @@ def get_nonce(client_key, timestamp, nonce, request_token,
                 return Nonce.get("...")
         """
         self._noncegetter = f
+        return f
 
     def noncesetter(self, f):
         """Register a function as the nonce and timestamp setter.
@@ -317,6 +325,7 @@ def save_nonce(client_key, timestamp, nonce, request_token,
         if you put timestamp and nonce object in a cache.
         """
         self._noncesetter = f
+        return f
 
     def verifiergetter(self, f):
         """Register a function as the verifier getter.
@@ -335,6 +344,7 @@ def load_verifier(verifier, token):
                 return data
         """
         self._verifiergetter = f
+        return f
 
     def verifiersetter(self, f):
         """Register a function as the verifier setter.
@@ -356,6 +366,7 @@ def save_verifier(verifier, token, *args, **kwargs):
                 return data.save()
         """
         self._verifiersetter = f
+        return f
 
     def authorize_handler(self, f):
         """Authorization handler decorator.

flask_oauthlib/provider/oauth2.py

@@ -182,6 +182,7 @@ def limit_client_request():
                 track_request(client)
         """
         self._before_request_funcs.append(f)
+        return f
 
     def after_request(self, f):
         """Register functions to be invoked after accessing the resource.
@@ -196,6 +197,7 @@ def valid_after_request(valid, oauth):
                 return valid, oauth
         """
         self._after_request_funcs.append(f)
+        return f
 
     def clientgetter(self, f):
         """Register a function as the client getter.
@@ -225,6 +227,7 @@ def get_client(client_id):
                 return client
         """
         self._clientgetter = f
+        return f
 
     def usergetter(self, f):
         """Register a function as the user getter.
@@ -238,6 +241,7 @@ def get_user(username=username, password=password,
                 return get_user_by_username(username, password)
         """
         self._usergetter = f
+        return f
 
     def tokengetter(self, f):
         """Register a function as the token getter.
@@ -264,6 +268,7 @@ def bearer_token(access_token=None, refresh_token=None):
                 return None
         """
         self._tokengetter = f
+        return f
 
     def tokensetter(self, f):
         """Register a function to save the bearer token.
@@ -288,6 +293,7 @@ def set_token(token, request, *args, **kwargs):
         client object.
         """
         self._tokensetter = f
+        return f
 
     def grantgetter(self, f):
         """Register a function as the grant getter.
@@ -303,6 +309,7 @@ def grant(client_id, code):
             - delete: A function to delete itself
         """
         self._grantgetter = f
+        return f
 
     def grantsetter(self, f):
         """Register a function to save the grant code.
@@ -314,6 +321,7 @@ def set_grant(client_id, code, request, *args, **kwargs):
                 save_grant(client_id, code, request.user, request.scopes)
         """
         self._grantsetter = f
+        return f
 
     def authorize_handler(self, f):
         """Authorization handler decorator.