def user_bind_oauth(user, session): """Bind oauth info in session to given user, then clear the session""" oauth = UserOauth(user=user, provider=session['oauth_provider'], openid=session['oauth_openid'], token=session['oauth_token']) oauth.save() # Clear oauth info in session del session['oauth_provider'] del session['oauth_openid'] del session['oauth_name'] del session['oauth_token']
def oauth_callback(provider): current_app.logger.info("Oauth callback for %s" % provider) redirect_uri = current_app.config['OAUTH'][provider]['callback_url'] oauth_service = get_oauth_services()[provider] data = dict(code=request.args.get('code'), grant_type='authorization_code', redirect_uri=redirect_uri) if provider == 'google': oauth_session = oauth_service.get_auth_session(data=data, decoder=json.loads) user_info = oauth_session.get('userinfo').json() email = session['email'] = user_info['email'] username = user_info['given_name'] elif provider == 'github': oauth_session = oauth_service.get_auth_session(data=data) user_info = oauth_session.get('user').json() email = session['email'] = user_info['email'] username = user_info['login'] access_token = oauth_session.access_token user_info['id'] = str(user_info['id']) current_app.logger.info("%s oauth access token is: %s" % (provider, access_token)) current_app.logger.info("%s oauth user info is %s" % (provider, user_info)) user = User.find_by_oauth(provider, user_info['id']) if user: # TODO: 直接登录时更新 token. user_mixin = LoginManagerUser(user) login_user(user_mixin) flash(u"登录成功", category='info') return redirect('/') else: user = User.objects(email=email).first() if user: user_oauth = UserOauth(provider=provider, openid=user_info['id'], token=access_token) user_oauth.save() user_mixin = LoginManagerUser(user) login_user(user_mixin) flash(u"登录成功", category='info') return redirect('/') else: return render_template('users/finish_signup.html', form=UserInfoForm(email=email, username=username))
def oauth_callback(provider): logger.info("Oauth callback for %s" % provider) redirect_uri = app.config['oauth.%s.callback_url' % provider] oauth_service = oauth_services[provider] session = get_session(request) data = dict(code=request.params.get('code'), grant_type='authorization_code', redirect_uri=redirect_uri) if provider == 'google': oauth_session = oauth_service.get_auth_session(data=data, decoder=json.loads) user_info = oauth_session.get('userinfo').json() email = session['email'] = user_info['email'] username = user_info['given_name'] elif provider == 'github': oauth_session = oauth_service.get_auth_session(data=data) user_info = oauth_session.get('user').json() email = session['email'] = user_info['email'] username = user_info['login'] access_token = oauth_session.access_token user_info['id'] = str(user_info['id']) logger.info("%s oauth access token is: %s" % (provider, access_token)) logger.info("%s oauth user info is %s" % (provider, user_info)) user = User.find_by_oauth(provider, user_info['id']) if user: # TODO: 直接登录时更新 token. login.login_user(str(user.id)) return redirect('/') else: user = User.objects(email=email).first() if user: user_oauth = UserOauth(provider=provider, openid=user_info['id'], token=access_token) user_oauth.save() login.login_user(str(user.id)) return redirect('/') else: return { 'form': UserInfoForm(email=email, username=username), 'token': request.csrf_token }
def oauth_callback(provider): logger.info("Oauth callback for %s" % provider) redirect_uri = app.config['oauth.%s.callback_url' % provider] oauth_service = oauth_services[provider] session = get_session(request) data = dict(code=request.params.get('code'), grant_type='authorization_code', redirect_uri=redirect_uri) if provider == 'google': oauth_session = oauth_service.get_auth_session(data=data, decoder=json.loads) user_info = oauth_session.get('userinfo').json() email = session['email'] = user_info['email'] username = user_info['given_name'] elif provider == 'github': oauth_session = oauth_service.get_auth_session(data=data) user_info = oauth_session.get('user').json() email = session['email'] = user_info['email'] username = user_info['login'] access_token = oauth_session.access_token user_info['id'] = str(user_info['id']) logger.info("%s oauth access token is: %s" % (provider, access_token)) logger.info("%s oauth user info is %s" % (provider, user_info)) user = User.find_by_oauth(provider, user_info['id']) if user: # TODO: 直接登录时更新 token. login.login_user(str(user.id)) return redirect('/') else: user = User.objects(email=email).first() if user: user_oauth = UserOauth(provider=provider, openid=user_info['id'], token=access_token) user_oauth.save() login.login_user(str(user.id)) return redirect('/') else: return {'form': UserInfoForm(email=email, username=username), 'token': request.csrf_token}