ClearML/clearml-server
1
0
Fork 0
mirror of https://github.com/clearml/clearml-server synced 2025-03-15 18:13:16 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix UPLOAD_FOLDER handling when provided as env var or when fileserver is run by gunicorn

Browse Source
This commit is contained in:
allegroai 2020-06-01 13:05:45 +03:00
parent 9403942ef7
commit e1309e30b7
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
fileserver/fileserver.py
View File

@ -10,12 +10,13 @@ from flask_cors import CORS
from config import config from config import config
DEFAULT_UPLOAD_FOLDER = "/mnt/fileserver"
app = Flask(__name__) app = Flask(__name__)
CORS(app, **config.get("fileserver.cors")) CORS(app, **config.get("fileserver.cors"))
Compress(app) Compress(app)
if os.environ.get("TRAINS_UPLOAD_FOLDER"): app.config["UPLOAD_FOLDER"] = os.environ.get("TRAINS_UPLOAD_FOLDER") or DEFAULT_UPLOAD_FOLDER
app.config["UPLOAD_FOLDER"] = os.environ.get("TRAINS_UPLOAD_FOLDER")
@app.route("/", methods=["POST"]) @app.route("/", methods=["POST"])
@ -57,11 +58,12 @@ def main():
parser.add_argument( parser.add_argument(
"--upload-folder", "--upload-folder",
"-u", "-u",
default="/mnt/fileserver", default=DEFAULT_UPLOAD_FOLDER,
help="Upload folder (default %(default)s)", help="Upload folder (default %(default)s)",
) )
args = parser.parse_args() args = parser.parse_args()
if app.config.get("UPLOAD_FOLDER") is None:
app.config["UPLOAD_FOLDER"] = args.upload_folder app.config["UPLOAD_FOLDER"] = args.upload_folder
app.run(debug=args.debug, host=args.ip, port=args.port, threaded=True) app.run(debug=args.debug, host=args.ip, port=args.port, threaded=True)