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

added django-allauth support for obtaining google token

Browse source
This commit is contained in:
ak4zh 2020-08-30 14:17:37 +05:30
parent c8751f0c9a
commit 84543396f7
1 changed files with 28 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

37
django_mailbox/google_utils.py
View file

@ -3,7 +3,10 @@ import logging
from django.conf import settings from django.conf import settings
import requests import requests
from social.apps.django_app.default.models import UserSocialAuth from social.apps.django_app.default.models import UserSocialAuth
try:
from allauth.socialaccount.models import SocialAccount, SocialApp
except ImportError:
SocialAccount, SocialApp = None, None
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -17,34 +20,46 @@ class RefreshTokenNotFound(Exception):
def get_google_consumer_key(): def get_google_consumer_key():
if SocialApp:
app = SocialApp.objects.filter(provider='google').first()
if app:
return app.client_id
return settings.SOCIAL_AUTH_GOOGLE_OAUTH2_KEY return settings.SOCIAL_AUTH_GOOGLE_OAUTH2_KEY
def get_google_consumer_secret(): def get_google_consumer_secret():
if SocialApp:
app = SocialApp.objects.filter(provider='google').first()
if app:
return app.secret
return settings.SOCIAL_AUTH_GOOGLE_OAUTH2_SECRET return settings.SOCIAL_AUTH_GOOGLE_OAUTH2_SECRET
def get_google_account(email, key=None):
try:
if SocialAccount:
me = SocialAccount.objects.get(uid=email, provider='google')
else:
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
def get_google_access_token(email): def get_google_access_token(email):
# TODO: This should be cacheable # TODO: This should be cacheable
try: return get_google_account(email, key='access_token')
me = UserSocialAuth.objects.get(uid=email, provider="google-oauth2")
return me.extra_data['access_token']
except (UserSocialAuth.DoesNotExist, KeyError):
raise AccessTokenNotFound
def update_google_extra_data(email, extra_data): def update_google_extra_data(email, extra_data):
try: me = get_google_account(email)
me = UserSocialAuth.objects.get(uid=email, provider="google-oauth2")
me.extra_data = extra_data me.extra_data = extra_data
me.save() me.save()
except (UserSocialAuth.DoesNotExist, KeyError):
raise AccessTokenNotFound
def get_google_refresh_token(email): def get_google_refresh_token(email):
try: try:
me = UserSocialAuth.objects.get(uid=email, provider="google-oauth2") me = get_google_account(email)
return me.extra_data['refresh_token'] return me.extra_data['refresh_token']
except (UserSocialAuth.DoesNotExist, KeyError): except (UserSocialAuth.DoesNotExist, KeyError):
raise RefreshTokenNotFound raise RefreshTokenNotFound