ClearML/clearml-server
1
0
Fork 0
mirror of https://github.com/clearml/clearml-server synced 2025-02-12 07:38:28 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Move fileserver to clearml

Browse Source
This commit is contained in:
allegroai 2021-05-03 17:00:38 +03:00
parent 074842a122
commit cb3a7c90a8
2 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
fileserver/fileserver.py
View File

@ -4,6 +4,7 @@ import os
from argparse import ArgumentParser from argparse import ArgumentParser
from pathlib import Path from pathlib import Path
from boltons.iterutils import first
from flask import Flask, request, send_from_directory, safe_join, abort, Response from flask import Flask, request, send_from_directory, safe_join, abort, Response
from flask_compress import Compress from flask_compress import Compress
from flask_cors import CORS from flask_cors import CORS
@ -16,8 +17,9 @@ app = Flask(__name__)
CORS(app, **config.get("fileserver.cors")) CORS(app, **config.get("fileserver.cors"))
Compress(app) Compress(app)
app.config["UPLOAD_FOLDER"] = ( app.config["UPLOAD_FOLDER"] = first(
os.environ.get("TRAINS_UPLOAD_FOLDER") or DEFAULT_UPLOAD_FOLDER (os.environ.get(f"{prefix}_UPLOAD_FOLDER") for prefix in ("CLEARML", "TRAINS")),
default=DEFAULT_UPLOAD_FOLDER,
) )
app.config["SEND_FILE_MAX_AGE_DEFAULT"] = config.get( app.config["SEND_FILE_MAX_AGE_DEFAULT"] = config.get(
"fileserver.download.cache_timeout_sec", 5 * 60 "fileserver.download.cache_timeout_sec", 5 * 60
@ -35,7 +37,7 @@ def upload():
target.parent.mkdir(parents=True, exist_ok=True) target.parent.mkdir(parents=True, exist_ok=True)
file.save(str(target)) file.save(str(target))
results.append(file_path) results.append(file_path)
return (json.dumps(results), 200) return json.dumps(results), 200
@app.route("/<path:path>", methods=["GET"]) @app.route("/<path:path>", methods=["GET"])
@ -57,8 +59,7 @@ def download(path):
@app.route("/<path:path>", methods=["DELETE"]) @app.route("/<path:path>", methods=["DELETE"])
def delete(path): def delete(path):
path = Path(safe_join(app.config["UPLOAD_FOLDER"], path)) path = Path(safe_join(app.config["UPLOAD_FOLDER"], path))
if not path.exists() or path.is_file():
if not (path.exists() and path.is_file()):
abort(Response(f"File {str(path)} not found", 404)) abort(Response(f"File {str(path)} not found", 404))
path.unlink() path.unlink()

1
fileserver/requirements.txt
View File

@ -1,3 +1,4 @@
boltons>=19.1.0
Flask Flask
Flask-Cors>=3.0.5 Flask-Cors>=3.0.5
Flask-Compress>=1.4.0 Flask-Compress>=1.4.0