diff --git a/ruoyi-fastapi-backend/utils/compreface_util.py b/ruoyi-fastapi-backend/utils/compreface_util.py index 67634f9..0dbd1c2 100644 --- a/ruoyi-fastapi-backend/utils/compreface_util.py +++ b/ruoyi-fastapi-backend/utils/compreface_util.py @@ -99,6 +99,39 @@ class ComprefaceUtil: , headers={'x-api-key': cls.api_key_detection} , files=m) return result + + # 获取所有subjects + @classmethod + async def get_all_subjects(cls) -> dict: + """获取所有已注册的人脸subjects + + Returns: + dict: 包含subjects列表的字典 + """ + client_url: str = SUBJECTS_CRUD_API + url: str = ComprefaceConfig.COMPREFACE_BASE_URL + ':' + ComprefaceConfig.COMPERFACE_BASE_PORT + client_url + result = await cls.send_request("get" + , url + , headers={'x-api-key': cls.api_key}) + return result + + # 删除指定的subject + @classmethod + async def delete_subject(cls, subject: str) -> dict: + """删除指定的subject及其所有人脸数据 + + Args: + subject (str): 要删除的subject名称 + + Returns: + dict: 删除操作的结果 + """ + client_url: str = SUBJECTS_CRUD_API + '/' + subject + url: str = ComprefaceConfig.COMPREFACE_BASE_URL + ':' + ComprefaceConfig.COMPERFACE_BASE_PORT + client_url + result = await cls.send_request("delete" + , url + , headers={'x-api-key': cls.api_key}) + return result async def get_file(image: str = '' or bytes): if not os.path.isfile(image):