Ensure service calls are typed [e-g] (#62912)

Co-authored-by: epenet <epenet@users.noreply.github.com>
This commit is contained in:
epenet 2021-12-28 14:23:01 +01:00 committed by GitHub
parent fb272f58fb
commit 656d383ba6
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
11 changed files with 40 additions and 30 deletions

View file

@ -24,7 +24,7 @@ from homeassistant.const import (
STATE_ALARM_TRIGGERED,
STATE_UNKNOWN,
)
from homeassistant.core import callback
from homeassistant.core import ServiceCall, callback
import homeassistant.helpers.config_validation as cv
from homeassistant.helpers.dispatcher import async_dispatcher_connect
@ -74,10 +74,10 @@ async def async_setup_platform(hass, config, async_add_entities, discovery_info=
async_add_entities(devices)
@callback
def alarm_keypress_handler(service):
def alarm_keypress_handler(service: ServiceCall) -> None:
"""Map services to methods on Alarm."""
entity_ids = service.data.get(ATTR_ENTITY_ID)
keypress = service.data.get(ATTR_KEYPRESS)
entity_ids = service.data[ATTR_ENTITY_ID]
keypress = service.data[ATTR_KEYPRESS]
target_devices = [
device for device in devices if device.entity_id in entity_ids