mirror of
https://github.com/clearml/clearml-server
synced 2025-02-26 05:59:20 +00:00
Fix fileserver depends on deprecated flask._compat.fspath and safe_join
This commit is contained in:
parent
3f2304549d
commit
8bb26a6b0b
@ -5,10 +5,11 @@ from argparse import ArgumentParser
|
|||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
|
||||||
from boltons.iterutils import first
|
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, abort, Response
|
||||||
from flask._compat import fspath
|
|
||||||
from flask_compress import Compress
|
from flask_compress import Compress
|
||||||
from flask_cors import CORS
|
from flask_cors import CORS
|
||||||
|
from werkzeug.exceptions import NotFound
|
||||||
|
from werkzeug.security import safe_join
|
||||||
|
|
||||||
from config import config
|
from config import config
|
||||||
|
|
||||||
@ -34,7 +35,10 @@ def upload():
|
|||||||
if not filename:
|
if not filename:
|
||||||
continue
|
continue
|
||||||
file_path = filename.lstrip(os.sep)
|
file_path = filename.lstrip(os.sep)
|
||||||
target = Path(safe_join(app.config["UPLOAD_FOLDER"], file_path))
|
safe_path = safe_join(app.config["UPLOAD_FOLDER"], file_path)
|
||||||
|
if safe_path is None:
|
||||||
|
raise NotFound()
|
||||||
|
target = Path(safe_path)
|
||||||
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)
|
||||||
@ -61,8 +65,8 @@ def download(path):
|
|||||||
def delete(path):
|
def delete(path):
|
||||||
real_path = Path(
|
real_path = Path(
|
||||||
safe_join(
|
safe_join(
|
||||||
fspath(app.config["UPLOAD_FOLDER"]),
|
os.fspath(app.config["UPLOAD_FOLDER"]),
|
||||||
fspath(path)
|
os.fspath(path)
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
if not real_path.exists() or not real_path.is_file():
|
if not real_path.exists() or not real_path.is_file():
|
||||||
|
Loading…
Reference in New Issue
Block a user