app.py

import hashlib
import json
import os

from flask import Flask, request, send_file

app = Flask(__name__)
app.config["UPLOAD_FOLDER"] = os.getenv("UPLOAD_FOLDER", "/usr/share/nginx/html")

if not os.path.exists(app.config["UPLOAD_FOLDER"]):
    os.makedirs(app.config["UPLOAD_FOLDER"])


@app.route("/")
def index():
    return "You shouldn't be here."


@app.route("/save", methods=["POST"])
def save():
    if "file" not in request.files:
        return "no file uploaded", 400
    file = request.files["file"]
    if file.filename == "":
        return "no filename found", 400
    filename = hashlib.sha256(file.read()).hexdigest()
    file.seek(0)
    if "filename" in request.form:
        name, ext = json.loads(request.form["filename"])
        filename = "%s.%s.%s" % (name, filename, ext)
    else:
        if "prefix" in request.form:
            filename = "%s_%s" % (request.form["prefix"], filename)
        if "suffix" in request.form:
            filename = "%s_%s" % (filename, request.form["suffix"])
    file.save(os.path.join(app.config["UPLOAD_FOLDER"], filename))
    return filename


# This route should be used for debugging filestore locally.

@app.route("/static/<string:path>")
def serve(path):
    return send_file(os.path.join(app.config["UPLOAD_FOLDER"], path))


if __name__ == "__main__":
    port = int(os.getenv("FILESTORE_PORT", "7910"))
    app.run(use_debugger=True, use_reloader=True, port=port, host="0.0.0.0")