Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-03-16 02:17:46 +01:00
Code Issues Projects Releases Wiki Activity

Cleanup filename sanitization

Browse Source
This commit is contained in:
Nicolas Mowen
2025-08-16 07:08:04 -06:00
committed by Blake Blackshear
parent ccbaa74a8b
commit 856aab8e6e
2 changed files with 50 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
frigate/api/classification.py
View File

@@ -542,7 +542,7 @@ def delete_classification_dataset_images(
)
for id in list_of_ids:
file_path = os.path.join(folder, id)
file_path = os.path.join(folder, sanitize_filename(id))
if os.path.isfile(file_path):
os.unlink(file_path)
@@ -574,7 +574,9 @@ def categorize_classification_image(request: Request, name: str, body: dict = No
json: dict[str, Any] = body or {}
category = sanitize_filename(json.get("category", ""))
training_file_name = sanitize_filename(json.get("training_file", ""))
training_file = os.path.join(CLIPS_DIR, name, "train", training_file_name)
training_file = os.path.join(
CLIPS_DIR, sanitize_filename(name), "train", training_file_name
)
if training_file_name and not os.path.isfile(training_file):
return JSONResponse(
@@ -588,7 +590,9 @@ def categorize_classification_image(request: Request, name: str, body: dict = No
)
new_name = f"{category}-{datetime.datetime.now().timestamp()}.png"
new_file_folder = os.path.join(CLIPS_DIR, name, "dataset", category)
new_file_folder = os.path.join(
CLIPS_DIR, sanitize_filename(name), "dataset", category
)
if not os.path.exists(new_file_folder):
os.mkdir(new_file_folder)
@@ -627,7 +631,7 @@ def delete_classification_train_images(request: Request, name: str, body: dict =
folder = os.path.join(CLIPS_DIR, sanitize_filename(name), "train")
for id in list_of_ids:
file_path = os.path.join(folder, id)
file_path = os.path.join(folder, sanitize_filename(id))
if os.path.isfile(file_path):
os.unlink(file_path)