From 325a150cc40e6233286248da6ca92760eb3d3190 Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Thu, 8 May 2025 08:15:32 -0600 Subject: [PATCH] Use datetime as ID --- frigate/api/classification.py | 6 ++---- frigate/data_processing/real_time/face.py | 10 +++------- 2 files changed, 5 insertions(+), 11 deletions(-) diff --git a/frigate/api/classification.py b/frigate/api/classification.py index 8f0fb6462..d0fcf775c 100644 --- a/frigate/api/classification.py +++ b/frigate/api/classification.py @@ -1,10 +1,9 @@ """Object classification APIs.""" +import datetime import logging import os -import random import shutil -import string import cv2 from fastapi import APIRouter, Depends, Request, UploadFile @@ -120,8 +119,7 @@ def train_face(request: Request, name: str, body: dict = None): ) sanitized_name = sanitize_filename(name) - rand_id = "".join(random.choices(string.ascii_lowercase + string.digits, k=6)) - new_name = f"{sanitized_name}-{rand_id}.webp" + new_name = f"{sanitized_name}-{datetime.datetime.now().timestamp()}.webp" new_file_folder = os.path.join(FACE_DIR, f"{sanitized_name}") if not os.path.exists(new_file_folder): diff --git a/frigate/data_processing/real_time/face.py b/frigate/data_processing/real_time/face.py index d91ab9b80..8fe47b23a 100644 --- a/frigate/data_processing/real_time/face.py +++ b/frigate/data_processing/real_time/face.py @@ -5,9 +5,7 @@ import datetime import json import logging import os -import random import shutil -import string from typing import Optional import cv2 @@ -343,11 +341,7 @@ class FaceRealTimeProcessor(RealTimeProcessorApi): return {"success": True, "score": score, "face_name": sub_label} elif topic == EmbeddingsRequestEnum.register_face.value: - rand_id = "".join( - random.choices(string.ascii_lowercase + string.digits, k=6) - ) label = request_data["face_name"] - id = f"{label}-{rand_id}" if request_data.get("cropped"): thumbnail = request_data["image"] @@ -376,7 +370,9 @@ class FaceRealTimeProcessor(RealTimeProcessorApi): # write face to library folder = os.path.join(FACE_DIR, label) - file = os.path.join(folder, f"{id}.webp") + file = os.path.join( + folder, f"{label}_{datetime.datetime.now().timestamp()}.webp" + ) os.makedirs(folder, exist_ok=True) # save face image