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 signin():
form = SigninForm()
if request.method == 'GET':
return render_template('users/signin.html', form=SigninForm())
else:
if form.validate_on_submit():
user = User.objects.get_or_404(email=form.email.data)
user_mixin = LoginManagerUser(user)
login_user(user_mixin)
flash(u"登录成功", category='info')
return redirect(url_for('site_app.index'))
return render_template('users/signin.html', form=form)
def manage():
form = UserInfoForm(request.form)
if form.validate():
if current_user.is_authenticated:
current_user.user.username = form.username.data
return redirect('/')
else:
user = User(email=form.email.data, username=form.username.data,
is_email_confirmed=True)
user.save()
user_mixin = LoginManagerUser(user)
login_user(user_mixin)
if 'email' in session:
del(session['email'])
return redirect('/')
return render_template('users/manage.html',
form=form)
def signup():
form = SignupForm()
if request.method == 'GET':
return render_template('users/signup.html', form=form)
else:
if form.validate_on_submit():
user = User()
form.populate_obj(user)
user.save()
bookmark = Bookmark(user=user,
title=u"%s 的收藏夹" % user.username,
is_default=True)
bookmark.save()
user_mixin = LoginManagerUser(user)
login_user(user_mixin)
send_confirm_email(current_app.config, user.email)
return redirect(url_for('site_app.index'))
return render_template('users/signup.html', form=form)
def finish_signup():
form = UserInfoForm(request.form)
if form.validate():
if current_user.is_authenticated:
current_user.user.username = form.username.data
return redirect('/')
else:
user = User(email=form.email.data,
username=form.username.data,
is_email_confirmed=True)
user.save()
bookmark = Bookmark(user=user,
title=u"%s 的收藏夹" % user.username,
is_default=True)
bookmark.save()
user_mixin = LoginManagerUser(user)
login_user(user_mixin)
flash(u"登录成功", category='info')
if 'email' in session:
del (session['email'])
return redirect('/')
return render_template('users/finish_signup.html', form=form)
def load_user(user_id):
user = User.objects.get_or_404(id=user_id)
return LoginManagerUser(user)