From 9ade711d597a6eb5e1080bc7d3fa1fe96cd57ecc Mon Sep 17 00:00:00 2001 From: Slava Bacherikov Date: Fri, 13 Apr 2012 17:44:13 +0300 Subject: [PATCH] Litle change library, add pass response to user_data method of BaseOAuth2. --- social_auth/backends/__init__.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/social_auth/backends/__init__.py b/social_auth/backends/__init__.py index 10f597b..bf7767d 100644 --- a/social_auth/backends/__init__.py +++ b/social_auth/backends/__init__.py @@ -698,7 +698,16 @@ class BaseOAuth2(BaseOAuth): error = response.get('error_description') or response.get('error') raise AuthFailed(self, error) else: - response.update(self.user_data(response['access_token']) or {}) + try: + argnum = self.user_data.im_func.func_code.co_argcount + except AttributeError: + argnum = 2 + finally: + user_data_args = [response['access_token']] + if argnum == 3: + user_data_args.append(response) + + response.update(self.user_data(*user_data_args) or {}) kwargs.update({ 'auth': self, 'response': response, -- 2.39.5