Continue arg in login and logout routes

[cavote.git] / templates / layout.html

diff --git a/templates/layout.html b/templates/layout.html
index 6fe38aa..fff1187 100644 (file)
--- 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>