X-Git-Url: http://git.veekun.com/zzz-floof.git/blobdiff_plain/fca6f64bc81a02c61ba45a13de25f3da2e7630b3..79cb1c0d3048c165fd3b9ff287ee1fbe361f904d:/floof/lib/base.py?ds=sidebyside

diff --git a/floof/lib/base.py b/floof/lib/base.py
index 169493c..c93803b 100644
--- a/floof/lib/base.py
+++ b/floof/lib/base.py
@@ -3,8 +3,9 @@
 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, session, tmpl_context as c
+from pylons import config, request, session, tmpl_context as c
 from routes import request_config
 
 from floof import model
@@ -20,6 +21,7 @@ class BaseController(WSGIController):
         try:
             c.user = User.query.get(session['user_id'])
         except:
+            c.user = None
             pass
 
     def __call__(self, environ, start_response):
@@ -34,3 +36,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)