From ddf75616abfd9dcf7bd8836b6de39c649a32410f Mon Sep 17 00:00:00 2001
From: Fulgen301 <tokmajigeorge@gmail.com>
Date: Fri, 26 Jul 2019 13:50:45 +0200
Subject: Add file deletion

---
 routes/media.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'routes/media.py')

diff --git a/routes/media.py b/routes/media.py
index 37f57c9..34a53ff 100644
--- a/routes/media.py
+++ b/routes/media.py
@@ -14,6 +14,7 @@
 
 from ..helpers import *
 import magic
+from bottle import delete
 from datetime import datetime
 
 def _upload_file(file, entry=None):
@@ -66,3 +67,17 @@ def get_media_id(id):
             return static_file(str(file.id), os.path.join(os.getcwd(), "media"), file.content_type, download=file.name if request.params.download else False)
     
     raise HTTPResponse(status=404)
+
+@delete("/api/media/<id>")
+@jwt_auth_required
+def delete_media_id(id):
+    session = DBSession()
+    try:
+        file = session.query(File).filter_by(id=id).one()
+    except db.orm.exc.NoResultFound:
+        raise HTTPResponse(status=404)
+    
+    session.delete(file)
+    _delete_file(file)
+    session.commit()
+    return HTTPResponse(status=204)
-- 
cgit v1.2.3-54-g00ecf