Delete the dataset source file too

1 files changed, 9 insertions, 0 deletions

diff --git a/app/admin/routes.py b/app/admin/routes.py
index bed0d18..0317e44 100644
--- a/app/admin/routes.py
+++ b/app/admin/routes.py
@@ -103,6 +103,14 @@ def manage_datasets():
             flash("Dataset doesn't exist.", "error")
             return redirect(url_for("admin.manage_datasets"))
 
+        dataset_dir = os.path.join(
+            current_app.instance_path, current_app.config["DATASET_DIR"]
+        )
+        filename = os.path.join(dataset_dir, dataset.name + ".json")
+        if not os.path.exists(filename):
+            flash("Internal error: dataset file doesn't exist!", "error")
+            return redirect(url_for("admin.manage_datasets"))
+
         tasks = Task.query.filter_by(dataset_id=dataset.id).all()
         for task in tasks:
             for ann in Annotation.query.filter_by(task_id=task.id).all():
@@ -110,6 +118,7 @@ def manage_datasets():
             db.session.delete(task)
         db.session.delete(dataset)
         db.session.commit()
+        os.unlink(filename)
         flash("Dataset deleted successfully.", "success")
 
     overview = []