Add login_required decorator

views/loginlogout.py

@@ -6,6 +6,7 @@ from socket import gethostname
 from os import remove, system 
 from tools.utils import email_disp, valid_token_register, valid_passwd, valid_username, gen_token, totp_is_valid
 from tools.mailer import Mailer
+from tools.utils import login_required
 
 app = Flask( 'pywallter' )
 app.config.from_pyfile('config.py')
@@ -29,8 +30,34 @@ BACKUP_TIME = app.config['BACKUP_TIME']
 
 loginlogout = Blueprint('loginlogout', __name__, template_folder='templates')
 
+
+@loginlogout.route( '/' )
+def index():
+    conn = sqlite3.connect(DATABASE) # Connexion à la base de donnée
+    cursor = conn.cursor() # Création de l'objet "curseur"
+    cursor.execute("""SELECT token passwd FROM users where name=? """, ("pywallter", ))
+    tmp = cursor.fetchone()
+    conn.close
+    if tmp:
+        token = tmp[0]
+    else:
+        token = None
+
+    if 'username' in session :
+        return redirect(url_for('profil.profile'))
+    else :
+        if token:
+            hostname = gethostname()
+            url_inscription = BASE_URL+'inscription/'+token
+            return render_template('inscription.html', signin_enable=app.config['SIGNIN_ENABLE'],
+                                   token=token, hostname=hostname,
+                                   url_inscription=url_inscription,
+                                   MAIL_SERVER=MAIL_SERVER)
+        else:
+            return redirect(url_for('loginlogout.login', _external=True))
+
 @loginlogout.route( '/login/', methods=['GET','POST'] )
-def login() :
+def login():
     if 'username' in session :
         resp = redirect(url_for('profil.profile', _external=True))
     else :
@@ -63,65 +90,66 @@ def login() :
 
 
 @loginlogout.route( '/logout/' )
+@login_required
 def logout():
     session.pop('username', None) # Supprimer username de la session s'il s'y trouve
     return redirect(url_for('loginlogout.index'))
 
 
 @loginlogout.route( '/delete_me/', methods=['GET','POST'])
+@login_required
 def delete_account():
-    if 'username' in session :
-        user='%s'% escape(session['username'])
-        resp = render_template('delete_account.html', time_backup=BACKUP_TIME)
-        if request.method == 'POST' :
-            conn = sqlite3.connect(DATABASE) # Connexion à la base de donnée
-            cursor = conn.cursor() # Création de l'objet "curseur"
-            cursor.execute("""SELECT passwd FROM users WHERE name=?""", (user,))
-            passwd = cursor.fetchone()[0].decode()
-            conn.close()
-            password = request.form['passwd']
-            if bcrypt.check_password_hash(passwd, password) is True:
-                not_error = True 
+    user='%s'% escape(session['username'])
+    resp = render_template('delete_account.html', time_backup=BACKUP_TIME)
+    if request.method == 'POST' :
+        conn = sqlite3.connect(DATABASE) # Connexion à la base de donnée
+        cursor = conn.cursor() # Création de l'objet "curseur"
+        cursor.execute("""SELECT passwd FROM users WHERE name=?""", (user,))
+        passwd = cursor.fetchone()[0].decode()
+        conn.close()
+        password = request.form['passwd']
+        if bcrypt.check_password_hash(passwd, password) is True:
+            not_error = True 
+            try:
+                cmd = 'rm -r ' + DATAS_USER + '/' + user
+                if system(cmd) != 0:
+                    raise TypeError("Remove directory error")
+            except:
+                not_error = False
+                flash(u'Erreur lors de la suppression de votre dossier utilisateur.', 'error')
+
+            if MAIL_SERVER:
                 try:
-                    cmd = 'rm -r ' + DATAS_USER + '/' + user
-                    if system(cmd) != 0:
-                        raise TypeError("Remove directory error")
+                    cmd = SETUID + ' set_mail_passwd del' + '"'+mail+'"'
+                    system(cmd)
                 except:
                     not_error = False
-                    flash(u'Erreur lors de la suppression de votre dossier utilisateur.', 'error')
-
-                if MAIL_SERVER:
-                    try:
-                        cmd = SETUID + ' set_mail_passwd del' + '"'+mail+'"'
-                        system(cmd)
-                    except:
-                        not_error = False
-                        flash(u'Erreur lors de la suppression de votre compte Mail.', 'error')
+                    flash(u'Erreur lors de la suppression de votre compte Mail.', 'error')
                 
-                if XMPP_SERVER:
-                    try:
-                        tmp = mail.split('@')
-                        cmd = SETUID+ ' prosodyctl deluser ' "'"+tmp[0]+"' " + "'"+tmp[1]+"'"
-                        system(cmd)
-                    except:
-                        not_error = False
-                        flash(u'Erreur lors de la suppression de votre compte XMPP.', 'error')
+            if XMPP_SERVER:
+                try:
+                    tmp = mail.split('@')
+                    cmd = SETUID+ ' prosodyctl deluser ' "'"+tmp[0]+"' " + "'"+tmp[1]+"'"
+                    system(cmd)
+                except:
+                    not_error = False
+                    flash(u'Erreur lors de la suppression de votre compte XMPP.', 'error')
 
-                if not_error:
-                    try:
-                        conn = sqlite3.connect(DATABASE)
-                        cursor = conn.cursor()
-                        cursor.execute("""DELETE FROM users WHERE name=?""", (user,))
-                        conn.commit()
-                        conn.close()
-                    except:
-                        flash(u'Erreur lors de la suppression de votre compte.', 'error')
-                    else:
-                        flash(u'Désinscription réalisé avec succés, y\'a plus rien !', 'succes')
-                        resp = redirect(url_for('loginlogout.logout'))
+            if not_error:
+                try:
+                    conn = sqlite3.connect(DATABASE)
+                    cursor = conn.cursor()
+                    cursor.execute("""DELETE FROM users WHERE name=?""", (user,))
+                    conn.commit()
+                    conn.close()
+                except:
+                    flash(u'Erreur lors de la suppression de votre compte.', 'error')
             else:
-                flash(u'Mauvais mot de passe', 'error')
-        return resp
+                flash(u'Désinscription réalisé avec succés, y\'a plus rien !', 'succes')
+                resp = redirect(url_for('loginlogout.logout'))
+        else:
+            flash(u'Mauvais mot de passe', 'error')
+    return resp
     
 
 @loginlogout.route( '/lost_password/', methods=['GET', 'POST'])
@@ -160,27 +188,3 @@ def lost_password():
     
     return render_template('lost_password.html')
     
-@loginlogout.route( '/' )
-def index():
-    conn = sqlite3.connect(DATABASE) # Connexion à la base de donnée
-    cursor = conn.cursor() # Création de l'objet "curseur"
-    cursor.execute("""SELECT token passwd FROM users where name=? """, ("pywallter", ))
-    tmp = cursor.fetchone()
-    conn.close
-    if tmp:
-        token = tmp[0]
-    else:
-        token = None
-
-    if 'username' in session :
-        return redirect(url_for('profil.profile'))
-    else :
-        if token:
-            hostname = gethostname()
-            url_inscription = BASE_URL+'inscription/'+token
-            return render_template('inscription.html', signin_enable=app.config['SIGNIN_ENABLE'],
-                                   token=token, hostname=hostname,
-                                   url_inscription=url_inscription,
-                                   MAIL_SERVER=MAIL_SERVER)
-        else:
-            return redirect(url_for('loginlogout.login', _external=True))