Amélioration des post-it Ajout supp son compte

views/profil.py

@@ -22,10 +22,12 @@ DOSSIER_PERSO = app.config['DOSSIER_APP']
 extensionimg = app.config['EXT_IMG']
 
 DATABASE = app.config['DATABASE']
+DATAS_USER = app.config['DOSSIER_APP']
 MAIL_SERVER = app.config['MAIL_SERVER']
 XMPP_SERVER = app.config['XMPP_SERVER']
 SETUID = app.config['SETUID']
 BASE_URL = app.config['BASE_URL']
+BACKUP_TIME = app.config['BACKUP_TIME']
 
 ##################################################################################################
 
@@ -312,3 +314,61 @@ def generate_token():
       return redirect(BASE_URL+'invitation/')
    else:
       return redirect(BASE_URL, code=401)
+
+
+@profil.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,))
+                        cursor.execute("""DELETE FROM posts WHERE author=?""", (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
+