add a logout link

1 files changed, 5 insertions, 0 deletions

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: