feat: add duplicate_check flag when adding person to group

Author: Vitor Pedro

tests/test_face.py

@@ -18,6 +18,7 @@
 
 __image_file = './sample/detection1.jpg'
 __person_id = "person1"
+__person_id_duplicate = "person1_duplicate"
 __template: str
 
 __group_id_async = "test_face_sdk_py_group_async"
@@ -298,6 +299,39 @@ def test_group_add_person(use_async: bool, group_id: str, loop: asyncio.Abstract
         assert False
 
 
+@pytest.mark.parametrize('use_async, group_id', [
+    (True, __group_id_async),
+    (False, __group_id)
+])
+def test_group_add_duplicate_person(use_async: bool, group_id: str, loop: asyncio.AbstractEventLoop):
+    """
+    Test sync and async group add_person request with duplicate check.
+    :param use_async: flag to use the async function
+    :param group_id: group identifier
+    :param loop: event loop for the current test
+    :return:
+        Passes if YoonikApiException is raised.
+    """
+    with pytest.raises(YoonikApiException) as exception:
+        if use_async:
+            loop.run_until_complete(
+                YKF.group.add_person_async(
+                    group_id=group_id,
+                    person_id=__person_id_duplicate,
+                    face_template=__template,
+                    duplicate_check=True
+                )
+            )
+        else:
+            YKF.group.add_person(
+                group_id=group_id,
+                person_id=__person_id_duplicate,
+                face_template=__template,
+                duplicate_check=True
+            )
+
+    assert exception.value.status_code == 409
+
 # @pytest.mark.parametrize('use_async, group_id', [
 #     (True, 'invalid_group1'),
 #     (False, 'invalid_group3')

yk_face/group.py

@@ -91,14 +91,16 @@ async def list_ids_async(group_id: str) -> List[str]:
     return await request_async('GET', url)
 
 
-def add_person(group_id: str, person_id: str, face_template: str):
+def add_person(group_id: str, person_id: str, face_template: str, duplicate_check=False):
     """Add a person to a group.
     :param group_id:
          ID of the group. `group_id` is created in `group.create`.
     :param person_id:
         Person ID.
     :param face_template:
         Biometric template to be associated with the provided `person_id` (obtained from `face.process`).
+    :param duplicate_check:
+         Set True to check if a template already exists with a different person_id.
     :return:
     """
     if group_id is None:
@@ -107,11 +109,12 @@ def add_person(group_id: str, person_id: str, face_template: str):
         raise ValueError("Person ID must be specified.")
 
     url = f'gallery/{group_id}/{person_id}'
-    template_request = Template(template=face_template).model_dump(mode='json')
+    template_request = Template(
+        template=face_template, duplicate_check=duplicate_check).model_dump(mode='json')
     request('POST', url, json=template_request)
 
 
-async def add_person_async(group_id: str, person_id: str, face_template: str):
+async def add_person_async(group_id: str, person_id: str, face_template: str, duplicate_check=False):
     """
     Add a person to a group.
     Performs the request asynchronously.
@@ -121,6 +124,8 @@ async def add_person_async(group_id: str, person_id: str, face_template: str):
         Person ID.
     :param face_template:
         Biometric template to be associated with the provided `person_id` (obtained from `face.process`).
+    :param duplicate_check:
+        Set True to check if a template already exists with a different person_id.
     :return:
     """
     if group_id is None:
@@ -129,7 +134,8 @@ async def add_person_async(group_id: str, person_id: str, face_template: str):
         raise ValueError("Person ID must be specified.")
 
     url = f'gallery/{group_id}/{person_id}'
-    template_request = Template(template=face_template).model_dump(mode='json')
+    template_request = Template(
+        template=face_template, duplicate_check=duplicate_check).model_dump(mode='json')
     await request_async('POST', url, json=template_request)