mirror of
https://github.com/TurTaskProject/TurTaskWeb.git
synced 2025-12-19 05:54:07 +01:00
50 lines
1.9 KiB
Python
50 lines
1.9 KiB
Python
from django.core.cache import cache
|
|
from django.conf import settings
|
|
|
|
from google.auth.transport.requests import Request
|
|
from google.oauth2.credentials import Credentials
|
|
from googleapiclient.discovery import build
|
|
|
|
from .models import CustomUser
|
|
|
|
|
|
def store_token(user_id, token, token_type):
|
|
cache_key = f"user_{token_type}_token:{user_id}"
|
|
cache.set(cache_key, token, timeout=3600)
|
|
|
|
|
|
def get_credential_from_cache_token(user_id):
|
|
access_token = cache.get(f"user_access_token:{user_id}")
|
|
id_token = cache.get(f"user_id_token:{user_id}")
|
|
refresh_token = CustomUser.objects.get(id=user_id).refresh_token
|
|
scopes = [
|
|
'https://www.googleapis.com/auth/userinfo.email',
|
|
'https://www.googleapis.com/auth/userinfo.profile',
|
|
'https://www.googleapis.com/auth/calendar.readonly',
|
|
]
|
|
# credentials = Credentials.from_authorized_user_info(
|
|
# {
|
|
# 'access_token': access_token,
|
|
# 'token_uri': 'https://oauth2.googleapis.com/token',
|
|
# 'refresh_token': refresh_token,
|
|
# 'client_id': settings.GOOGLE_CLIENT_ID,
|
|
# 'client_secret': settings.GOOGLE_CLIENT_SECRET,
|
|
# 'id_token': id_token,
|
|
# }
|
|
|
|
credentials = Credentials(token=access_token,
|
|
refresh_token=refresh_token,
|
|
token_uri='https://oauth2.googleapis.com/token',
|
|
client_id=settings.GOOGLE_CLIENT_ID,
|
|
client_secret=settings.GOOGLE_CLIENT_SECRET,
|
|
scopes=scopes,
|
|
id_token=id_token
|
|
)
|
|
|
|
# If can refresh, refresh
|
|
if credentials.expired and credentials.refresh_token:
|
|
credentials.refresh(Request())
|
|
store_token(user_id, credentials.token, 'access')
|
|
store_token(user_id, credentials.id_token, 'id')
|
|
|
|
return credentials |