Skip to content

Commit da61f80

Browse files
committed
Merge branch 'customizable-on_obtaining_tokens' into dev
2 parents 4095aa4 + 0d4b8c2 commit da61f80

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

oauth2cli/oauth2.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)