X-Git-Url: http://git.veekun.com/zzz-floof.git/blobdiff_plain/324ea2adc265de0c02ea56328f5f0a420dbaa999..efdbb22873232827398ba7605f8b35c605c043e9:/floof/lib/base.py diff --git a/floof/lib/base.py b/floof/lib/base.py index d05c24d..b42d2be 100644 --- a/floof/lib/base.py +++ b/floof/lib/base.py @@ -3,17 +3,20 @@ Provides the BaseController class for subclassing. """ from pylons.controllers import WSGIController +from pylons.controllers.util import abort, redirect from pylons.templating import render_mako as render -from pylons import config -from floof import model +from pylons import config, request, session, tmpl_context as c +from routes import request_config -from pylons import session, tmpl_context as c +from floof import model from floof.model.users import User class BaseController(WSGIController): # NOTE: This could have been implemented as a middleware =] def __before__(self): + c.route = request_config().mapper_dict + # Fetch current user object try: c.user = User.query.get(session['user_id']) @@ -32,3 +35,11 @@ class BaseController(WSGIController): return WSGIController.__call__(self, environ, start_response) finally: model.Session.remove() + + + def redirect_to_referrer(self): + """Performs a redirect_to to wherever we came from. Used for stuff + like logging in. + """ + referrer = request.headers.get('REFERER', '/') + redirect(referrer, code=302)