Spartan/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-02-20 11:48:40 -06:00
Code Issues Packages Projects Releases Wiki Activity

Sanitize provided name

Browse Source
This commit is contained in:
Nicolas Mowen 2025-02-07 11:09:37 -07:00 committed by Blake Blackshear
parent 060ad34e1d
commit f3485bfc13
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
frigate/api/classification.py
View File

@ -100,7 +100,7 @@ def train_face(request: Request, name: str, body: dict = None):
rand_id = "".join(random.choices(string.ascii_lowercase + string.digits, k=6)) rand_id = "".join(random.choices(string.ascii_lowercase + string.digits, k=6))
new_name = f"{name}-{rand_id}.webp" new_name = f"{name}-{rand_id}.webp"
new_file = os.path.join(FACE_DIR, f"{name}/{new_name}") new_file = sanitize_filename(os.path.join(FACE_DIR, f"{name}/{new_name}"))
shutil.move(training_file, new_file) shutil.move(training_file, new_file)
context: EmbeddingsContext = request.app.embeddings context: EmbeddingsContext = request.app.embeddings