From 4595117d912513df8f27f570b270b828b735bc63 Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Wed, 26 Feb 2020 17:26:46 +0200
Subject: [PATCH] Support setting fileserver upload folder using an environment
 variable

---
 fileserver/fileserver.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/fileserver/fileserver.py b/fileserver/fileserver.py
index 387c385..ec93370 100644
--- a/fileserver/fileserver.py
+++ b/fileserver/fileserver.py
@@ -14,6 +14,9 @@ app = Flask(__name__)
 CORS(app, **config.get("fileserver.cors"))
 Compress(app)
 
+if os.environ.get("TRAINS_UPLOAD_FOLDER"):
+    app.config["UPLOAD_FOLDER"] = os.environ.get("TRAINS_UPLOAD_FOLDER")
+
 
 @app.route("/", methods=["POST"])
 def upload():