diff options
Diffstat (limited to 'wedding.py')
| -rw-r--r-- | wedding.py | 15 |
1 files changed, 11 insertions, 4 deletions
@@ -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')) |