From 206362de0e315d235e9abfdf7119fd2073650efe Mon Sep 17 00:00:00 2001
From: Ben Adida <ben@adida.net>
Date: Mon, 26 Jan 2015 01:32:23 +0000
Subject: [PATCH] fixed google auth session

---
 helios_auth/auth_systems/google.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/helios_auth/auth_systems/google.py b/helios_auth/auth_systems/google.py
index e13a9f4..00a6bb7 100644
--- a/helios_auth/auth_systems/google.py
+++ b/helios_auth/auth_systems/google.py
@@ -28,11 +28,13 @@ def get_flow(redirect_url=None):
 def get_auth_url(request, redirect_url):
   flow = get_flow(redirect_url)
 
-  request.session['google-flow'] = flow
+  request.session['google-redirect-url'] = redirect_url
   return flow.step1_get_authorize_url()
 
 def get_user_info_after_auth(request):
-  flow = request.session['google-flow']
+  flow = get_flow(request.session['google-redirect-url'])
+  del request.session['google-redirect-url']
+
   code = request.GET['code']
   credentials = flow.step2_exchange(code)
 
@@ -79,4 +81,4 @@ def check_constraint(constraint, user_info):
   """
   for eligibility
   """
-  pass
\ No newline at end of file
+  pass
-- 
GitLab