evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Minor refactor of template alarm (#61862)

Browse source
This commit is contained in:
Erik Montnemery 2021-12-16 16:41:40 +01:00 • committed by GitHub
parent 682f29f131
commit 859bcb6eb4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 38 deletions
Download patch file Download diff file Expand all files Collapse all files

54
homeassistant/components/template/alarm_control_panel.py
View file

@ -97,29 +97,14 @@ async def _async_create_entities(hass, config):
"""Create Template Alarm Control Panels.""" """Create Template Alarm Control Panels."""
alarm_control_panels = [] alarm_control_panels = []
for device, device_config in config[CONF_ALARM_CONTROL_PANELS].items(): for object_id, entity_config in config[CONF_ALARM_CONTROL_PANELS].items():
name = device_config.get(CONF_NAME, device) unique_id = entity_config.get(CONF_UNIQUE_ID)
state_template = device_config.get(CONF_VALUE_TEMPLATE)
disarm_action = device_config.get(CONF_DISARM_ACTION)
arm_away_action = device_config.get(CONF_ARM_AWAY_ACTION)
arm_home_action = device_config.get(CONF_ARM_HOME_ACTION)
arm_night_action = device_config.get(CONF_ARM_NIGHT_ACTION)
code_arm_required = device_config[CONF_CODE_ARM_REQUIRED]
code_format = device_config[CONF_CODE_FORMAT]
unique_id = device_config.get(CONF_UNIQUE_ID)
alarm_control_panels.append( alarm_control_panels.append(
AlarmControlPanelTemplate( AlarmControlPanelTemplate(
hass, hass,
device, object_id,
name, entity_config,
state_template,
disarm_action,
arm_away_action,
arm_home_action,
arm_night_action,
code_arm_required,
code_format,
unique_id, unique_id,
) )
) )
@ -138,37 +123,30 @@ class AlarmControlPanelTemplate(TemplateEntity, AlarmControlPanelEntity):
def __init__( def __init__(
self, self,
hass, hass,
device_id, object_id,
name, config,
state_template,
disarm_action,
arm_away_action,
arm_home_action,
arm_night_action,
code_arm_required,
code_format,
unique_id, unique_id,
): ):
"""Initialize the panel.""" """Initialize the panel."""
super().__init__() super().__init__(config=config)
self.entity_id = async_generate_entity_id( self.entity_id = async_generate_entity_id(
ENTITY_ID_FORMAT, device_id, hass=hass ENTITY_ID_FORMAT, object_id, hass=hass
) )
self._name = name self._name = name = config.get(CONF_NAME, object_id)
self._template = state_template self._template = config.get(CONF_VALUE_TEMPLATE)
self._disarm_script = None self._disarm_script = None
self._code_arm_required = code_arm_required self._code_arm_required = config[CONF_CODE_ARM_REQUIRED]
self._code_format = code_format self._code_format = config[CONF_CODE_FORMAT]
if disarm_action is not None: if (disarm_action := config.get(CONF_DISARM_ACTION)) is not None:
self._disarm_script = Script(hass, disarm_action, name, DOMAIN) self._disarm_script = Script(hass, disarm_action, name, DOMAIN)
self._arm_away_script = None self._arm_away_script = None
if arm_away_action is not None: if (arm_away_action := config.get(CONF_ARM_AWAY_ACTION)) is not None:
self._arm_away_script = Script(hass, arm_away_action, name, DOMAIN) self._arm_away_script = Script(hass, arm_away_action, name, DOMAIN)
self._arm_home_script = None self._arm_home_script = None
if arm_home_action is not None: if (arm_home_action := config.get(CONF_ARM_HOME_ACTION)) is not None:
self._arm_home_script = Script(hass, arm_home_action, name, DOMAIN) self._arm_home_script = Script(hass, arm_home_action, name, DOMAIN)
self._arm_night_script = None self._arm_night_script = None
if arm_night_action is not None: if (arm_night_action := config.get(CONF_ARM_NIGHT_ACTION)) is not None:
self._arm_night_script = Script(hass, arm_night_action, name, DOMAIN) self._arm_night_script = Script(hass, arm_night_action, name, DOMAIN)
self._state = None self._state = None