Add file deletion

1 files changed, 15 insertions, 0 deletions

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)