Migrate foscam to use entity platform entity services (#43775)
This commit is contained in:
parent
39e7b30ab6
commit
935ec59c56
1 changed files with 22 additions and 29 deletions
|
@ -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(
|
||||
|
|
Loading…
Add table
Reference in a new issue