def callback():
"""
Google authentication Callback
"""
# Redirects if not logged in
if current_user is not None and current_user.is_authenticated:
return redirect(url_for('index'))
# checks for authntication error
if 'error' in request.args:
if request.args.get('error') == 'access_denied':
return 'You denied access.'
return 'Error encountered.'
if 'code' not in request.args and 'state' not in request.args:
return redirect(url_for('login'))
else:
google = get_google_auth(state=session['oauth_state'])
try:
# Fetch tonken
token = google.fetch_token(
Auth.TOKEN_URI,
client_secret=Auth.CLIENT_SECRET,
authorization_response=request.url)
except HTTPError:
return 'HTTPError occurred.'
google = get_google_auth(token=token)
# get user info if availabale
resp = google.get(Auth.USER_INFO)
if resp.status_code == 200:
user_data = resp.json()
email = user_data['email']
user = dbsession.query(User).filter_by(email=email).first()
# sets details if not avialbale
if user is None:
user = User()
user.email = email
user.name = user_data['name']
print(token)
user.tokens = json.dumps(token)
# Adds user to databse
dbsession.add(User(email=user.email, token=user.token, name=user.name)) # noqa
dbsession.commit()
login_user(user)
# redirects to homepage
return redirect(url_for('index'))
return 'Could not fetch your information.'
def callback():
# Redirect user to home page if already logged in.
if current_user is not None and current_user.is_authenticated:
return redirect(url_for('restaurants'))
if 'error' in request.args:
if request.args.get('error') == 'access_denied':
flash('You denied access.')
return
flash('Error encountered')
return
if 'code' not in request.args and 'state' not in request.args:
return redirect(url_for('login'))
else:
# Execution reaches here when user has
# successfully authenticated our app.
print 'my still session', session
google = get_google_auth(state=session['oauth_state'])
try:
token = google.fetch_token(Auth.TOKEN_URI,
client_secret=Auth.CLIENT_SECRET,
authorization_response=request.url)
except HTTPError:
return 'HTTPError occurred.'
google = get_google_auth(token=token)
resp = google.get(Auth.USER_INFO)
if resp.status_code == 200:
user_data = resp.json()
email = user_data['email']
print 'my email', email
user = db_session.query(User).filter_by(email=email).first()
# print 'my google user', user.fetchall()
if user is None:
user = User()
user.email = email
user.username = user_data['name']
print(token)
user.tokens = json.dumps(token)
user.avatar = user_data['picture']
db_session.add(user)
db_session.commit()
login_user(user)
return redirect(url_for('all_restaurants'))
return 'Could not fetch your information.'
