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('index')) 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: # Execution reaches here when user has successfully authenticated our app. google = get_google_auth(state=session['oauth_state']) try: token = google.fetch_token(Config.TOKEN_URI, client_secret=Config.CLIENT_SECRET, authorization_response=request.url) except HTTPError: return 'HTTPError occurred.' google = get_google_auth(token=token) resp = google.get(Config.USER_INFO) if resp.status_code == 200: user_data = resp.json() email = user_data['email'] user = Users.query.filter_by(email=email).first() if user is None: user = Users() user.email = email user.name = user_data['name'] user.tokens = json.dumps(token) user.avatar = user_data['picture'] db.session.add(user) db.session.commit() login_user(user) return redirect(url_for('index')) return 'Could not fetch your information.'
def test_avatar(self): u = Users(username='******', email='*****@*****.**') self.assertEqual(u.avatar(128), ('https://www.gravatar.com/avatar/' 'd4c74594d841139328695756648b6bd6' '?d=identicon&s=128'))
def test_avatar(self): # create a user u = Users(username='******', email='*****@*****.**', password = '******') avatar = u.avatar(128) expected = 'http://www.gravatar.com/avatar/d4c74594d841139328695756648b6bd6' assert avatar[0:len(expected)] == expected