Full delete removes image from disk

jsbroks · Feb 14, 2019 · 06c8c8d · 06c8c8d
1 parent fa1ea6a
commit 06c8c8d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/app/api/undo.py b/app/api/undo.py
@@ -1,6 +1,7 @@
 from flask_restplus import Namespace, Resource, reqparse
 from flask_login import login_required, current_user
 
+import os
 import datetime
 from ..models import *
 
@@ -100,6 +101,9 @@ def delete(self):
         if model_object is None:
             return {"message": "Invalid id"}, 400
 
+        if isinstance(model_object, ImageModel):
+            os.remove(model_object.path)
+
         model_object.delete()
 
         return {"success": True}