From 2d9b1654749a61c05796c73e049e8e1c6bb1722c Mon Sep 17 00:00:00 2001 From: Guillaume Horel Date: Sat, 21 Mar 2015 18:33:17 -0400 Subject: handle wrong login/password and improve translations --- wedding.py | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) (limited to 'wedding.py') 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')) -- cgit v1.2.3-70-g09d2