Subclass HTTPReponse in order to ensure the presence of CORS headers

1 files changed, 4 insertions, 8 deletions

diff --git a/routes/media.py b/routes/media.py
index 7986960..125284b 100644
--- a/routes/media.py
+++ b/routes/media.py
@@ -36,10 +36,6 @@ def _delete_file(file):
         pass
     except OSError:
         print("Failed to unlink", file.id, file=sys.stderr)
-    
-@get("/api/media")
-def get_media():
-    notAllowed()
 
 @post("/api/media")
 @jwt_auth_required
@@ -48,7 +44,7 @@ def post_media():
     f = _upload_file(next(request.files.values()))
     session.add(f)
     session.commit()
-    return HTTPResponse(f.json(), status=201) 
+    return ParryHTTPResponse(f.json(), status=201) 
 
 @get("/api/media/<id>")
 def get_media_id(id):
@@ -56,7 +52,7 @@ def get_media_id(id):
     try:
         file = session.query(File).filter_by(id=id).one()
     except db.orm.exc.NoResultFound:
-        raise HTTPResponse(status=404)
+        raise ParryHTTPResponse(status=404)
 
     response.set_header("Content-Type", file.content_type)
     response.set_header("Content-Length", file.length)
@@ -65,8 +61,8 @@ def get_media_id(id):
     if request.method == "GET":
         if file.download_url:
             #return requests.request(request.method, file.download_url, allow_redirects=True)
-            return HTTPResponse(status=302, headers={"Location" : file.download_url})
+            return ParryHTTPResponse(status=302, headers={"Location" : file.download_url})
         else:
             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)
+    raise ParryHTTPResponse(status=404)