feat($opencv): persistence detected face info

Author: johnnymillergh

home_guardian/__main__.py

@@ -1,8 +1,4 @@
-import random
-
 from home_guardian.configuration.thread_pool_configuration import simulate_multi_thread
-from home_guardian.repository.user_repository import create_user
 
 if __name__ == "__main__":
-    create_user(f"user{random.random()}")
     simulate_multi_thread()

home_guardian/message/email.py

@@ -36,7 +36,7 @@ def build_message(receiver: str) -> MIMEMultipart:
     return message
 
 
-def send_email():
+def send_email() -> None:
     smtp: smtplib.SMTP = smtplib.SMTP(mail_host, 25)
     smtp.connect(mail_host, 25)
     smtp.login(sender, mail_password)

home_guardian/opencv/face_detection.py

@@ -8,6 +8,7 @@
 from home_guardian.common.debounce_throttle import throttle
 from home_guardian.configuration.thread_pool_configuration import executor
 from home_guardian.function_collection import get_data_dir, get_resources_dir
+from home_guardian.repository.detected_face_repository import save
 
 detected_face_dir = f"{get_data_dir()}/detection"
 os.makedirs(detected_face_dir, exist_ok=True)
@@ -37,10 +38,20 @@ def detect_and_take_photo() -> None:
 
 @throttle(2)
 def save_capture(frame) -> None:
+    logger.debug(f"type of frame: {type(frame)}")
     executor.submit(async_save_capture, frame)
 
 
+@logger.catch
 def async_save_capture(frame) -> None:
-    file_name = f"{get_data_dir()}/detection/detected_face_{datetime.datetime.now().__str__().replace(':', '_')}.jpeg"
-    cv2.imwrite(file_name, frame)
-    logger.info("Saved image to: {}", file_name)
+    picture_path = (
+        f"{get_data_dir()}/detection"
+        f"/detected_face_{datetime.datetime.now().__str__().replace(':', '_')}.jpeg"
+    )
+    cv2.imwrite(picture_path, frame)
+    detected_face = save(picture_path)
+    logger.info(
+        "Saved picture, picture_path: {}, detected_face: {}",
+        picture_path,
+        detected_face,
+    )

home_guardian/repository/detected_face_repository.py

@@ -0,0 +1,18 @@
+from loguru import logger
+
+from home_guardian.repository.model.detected_face import DetectedFace
+
+
+@logger.catch
+def save(picture_path: str, detected_username: str = None) -> DetectedFace:
+    """
+    Save a new detected face.
+    :param picture_path: the picture path
+    :param detected_username: the detected username
+    :return: a DetectedFace object
+    """
+    detected_face: DetectedFace = DetectedFace(
+        picture_path=picture_path, detected_username=detected_username
+    )
+    detected_face.save()
+    return detected_face

home_guardian/repository/model/base/model.py

@@ -18,3 +18,4 @@ class BaseModel(Model):
 
     class Meta:
         database = db
+        legacy_table_names = False

home_guardian/repository/model/detected_face.py

@@ -0,0 +1,18 @@
+from datetime import datetime
+
+from peewee import CharField, DateTimeField
+
+from home_guardian.repository.model.base.model import BaseModel, db
+
+
+class DetectedFace(BaseModel):
+    """
+    DetectedFace model
+    """
+
+    picture_path = CharField(max_length=500)
+    detected_username = CharField(max_length=50, null=True)
+    created_time = DateTimeField(default=datetime.now)
+
+
+db.create_tables([DetectedFace])