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 find_by_oauth(cls, provider, openid):
"""Find user that has oauth info with given provider and openid"""
oauth = UserOauth.objects(provider=provider, openid=openid).first()
if oauth and oauth.user:
return oauth.user
def find_by_oauth(cls, provider, openid):
"""Find user that has oauth info with given provider and openid"""
oauth = UserOauth.objects(provider=provider, openid=openid).first()
if oauth and oauth.user:
return oauth.user
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):
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}
