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

Added ENTITY_ID_FORMAT import and set entity_id in __init__ (#13642)

Browse source
This commit is contained in:
Timmo 2018-04-06 09:59:09 +01:00 • committed by Daniel Høyer Iversen
parent 703eea0c93
commit 1d7ecc22f9
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
homeassistant/components/switch/broadlink.py
View file

@ -14,7 +14,7 @@ import socket
import voluptuous as vol
from homeassistant.components.switch import (
DOMAIN, PLATFORM_SCHEMA, SwitchDevice)
DOMAIN, PLATFORM_SCHEMA, SwitchDevice, ENTITY_ID_FORMAT)
from homeassistant.const import (
CONF_COMMAND_OFF, CONF_COMMAND_ON, CONF_FRIENDLY_NAME, CONF_HOST, CONF_MAC,
CONF_SWITCHES, CONF_TIMEOUT, CONF_TYPE)
@ -150,6 +150,7 @@ def setup_platform(hass, config, add_devices, discovery_info=None):
for object_id, device_config in devices.items():
switches.append(
BroadlinkRMSwitch(
object_id,
device_config.get(CONF_FRIENDLY_NAME, object_id),
broadlink_device,
device_config.get(CONF_COMMAND_ON),
@ -184,8 +185,9 @@ def setup_platform(hass, config, add_devices, discovery_info=None):
class BroadlinkRMSwitch(SwitchDevice):
"""Representation of an Broadlink switch."""
def __init__(self, friendly_name, device, command_on, command_off):
def __init__(self, name, friendly_name, device, command_on, command_off):
"""Initialize the switch."""
self.entity_id = ENTITY_ID_FORMAT.format(name)
self._name = friendly_name
self._state = False
self._command_on = b64decode(command_on) if command_on else None