orbit-oss/flask
4
5
Fork 2
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Added snippet admin and profile editing

Browse source
This commit is contained in:
Armin Ronacher 2010-05-16 13:51:12 +02:00
parent aa79c72ae0
commit d87cbff134
14 changed files with 228 additions and 40 deletions
Download patch file Download diff file Expand all files Collapse all files

78
flask_website/views/general.py
View file

@ -1,6 +1,8 @@
from flask import Module, render_template, session, redirect, url_for, \
request, flash, g, Response
from flaskext.openid import COMMON_PROVIDERS
from flask_website import oid, twitter
from flask_website.utils import requires_login
from flask_website.database import db_session, User
general = Module(__name__)
@ -30,28 +32,20 @@ def logout():
def login():
if g.user is not None:
return redirect(url_for('general.index'))
if request.method == 'POST':
openid = request.values.get('openid')
if openid:
return oid.try_login(openid, ask_for=['fullname', 'nickname'])
if reassign and 'cancel' in request.form:
flash(u'Cancelled. The OpenID was not changed.')
return redirect(oid.get_next_url())
openid = request.values.get('openid')
if not openid:
openid = COMMON_PROVIDERS.get(request.args.get('provider'))
if openid:
return oid.try_login(openid, ask_for=['fullname', 'nickname'])
error = oid.fetch_error()
if error:
flash(u'Error: ' + error)
return render_template('general/login.html', next=oid.get_next_url())
@oid.after_login
def create_or_login(resp):
session['openid'] = resp.identity_url
user = User.query.filter_by(openid=resp.identity_url).first()
if user is not None:
flash(u'Successfully signed in')
g.user = user
return redirect(oid.get_next_url())
return redirect(url_for('first_login', next=oid.get_next_url(),
name=resp.fullname or resp.nickname))
@general.route('/first-login/', methods=['GET', 'POST'])
def first_login():
if g.user is not None or 'openid' not in session:
@ -68,3 +62,55 @@ def first_login():
return render_template('general/first_login.html',
next=oid.get_next_url(),
openid=session['openid'])
@general.route('/profile/', methods=['GET', 'POST'])
@requires_login
def profile():
name = g.user.name
if request.method == 'POST':
name = request.form['name'].strip()
if not name:
flash(u'Error: a name is required')
else:
g.user.name = name
db_session.commit()
flash(u'User profile updated')
return redirect(url_for('index'))
return render_template('general/profile.html', name=name)
@general.route('/profile/change-openid/', methods=['GET', 'POST'])
@requires_login
@oid.loginhandler
def change_openid():
if request.method == 'POST':
if 'cancel' in request.form:
flash(u'Cancelled. The OpenID was not changed.')
return redirect(oid.get_next_url())
openid = request.values.get('openid')
if not openid:
openid = COMMON_PROVIDERS.get(request.args.get('provider'))
if openid:
return oid.try_login(openid)
error = oid.fetch_error()
if error:
flash(u'Error: ' + error)
return render_template('general/change_openid.html',
next=oid.get_next_url())
@oid.after_login
def create_or_login(resp):
session['openid'] = resp.identity_url
user = g.user or User.query.filter_by(openid=resp.identity_url).first()
if user is None:
return redirect(url_for('first_login', next=oid.get_next_url(),
name=resp.fullname or resp.nickname))
if user.openid != resp.identity_url:
user.openid = resp.identity_url
db_session.commit()
flash(u'OpenID identity changed')
else:
flash(u'Successfully signed in')
return redirect(oid.get_next_url())