From 55b9a5cbfb8c9266324ae0d7cb0bebd5129f35ee Mon Sep 17 00:00:00 2001
From: haotianmingyue <2421912570@qq.com>
Date: Fri, 19 Sep 2025 09:48:05 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AE=9E=E7=8E=B0=E5=90=91=E5=88=86=E7=BB=84?=
 =?UTF-8?q?=E4=B8=AD=E6=B7=BB=E5=8A=A0=E5=8D=95=E4=B8=AA=E4=BA=BA=E8=84=B8?=
 =?UTF-8?q?=E6=9C=8D=E5=8A=A1?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 001测试海康api.py    | 36 ++++++++++++++++++++++++++++++++++--
 config/env.py        |  2 ++
 util/haikang_util.py | 21 ++++++++++++++++++++-
 3 files changed, 56 insertions(+), 3 deletions(-)

diff --git a/001测试海康api.py b/001测试海康api.py
index 4f90cc4..2e2ab1c 100644
--- a/001测试海康api.py
+++ b/001测试海康api.py
@@ -168,7 +168,7 @@ async def get_all_employee_pictures(pageNo=1, pageSize=500):
             # 获取图片
             try:
                 await HaikangUtil.get_person_picture(employee["serverIndexCode"], employee["picUri"], f"./employee/face_images/{employee['personName']}.jpg")
-                await asyncio.sleep(1)
+                await asyncio.sleep(0.5)
             except Exception as e:
                 print(e, employee["visitorId"])
 
@@ -184,6 +184,26 @@ async def face_group_addition_service(name, description):
     with open("face_group_addition_service.json", "w", encoding="utf-8") as f:
         f.write(json.dumps(result[1]))
 
+# 向分组中添加单个人脸
+async def face_single_addition_service(faceGroupIndexCode, image_path, name):
+
+    with open(image_path, 'rb') as f:
+        image_data = f.read()
+
+        encoded_image = base64.b64encode(image_data).decode('utf-8')
+
+    faceInfo = {
+        "name": name,
+    }
+    facePic = {
+        "picBinaryData": encoded_image,
+    }
+
+
+    result = await HaikangUtil.face_single_addition(faceGroupIndexCode, faceInfo, facePic)
+    print(result)
+    with open("face_single_addition_service.json", "w", encoding="utf-8") as f:
+        f.write(json.dumps(result[1]))
 
 
 if __name__ == '__main__':
@@ -206,5 +226,17 @@ if __name__ == '__main__':
     # asyncio.run(query_visitor_record_pictures_service("80890167-b8d4-4e33-af35-94db87c90816"
 	# , "/pic?0de900=71f0ip-4eo881-342*fo1=9=8522*4l5373169147*1t1=0*0ps==418b*=6127*38da35a12-8338cd-9*l168od0f1f1=002"))
     #asyncio.run(door_online_status_service(["d4a610b6554d4544bc35dd3138f128b0"]))
-    asyncio.run(get_all_visitor_pictures())
+    # asyncio.run(get_all_visitor_pictures())
+
+    # 测试获取访客预约记录图片
+    asyncio.run(get_all_employee_pictures())
+
+    # 测试获取员工图片
+    asyncio.run(get_all_employee_pictures())
+
+    # 添加单个人脸分组
+    asyncio.run(face_group_addition_service("访客", "访客分组"))
+
+    # 向分组中添加单个人脸
+    asyncio.run(face_single_addition_service("d4a610b6554d4544bc35dd3138f128b0", "./a7b3c794281902041c9b823df7667ce7.jpg", "测试"))
 
diff --git a/config/env.py b/config/env.py
index efe2d26..e621e84 100644
--- a/config/env.py
+++ b/config/env.py
@@ -40,6 +40,8 @@ class HaiKangSettings:
 
     HAIKANG_FACE_GROUP_ADDITION = '/api/frs/v1/face/group/single/addition'
 
+    HAIKANG_FACE_SIGLE_ADDITION = '/api/frs/v1/face/single/addition'
+
     HAIKANG_FACE_SAVE_PATH = "./face_images"
 
 
diff --git a/util/haikang_util.py b/util/haikang_util.py
index 2163db0..1abc507 100644
--- a/util/haikang_util.py
+++ b/util/haikang_util.py
@@ -558,4 +558,23 @@ class HaikangUtil:
         if back["code"] == "0":
             return [True, back["data"]]
         else:
-            return [False, back["code"], back["msg"]]
\ No newline at end of file
+            return [False, back["code"], back["msg"]]
+
+    # 向分组中单个添加人脸图片
+    @classmethod
+    async def face_single_addition(cls, faceGroupIndexCode: str, faceInfo: dict, facePic: dict):
+        url = f"{HaiKangConfig.HAIKANG_URL}:{HaiKangConfig.HAIKANG_PORT}/artemis{HaiKangConfig.HAIKANG_FACE_SIGLE_ADDITION}"
+        body_dict = {
+            "faceGroupIndexCode": faceGroupIndexCode,
+            "faceInfo": faceInfo,
+            "facePic": facePic,
+        }
+        body_json = json.dumps(body_dict, separators=(",", ":"))
+        headers = cls.build_signed_headers(
+            "POST", url, body_json, HaiKangConfig.HAIKANG_AK, HaiKangConfig.HAIKANG_SK
+        )
+        back = await cls.send_request("POST", url, headers, body_json)
+        if back["code"] == "0":
+            return [True, back["data"]]
+        else:
+            return [False, back["code"], back["msg"]]