From 3c19321cfaca486ce5a3b7c68a58e9c1dfceefa2 Mon Sep 17 00:00:00 2001
From: Guillaume Subiron <maethor@subiron.org>
Date: Fri, 8 Jun 2012 09:41:44 +0200
Subject: [PATCH] Continue arg in login and logout routes

---
 main.py               | 4 ++++
 templates/layout.html | 4 ++--
 templates/login.html  | 2 +-
 templates/votes.html  | 2 +-
 4 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/main.py b/main.py
index 6b391c3..956e438 100755
--- a/main.py
+++ b/main.py
@@ -92,6 +92,8 @@ def login():
         else:
             connect_user(user)
             flash(u'Vous êtes connecté. Bienvenue, %s !' % user['name'], 'success')
+            if request.args.get('continue'):
+                return redirect(request.args['continue'])
             return redirect(url_for('home'))
     return render_template('login.html')
 
@@ -99,6 +101,8 @@ def login():
 def logout():
     disconnect_user()
     flash(u'Vous avez été déconnecté.', 'info')
+    if request.args.get('continue') and not "admin" in request.args.get('continue'):
+        return redirect(request.args['continue'])
     return redirect(url_for('home'))
 
 #-----------------
diff --git a/templates/layout.html b/templates/layout.html
index 6fe38aa..fff1187 100644
--- a/templates/layout.html
+++ b/templates/layout.html
@@ -44,10 +44,10 @@
         <li><a href="{{ url_for('votes', votes='waiting') }}"><i class="icon-comment"></i> Votes en attente</a></li>
         <li><a href="{{ url_for('user_edit', userid=session.user.id) }}"><i class="icon-cog"></i> Paramètres</a></li>
         <li class="divider"></li>
-        <li><a href="{{ url_for('logout') }}"><i class="icon-off"></i> Déconnexion</a></li>
+        <li><a href="{{ url_for('logout', continue=request.url) }}"><i class="icon-off"></i> Déconnexion</a></li>
       </ul>
       {% else %}
-      <a class="btn btn-primary" href="{{ url_for('login') }}"><i class="icon-user icon-white"></i> Connexion</a>
+      <a class="btn btn-primary" href="{{ url_for('login', continue=request.url) }}"><i class="icon-user icon-white"></i> Connexion</a>
       {% endif %}
     </div>
   </div>
diff --git a/templates/login.html b/templates/login.html
index 332195e..b00dc72 100644
--- a/templates/login.html
+++ b/templates/login.html
@@ -2,7 +2,7 @@
 {% block body %}
 <div class="row">
   <div class="span3 well offset1">
-    <form action="{{ url_for('login') }}" method="post">
+    <form action="{{ url_for('login', continue=request.args.continue) }}" method="post">
     <fieldset><legend>Connexion</legend>
       <label for="username">E-mail</label>
       <input type="text" name="username" id="username" value="{{ request.form.username }}"/>
diff --git a/templates/votes.html b/templates/votes.html
index 0cb1b51..616e3b3 100644
--- a/templates/votes.html
+++ b/templates/votes.html
@@ -31,7 +31,7 @@ Liste des votes
       {% if vote.description %}
       <h4>Description</h4>
       <p>
-      {{ vote.description|safe }}
+      {{ vote.description }}
       </p>
       {% endif %}
     </div>
-- 
2.20.1