Improve type hints in dlib_face_detect (#145422)

This commit is contained in:
epenet 2025-05-22 10:37:03 +02:00 committed by GitHub
parent 7893eaa389
commit a7f6a6f22c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -25,37 +25,28 @@ def setup_platform(
discovery_info: DiscoveryInfoType | None = None, discovery_info: DiscoveryInfoType | None = None,
) -> None: ) -> None:
"""Set up the Dlib Face detection platform.""" """Set up the Dlib Face detection platform."""
source: list[dict[str, str]] = config[CONF_SOURCE]
add_entities( add_entities(
DlibFaceDetectEntity(camera[CONF_ENTITY_ID], camera.get(CONF_NAME)) DlibFaceDetectEntity(camera[CONF_ENTITY_ID], camera.get(CONF_NAME))
for camera in config[CONF_SOURCE] for camera in source
) )
class DlibFaceDetectEntity(ImageProcessingFaceEntity): class DlibFaceDetectEntity(ImageProcessingFaceEntity):
"""Dlib Face API entity for identify.""" """Dlib Face API entity for identify."""
def __init__(self, camera_entity, name=None): def __init__(self, camera_entity: str, name: str | None) -> None:
"""Initialize Dlib face entity.""" """Initialize Dlib face entity."""
super().__init__() super().__init__()
self._camera = camera_entity self._attr_camera_entity = camera_entity
if name: if name:
self._name = name self._attr_name = name
else: else:
self._name = f"Dlib Face {split_entity_id(camera_entity)[1]}" self._attr_name = f"Dlib Face {split_entity_id(camera_entity)[1]}"
@property def process_image(self, image: bytes) -> None:
def camera_entity(self):
"""Return camera entity id from process pictures."""
return self._camera
@property
def name(self):
"""Return the name of the entity."""
return self._name
def process_image(self, image):
"""Process image.""" """Process image."""
fak_file = io.BytesIO(image) fak_file = io.BytesIO(image)