def authorized(oauth_token): next_url = request.args.get('next') or url_for('index') if oauth_token is None: flash('Authorization failed.', 'alert') return redirect(next_url) user = db_session.query(User).filter_by(access_token=oauth_token).first() if user is None: user = User(oauth_token) db_session.add(user) user.access_token = oauth_token db_session.commit() # Store current user details in a session session['usr_id'] = user.id session['usr_token'] = user.access_token flash('Logged in', 'success') return redirect(url_for('index'))
def authorized(oauth_token): """ callback from github oauth """ next_url = request.args.get('next') if oauth_token is None: flash('Authorization failed.', 'danger') return redirect(next_url) user = db_session.query(User).filter_by(access_token=oauth_token).first() if user is None: user = User(oauth_token) db_session.add(user) # store access token into database user.access_token = oauth_token db_session.commit() # store user id and access token in session hash session['user_id'] = user.id session['user_token'] = user.access_token flash('You\'re logged in! Now you have incredible superpowers...', 'success') return redirect(url_for('index'))