handle wrong login/password and improve translations

1 files changed, 11 insertions, 4 deletions

diff --git a/wedding.py b/wedding.py
index f1e99e2..c8b5b54 100644
--- a/wedding.py
+++ b/wedding.py
@@ -2,6 +2,7 @@
 import sqlite3
 #all the imports
 from flask import Flask, request, session, g, redirect, url_for, render_template
+from flask import flash
 from functools import wraps
 
 app = Flask(__name__)
@@ -76,6 +77,12 @@ def main_page():
 
 @app.route('/login/', methods=['GET', 'POST'])
 def login():
+    wp = {'en': 'Wrong password',
+          'fr': 'Mot de passe erroné',
+          'sr': 'Pogrešna šifra'}
+    wn = {'en': 'Wrong name',
+          'fr': 'Prénom incorrect',
+          'sr': 'Pogrešno ime'}
     if 'user_name' in session:
         return redirect(url_for('main_page'))
 
@@ -86,14 +93,14 @@ def login():
         c.execute("SELECT * from guests where user_name=?",
                   (request.form['username'],))
         if not c.fetchone():
-            error = "User not registered"
             c.close()
-            return
+            flash(wn[request.form['language']])
+            return redirect(url_for('login'))
         else:
             c.close()
         if request.form['password'] != app.config['PASSWORD']:
-            error = "Wrong password"
-            return
+            flash(wp[request.form['language']])
+            return redirect(url_for('login'))
         session['user_name'] = request.form['username']
         session['lang'] = request.form['language']
         return redirect(url_for('main_page'))