mirror of
https://github.com/clearml/clearml-server
synced 2025-03-15 18:13:16 +00:00
Fix UPLOAD_FOLDER handling when provided as env var or when fileserver is run by gunicorn
This commit is contained in:
parent
9403942ef7
commit
e1309e30b7
@ -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)
|
||||||
|
Loading…
Reference in New Issue
Block a user