diff --git a/flaskr/auth.py b/flaskr/auth.py index abe769e..acdce91 100644 --- a/flaskr/auth.py +++ b/flaskr/auth.py @@ -1,6 +1,6 @@ # Authorization -import functools,time,os,uuid,cv2 +import functools,time,os,uuid,cv2, shutil from flaskr.db import get_db from flask import( Blueprint, @@ -118,4 +118,35 @@ def load_logged_in_user(): g.user = get_db().execute( 'SELECT * FROM USERS WHERE UserName = ?', (user_id,) ).fetchone() - \ No newline at end of file + + +@bp.route('/deleteAccount') +def deleteAccount(): + if 'username' not in session: + return 'error!', 500 + else: + username = session['username'] + db = get_db() + db.execute( + 'DELETE FROM USERS WHERE UserName =?', + (username,) + ) + db.commit() + db.execute( + 'DELETE FROM IMAGES WHERE User = ?', + (username,) + ) + db.commit() + db.execute( + 'DELETE FROM COMMENTS WHERE User = ?', + (username,) + ) + db.commit() + db.execute( + 'DELETE FROM ImagesLikedByUser WHERE User = ?', + (username,) + ) + db.commit() + shutil.rmtree((current_app.config['USERFILE_DIR'] + '/' + username)) + + return redirect(url_for('auth.logout')) \ No newline at end of file