Amélioration des post-it Ajout supp son compte

views/loginlogout.py

@@ -2,6 +2,7 @@ from flask import Blueprint, Flask, request, flash, render_template, url_for, se
 import sqlite3
 from flask_bcrypt import Bcrypt
 from socket import gethostname
+from os import remove, system 
 
 app = Flask( 'pywallter' )
 app.config.from_pyfile('config.py')
@@ -10,15 +11,17 @@ bcrypt = Bcrypt(app)
 #### Variables ####################################################################################
 
 bcrypt = Bcrypt(app)
-DOSSIER_PERSO= app.config['DOSSIER_APP']
+DATAS_USER = app.config['DOSSIER_APP']
 
 extensionimg = app.config['EXT_IMG']
 
 DATABASE = app.config['DATABASE']
 
 BASE_URL = app.config['BASE_URL']
-
+SETUID = app.config['SETUID']
 MAIL_SERVER = app.config['MAIL_SERVER']
+XMPP_SERVER = app.config['XMPP_SERVER']
+BACKUP_TIME = app.config['BACKUP_TIME']
 ##################################################################################################
 
 
@@ -27,7 +30,7 @@ loginlogout = Blueprint('loginlogout', __name__, template_folder='templates')
 @loginlogout.route( '/login/', methods=['GET','POST'] )
 def login() :
     if 'username' in session :
-        resp = redirect(url_for('filesupload.uploadfiles', _external=True))
+        resp = redirect(url_for('profil.profile', _external=True))
     else :
         resp = redirect(url_for('loginlogout.login', _external=True))
         if request.method == 'POST' :
@@ -54,10 +57,64 @@ 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'])
+def delete_account():
+    if 'username' in session :
+        UTILISATEUR='%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=?""", (UTILISATEUR,))
+            passwd = cursor.fetchone()[0]
+            conn.close()
+            password = request.form['passwd']
+            if bcrypt.check_password_hash(passwd, password) is True:
+                not_error = True 
+                try:
+                    cmd = 'rm -r ' + DATAS_USER + '/' + UTILISATEUR
+                    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 = SETUID + ' set_mail_passwd del' + '"'+mail+'"'
+                        system(cmd)
+                    except:
+                        not_error = False
+                        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 not_error:
+                    try:
+                        conn = sqlite3.connect(DATABASE)
+                        cursor = conn.cursor()
+                        cursor.execute("""DELETE FROM users WHERE name=?""", (UTILISATEUR,))
+                        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'))
+            else:
+                flash(u'Mauvais mot de passe', 'error')
+        return resp
+    
 
 @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", ))