@@ -734,6 +734,7 @@ def __init__(self,
734734 def _obtain_token (
735735 self , grant_type , params = None , data = None ,
736736 also_save_rt = False ,
737+ on_obtaining_tokens = None ,
737738 * args , ** kwargs ):
738739 _data = data .copy () # to prevent side effect
739740 resp = super (Client , self )._obtain_token (
@@ -753,7 +754,7 @@ def _obtain_token(
753754 # but our obtain_token_by_authorization_code(...) encourages
754755 # app developer to still explicitly provide a scope here.
755756 scope = _data .get ("scope" )
756- self .on_obtaining_tokens ({
757+ ( on_obtaining_tokens or self .on_obtaining_tokens ) ({
757758 "client_id" : self .client_id ,
758759 "scope" : scope ,
759760 "token_endpoint" : self .configuration ["token_endpoint" ],
@@ -767,6 +768,7 @@ def obtain_token_by_refresh_token(self, token_item, scope=None,
767768 rt_getter = lambda token_item : token_item ["refresh_token" ],
768769 on_removing_rt = None ,
769770 on_updating_rt = None ,
771+ on_obtaining_tokens = None ,
770772 ** kwargs ):
771773 # type: (Union[str, dict], Union[str, list, set, tuple], Callable) -> dict
772774 """This is an overload which will trigger token storage callbacks.
0 commit comments