From a877dcd617af40fa259fc7b90065c96e217f9c84 Mon Sep 17 00:00:00 2001 From: crunk Date: Sat, 19 Mar 2022 20:16:00 +0100 Subject: [PATCH] delete old uploaded files --- verse/editor.py | 8 ++++++-- verse/start.py | 3 ++- 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/verse/editor.py b/verse/editor.py index cc7ca30..e9d4e8a 100644 --- a/verse/editor.py +++ b/verse/editor.py @@ -1,6 +1,8 @@ import os -from flask import render_template import bleach +import shutil +from flask import render_template + from werkzeug.utils import secure_filename from sqlalchemy.exc import ( DataError, @@ -38,9 +40,11 @@ def Editor(): def ValidateEditCssForm(editorform, current_distribusi): newcssfolder = os.path.join("themes/userthemes", current_distribusi) + shutil.rmtree(newcssfolder) + publicfolder = os.path.join("themes/publicthemes", current_distribusi) + shutil.rmtree(publicfolder) if editorform.public.data: MakePublicTheme(editorform, current_distribusi) - publicfolder = os.path.join("themes/publicthemes", current_distribusi) if editorform.cssfile.data: SaveUploadCssFile(editorform, publicfolder) else: diff --git a/verse/start.py b/verse/start.py index 1fee0c3..923785c 100644 --- a/verse/start.py +++ b/verse/start.py @@ -146,7 +146,8 @@ def distribusi(): distribusify(args, userfolder) distribusi.visible = True user.currentdistribusi = None - os.remove(cssfile) + if os.path.exists(cssfile): + os.remove(cssfile) db.session.commit() return redirect(url_for("index")) template = render_template(