Add sighthound integration (#28824)
* Add component files * Add test state * Adds person detected event * Update CODEOWNERS * Updates requirements * remove unused datetime * Bump sighthound version * Update CODEOWNERS * Update CODEOWNERS * Create requirements_test_all.txt * Address reviewer comments * Add test for bad_api_key
This commit is contained in:
parent
73a55825af
commit
c71ae090fc
8 changed files with 234 additions and 0 deletions
120
homeassistant/components/sighthound/image_processing.py
Normal file
120
homeassistant/components/sighthound/image_processing.py
Normal file
|
@ -0,0 +1,120 @@
|
|||
"""Person detection using Sighthound cloud service."""
|
||||
import logging
|
||||
|
||||
import simplehound.core as hound
|
||||
import voluptuous as vol
|
||||
|
||||
from homeassistant.components.image_processing import (
|
||||
CONF_ENTITY_ID,
|
||||
CONF_NAME,
|
||||
CONF_SOURCE,
|
||||
PLATFORM_SCHEMA,
|
||||
ImageProcessingEntity,
|
||||
)
|
||||
from homeassistant.const import ATTR_ENTITY_ID, CONF_API_KEY
|
||||
from homeassistant.core import split_entity_id
|
||||
import homeassistant.helpers.config_validation as cv
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
EVENT_PERSON_DETECTED = "sighthound.person_detected"
|
||||
|
||||
ATTR_BOUNDING_BOX = "bounding_box"
|
||||
ATTR_PEOPLE = "people"
|
||||
CONF_ACCOUNT_TYPE = "account_type"
|
||||
DEV = "dev"
|
||||
PROD = "prod"
|
||||
|
||||
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
|
||||
{
|
||||
vol.Required(CONF_API_KEY): cv.string,
|
||||
vol.Optional(CONF_ACCOUNT_TYPE, default=DEV): vol.In([DEV, PROD]),
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
def setup_platform(hass, config, add_entities, discovery_info=None):
|
||||
"""Set up the platform."""
|
||||
# Validate credentials by processing image.
|
||||
api_key = config[CONF_API_KEY]
|
||||
account_type = config[CONF_ACCOUNT_TYPE]
|
||||
api = hound.cloud(api_key, account_type)
|
||||
try:
|
||||
api.detect(b"Test")
|
||||
except hound.SimplehoundException as exc:
|
||||
_LOGGER.error("Sighthound error %s setup aborted", exc)
|
||||
return
|
||||
|
||||
entities = []
|
||||
for camera in config[CONF_SOURCE]:
|
||||
sighthound = SighthoundEntity(
|
||||
api, camera[CONF_ENTITY_ID], camera.get(CONF_NAME)
|
||||
)
|
||||
entities.append(sighthound)
|
||||
add_entities(entities)
|
||||
|
||||
|
||||
class SighthoundEntity(ImageProcessingEntity):
|
||||
"""Create a sighthound entity."""
|
||||
|
||||
def __init__(self, api, camera_entity, name):
|
||||
"""Init."""
|
||||
self._api = api
|
||||
self._camera = camera_entity
|
||||
if name:
|
||||
self._name = name
|
||||
else:
|
||||
camera_name = split_entity_id(camera_entity)[1]
|
||||
self._name = f"sighthound_{camera_name}"
|
||||
self._state = None
|
||||
self._image_width = None
|
||||
self._image_height = None
|
||||
|
||||
def process_image(self, image):
|
||||
"""Process an image."""
|
||||
detections = self._api.detect(image)
|
||||
people = hound.get_people(detections)
|
||||
self._state = len(people)
|
||||
|
||||
metadata = hound.get_metadata(detections)
|
||||
self._image_width = metadata["image_width"]
|
||||
self._image_height = metadata["image_height"]
|
||||
for person in people:
|
||||
self.fire_person_detected_event(person)
|
||||
|
||||
def fire_person_detected_event(self, person):
|
||||
"""Send event with detected total_persons."""
|
||||
self.hass.bus.fire(
|
||||
EVENT_PERSON_DETECTED,
|
||||
{
|
||||
ATTR_ENTITY_ID: self.entity_id,
|
||||
ATTR_BOUNDING_BOX: hound.bbox_to_tf_style(
|
||||
person["boundingBox"], self._image_width, self._image_height
|
||||
),
|
||||
},
|
||||
)
|
||||
|
||||
@property
|
||||
def camera_entity(self):
|
||||
"""Return camera entity id from process pictures."""
|
||||
return self._camera
|
||||
|
||||
@property
|
||||
def name(self):
|
||||
"""Return the name of the sensor."""
|
||||
return self._name
|
||||
|
||||
@property
|
||||
def should_poll(self):
|
||||
"""Return the polling state."""
|
||||
return False
|
||||
|
||||
@property
|
||||
def state(self):
|
||||
"""Return the state of the entity."""
|
||||
return self._state
|
||||
|
||||
@property
|
||||
def unit_of_measurement(self):
|
||||
"""Return the unit of measurement."""
|
||||
return ATTR_PEOPLE
|
Loading…
Add table
Add a link
Reference in a new issue