add a logout link

2 files changed, 6 insertions, 0 deletions

diff --git a/templates/layout.html b/templates/layout.html
index 81c136d..f8ac3b9 100644
--- a/templates/layout.html
+++ b/templates/layout.html
@@ -21,6 +21,7 @@
           <li role="presentation" id="getting"><a href="../getting">{{_("Getting There")}}</a></li>
           <li role="presentation" id="lodging"><a href="../lodging">{{_("Lodging")}}</a></li>
           <li role="presentation" id="gifts"><a href="../gifts">{{_("Gifts")}}</a></li>
+          <li role="presentation" id="logout"><a href="../logout">{{_("Logout")}}</a></li>
         </ul>
       </div>
       <div class="col-md-9">
diff --git a/wedding.py b/wedding.py
index bb3df7a..14134c2 100644
--- a/wedding.py
+++ b/wedding.py
@@ -96,6 +96,11 @@ def rsvp():
 def main_page():
     return render_template("main.html")
 
+@app.route('/logout')
+def logout():
+    session.clear()
+    return redirect(url_for('login'))
+
 @app.route('/login/', methods=['GET', 'POST'])
 def login():
     if 'user_name' in session: