diff --git a/homeassistant/components/foscam/camera.py b/homeassistant/components/foscam/camera.py index 1c4c6bb9c8c..bc28e160b25 100644 --- a/homeassistant/components/foscam/camera.py +++ b/homeassistant/components/foscam/camera.py @@ -13,14 +13,9 @@ from homeassistant.const import ( CONF_PORT, CONF_USERNAME, ) -from homeassistant.helpers import config_validation as cv -from homeassistant.helpers.service import async_extract_entity_ids +from homeassistant.helpers import config_validation as cv, entity_platform -from .const import ( - DATA as FOSCAM_DATA, - DOMAIN as FOSCAM_DOMAIN, - ENTITIES as FOSCAM_ENTITIES, -) +from .const import DATA as FOSCAM_DATA, ENTITIES as FOSCAM_ENTITIES _LOGGER = logging.getLogger(__name__) @@ -90,28 +85,26 @@ SERVICE_PTZ_SCHEMA = vol.Schema( async def async_setup_platform(hass, config, async_add_entities, discovery_info=None): """Set up a Foscam IP Camera.""" - - async def async_handle_ptz(service): - """Handle PTZ service call.""" - movement = service.data[ATTR_MOVEMENT] - travel_time = service.data[ATTR_TRAVELTIME] - entity_ids = await async_extract_entity_ids(hass, service) - - if not entity_ids: - return - - _LOGGER.debug("Moving '%s' camera(s): %s", movement, entity_ids) - - all_cameras = hass.data[FOSCAM_DATA][FOSCAM_ENTITIES] - target_cameras = [ - camera for camera in all_cameras if camera.entity_id in entity_ids - ] - - for camera in target_cameras: - await camera.async_perform_ptz(movement, travel_time) - - hass.services.async_register( - FOSCAM_DOMAIN, SERVICE_PTZ, async_handle_ptz, schema=SERVICE_PTZ_SCHEMA + platform = entity_platform.current_platform.get() + assert platform is not None + platform.async_register_entity_service( + "ptz", + { + vol.Required(ATTR_MOVEMENT): vol.In( + [ + DIR_UP, + DIR_DOWN, + DIR_LEFT, + DIR_RIGHT, + DIR_TOPLEFT, + DIR_TOPRIGHT, + DIR_BOTTOMLEFT, + DIR_BOTTOMRIGHT, + ] + ), + vol.Optional(ATTR_TRAVELTIME, default=DEFAULT_TRAVELTIME): cv.small_float, + }, + "async_perform_ptz", ) camera = FoscamCamera(