From 50622f038d63490277d610a83fe095ee000f2b98 Mon Sep 17 00:00:00 2001 From: Fulgen301 Date: Wed, 22 Aug 2018 20:28:31 +0200 Subject: Initial commit --- routes/media.py | 49 +++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 49 insertions(+) create mode 100644 routes/media.py (limited to 'routes/media.py') diff --git a/routes/media.py b/routes/media.py new file mode 100644 index 0000000..6c3c831 --- /dev/null +++ b/routes/media.py @@ -0,0 +1,49 @@ +# Copyright (c) 2018, George Tokmaji + +# Permission to use, copy, modify, and/or distribute this software for any +# purpose with or without fee is hereby granted, provided that the above +# copyright notice and this permission notice appear in all copies. + +# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES +# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF +# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR +# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES +# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN +# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF +# OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. + +from ..helpers import * +from datetime import datetime + +@route("/api/media") +def get_media(): + notAllowed() + +@route("/api/media", method="POST") +def post_media(): + try: + entry = database["entries"][request.forms["id"]] + except KeyError as e: + raise HTTPResponse("Invalid id", 400) from e + + for f in request.files: + entry["files"].append({}) + + id = ObjectId() + +@route("/api/media/") +def get_media_id(id): + for entry in database["entries"].values(): + for file in entry["files"]: + if file["_id"] == id: + response.set_header("Content-Type", file["content-type"]) + response.set_header("Content-Length", file["length"]) + response.set_header("Date", datetime.fromisoformat(file["uploadDate"]).strftime("%a, %d %b %Y %H:%M:%S GMT")) + response.set_header("Content-Disposition", f"attachment; filename=\"{file['filename']}\"") + if request.method == "GET": + if "downloadURL" in file["metadata"]: + return HTTPResponse(status=302, headers={"Location" : file["metadata"]["downloadURL"]}) + else: + return static_file(file["filename"], root=f"media/{id}") + + raise HTTPResponse(status=404) -- cgit v1.2.3-54-g00ecf