From ac6b25e737aa9f533d27aa4711b3d3d35967dcaa Mon Sep 17 00:00:00 2001 From: ak4zh Date: Sun, 30 Aug 2020 14:21:40 +0530 Subject: [PATCH] removed duplicate codes for obtaining different data from google extra data dict --- django_mailbox/google_utils.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/django_mailbox/google_utils.py b/django_mailbox/google_utils.py index 422defc..a3e3fba 100644 --- a/django_mailbox/google_utils.py +++ b/django_mailbox/google_utils.py @@ -43,7 +43,7 @@ def get_google_account(email, key=None): me = UserSocialAuth.objects.get(uid=email, provider="google-oauth2") return me.extra_data[key] if key else me except (UserSocialAuth.DoesNotExist, SocialAccount.DoesNotExist, KeyError): - raise AccessTokenNotFound + raise RefreshTokenNotFound if key == 'refresh_token' else AccessTokenNotFound def get_google_access_token(email): @@ -58,11 +58,7 @@ def update_google_extra_data(email, extra_data): def get_google_refresh_token(email): - try: - me = get_google_account(email) - return me.extra_data['refresh_token'] - except (UserSocialAuth.DoesNotExist, KeyError): - raise RefreshTokenNotFound + return get_google_account(email, key='refresh_token') def google_api_get(email, url):