views.py 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  1. from django.http import HttpResponse
  2. from django.shortcuts import redirect
  3. from django.urls import reverse
  4. import google_auth_oauthlib.flow
  5. from django.contrib.auth.decorators import login_required
  6. from .models import UserModel
  7. @login_required
  8. def index(request):
  9. if 'credentials' not in request.session:
  10. return redirect(reverse('authorize'))
  11. cred = request.session['credentials']
  12. return HttpResponse(cred['access_token'])
  13. CLIENT_SECRETS_FILE = "client_secrets.json"
  14. SCOPES = ['https://www.googleapis.com/auth/analytics.readonly']
  15. flow = google_auth_oauthlib.flow.Flow.from_client_secrets_file(
  16. CLIENT_SECRETS_FILE,
  17. scopes=SCOPES,
  18. redirect_uri="http://127.0.0.1:8000/oauth2callback")
  19. def authorize(request):
  20. authorization_url, state = flow.authorization_url(access_type='offline')
  21. # Store the state so the callback can verify the auth server response.
  22. request.session['state'] = state
  23. return redirect(authorization_url)
  24. def oauth2callback(request):
  25. state = request.session['state']
  26. flow.state = state
  27. code = request.GET.get('code', False)
  28. flow.fetch_token(code=code)
  29. credentials = flow.credentials
  30. request.session['credentials'] = credentials_to_dict(credentials)
  31. refresh_token = credentials.refresh_token
  32. user_model = UserModel.objects.create(user=request.user, refresh_token=refresh_token)
  33. user_model.save()
  34. return redirect(reverse('home'))
  35. def credentials_to_dict(credentials):
  36. expiry = str(credentials.expiry.utcnow())
  37. return {
  38. 'access_token': credentials.token,
  39. 'expiry': expiry
  40. }