diff --git a/flask_oauthlib/provider/oauth1.py b/flask_oauthlib/provider/oauth1.py index f95547c8..627cf5ff 100644 --- a/flask_oauthlib/provider/oauth1.py +++ b/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. diff --git a/flask_oauthlib/provider/oauth2.py b/flask_oauthlib/provider/oauth2.py index eafa9970..878a1502 100644 --- a/flask_oauthlib/provider/oauth2.py +++ b/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.