jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-27 15:17:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

restore entity for switchbot (#22087)

Browse Source
This commit is contained in:
Daniel Høyer Iversen 2019-03-19 16:16:11 +01:00 committed by Pascal Vizeli
parent 1499485a71
commit 350904870e
1 changed files with 19 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
homeassistant/components/switchbot/switch.py
View File

@ -1,9 +1,4 @@
""" """Support for Switchbot."""
Support for Switchbot.
For more details about this platform, please refer to the documentation at
https://home-assistant.io/components/switch.switchbot
"""
import logging import logging
import voluptuous as vol import voluptuous as vol
@ -11,6 +6,7 @@ import voluptuous as vol
import homeassistant.helpers.config_validation as cv import homeassistant.helpers.config_validation as cv
from homeassistant.components.switch import SwitchDevice, PLATFORM_SCHEMA from homeassistant.components.switch import SwitchDevice, PLATFORM_SCHEMA
from homeassistant.const import CONF_NAME, CONF_MAC from homeassistant.const import CONF_NAME, CONF_MAC
from homeassistant.helpers.restore_state import RestoreEntity
REQUIREMENTS = ['PySwitchbot==0.5'] REQUIREMENTS = ['PySwitchbot==0.5']
@ -18,10 +14,12 @@ _LOGGER = logging.getLogger(__name__)
DEFAULT_NAME = 'Switchbot' DEFAULT_NAME = 'Switchbot'
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
{
vol.Required(CONF_MAC): cv.string, vol.Required(CONF_MAC): cv.string,
vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string, vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string,
}) }
)
def setup_platform(hass, config, add_entities, discovery_info=None): def setup_platform(hass, config, add_entities, discovery_info=None):
@ -31,18 +29,27 @@ def setup_platform(hass, config, add_entities, discovery_info=None):
add_entities([SwitchBot(mac_addr, name)]) add_entities([SwitchBot(mac_addr, name)])
class SwitchBot(SwitchDevice): class SwitchBot(SwitchDevice, RestoreEntity):
"""Representation of a Switchbot.""" """Representation of a Switchbot."""
def __init__(self, mac, name) -> None: def __init__(self, mac, name) -> None:
"""Initialize the Switchbot.""" """Initialize the Switchbot."""
# pylint: disable=import-error, no-member # pylint: disable=import-error, no-member
import switchbot import switchbot
self._state = False
self._state = None
self._name = name self._name = name
self._mac = mac self._mac = mac
self._device = switchbot.Switchbot(mac=mac) self._device = switchbot.Switchbot(mac=mac)
async def async_added_to_hass(self):
"""Run when entity about to be added."""
await super().async_added_to_hass()
state = await self.async_get_last_state()
if not state:
return
self._state = state.state
def turn_on(self, **kwargs) -> None: def turn_on(self, **kwargs) -> None:
"""Turn device on.""" """Turn device on."""
if self._device.turn_on(): if self._device.turn_on():