mirror of
https://github.com/home-assistant/core.git
synced 2025-07-11 15:27:08 +00:00
Improve type hints in dlib_face_detect (#145422)
This commit is contained in:
parent
7893eaa389
commit
a7f6a6f22c
@ -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)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user