mirror/django-mailbox
1
0
Fork 1
mirror of https://github.com/coddingtonbear/django-mailbox.git synced 2026-07-09 22:38:19 +02:00
Code Activity

removed duplicate codes for obtaining different data from google extra data dict

Browse source
This commit is contained in:
ak4zh 2020-08-30 14:21:40 +05:30
parent 84543396f7
commit ac6b25e737
1 changed files with 2 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

8
django_mailbox/google_utils.py
View file

@ -43,7 +43,7 @@ def get_google_account(email, key=None):
me = UserSocialAuth.objects.get(uid=email, provider="google-oauth2") me = UserSocialAuth.objects.get(uid=email, provider="google-oauth2")
return me.extra_data[key] if key else me return me.extra_data[key] if key else me
except (UserSocialAuth.DoesNotExist, SocialAccount.DoesNotExist, KeyError): except (UserSocialAuth.DoesNotExist, SocialAccount.DoesNotExist, KeyError):
raise AccessTokenNotFound raise RefreshTokenNotFound if key == 'refresh_token' else AccessTokenNotFound
def get_google_access_token(email): def get_google_access_token(email):
@ -58,11 +58,7 @@ def update_google_extra_data(email, extra_data):
def get_google_refresh_token(email): def get_google_refresh_token(email):
try: return get_google_account(email, key='refresh_token')
me = get_google_account(email)
return me.extra_data['refresh_token']
except (UserSocialAuth.DoesNotExist, KeyError):
raise RefreshTokenNotFound
def google_api_get(email, url): def google_api_get(email, url):