mirror of
https://github.com/azaion/detections.git
synced 2026-06-23 07:31:07 +00:00
This commit is contained in:
+13
-1
@@ -31,6 +31,16 @@ def _resolve_disk_path(filename: str, folder: str | None) -> Path | None:
|
||||
return None
|
||||
|
||||
|
||||
def _write_disk_path(filename: str, folder: str | None, data: bytes) -> Path:
|
||||
root = _models_root()
|
||||
safe_filename = Path(filename).name
|
||||
target_dir = root / folder if folder else root
|
||||
target_dir.mkdir(parents=True, exist_ok=True)
|
||||
target = target_dir / safe_filename
|
||||
target.write_bytes(data)
|
||||
return target
|
||||
|
||||
|
||||
def _should_fail_load() -> bool:
|
||||
global _first_fail_remaining
|
||||
if _mode == "error":
|
||||
@@ -73,7 +83,9 @@ def upload(filename):
|
||||
f = request.files.get("data")
|
||||
if not f:
|
||||
return "", 400
|
||||
_uploads[(folder, filename)] = f.read()
|
||||
data = f.read()
|
||||
_uploads[(folder, filename)] = data
|
||||
_write_disk_path(filename, folder, data)
|
||||
_upload_count += 1
|
||||
return "", 200
|
||||
|
||||
|
||||
Reference in New Issue
Block a user