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

Xiaomi MiIO Switch: Allow unavailable devices at startup by model setting (#12626)

Browse Source 
* Unavailable state introduced if the device isn't reachable.

* Redundancy removed.

* Pylint fixed.

* Missing space added.

* Pylint fixed.

* Use format instead of concatenation.
This commit is contained in:
Sebastian Muszynski 2018-03-05 08:25:12 +01:00 committed by Teemu R
parent b0e062b2f8
commit e7b84432f9
1 changed files with 59 additions and 43 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

102
homeassistant/components/switch/xiaomi_miio.py
View File

@ -19,10 +19,18 @@ _LOGGER = logging.getLogger(__name__)
DEFAULT_NAME = 'Xiaomi Miio Switch' DEFAULT_NAME = 'Xiaomi Miio Switch'
CONF_MODEL = 'model'
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
vol.Required(CONF_HOST): cv.string, vol.Required(CONF_HOST): cv.string,
vol.Required(CONF_TOKEN): vol.All(cv.string, vol.Length(min=32, max=32)), vol.Required(CONF_TOKEN): vol.All(cv.string, vol.Length(min=32, max=32)),
vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string, vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string,
vol.Optional(CONF_MODEL): vol.In(
['chuangmi.plug.v1',
'qmi.powerstrip.v1',
'zimi.powerstrip.v2',
'chuangmi.plug.m1',
'chuangmi.plug.v2']),
}) })
REQUIREMENTS = ['python-miio==0.3.7'] REQUIREMENTS = ['python-miio==0.3.7']
@ -43,48 +51,53 @@ def async_setup_platform(hass, config, async_add_devices, discovery_info=None):
host = config.get(CONF_HOST) host = config.get(CONF_HOST)
name = config.get(CONF_NAME) name = config.get(CONF_NAME)
token = config.get(CONF_TOKEN) token = config.get(CONF_TOKEN)
model = config.get(CONF_MODEL)
_LOGGER.info("Initializing with host %s (token %s...)", host, token[:5]) _LOGGER.info("Initializing with host %s (token %s...)", host, token[:5])
devices = [] devices = []
try:
plug = Device(host, token)
device_info = plug.info()
_LOGGER.info("%s %s %s initialized",
device_info.model,
device_info.firmware_version,
device_info.hardware_version)
if device_info.model in ['chuangmi.plug.v1']: if model is None:
from miio import PlugV1 try:
plug = PlugV1(host, token) miio_device = Device(host, token)
device_info = miio_device.info()
model = device_info.model
_LOGGER.info("%s %s %s detected",
model,
device_info.firmware_version,
device_info.hardware_version)
except DeviceException:
raise PlatformNotReady
# The device has two switchable channels (mains and a USB port). if model in ['chuangmi.plug.v1']:
# A switch device per channel will be created. from miio import PlugV1
for channel_usb in [True, False]: plug = PlugV1(host, token)
device = ChuangMiPlugV1Switch(
name, plug, device_info, channel_usb)
devices.append(device)
elif device_info.model in ['qmi.powerstrip.v1', # The device has two switchable channels (mains and a USB port).
'zimi.powerstrip.v2']: # A switch device per channel will be created.
from miio import PowerStrip for channel_usb in [True, False]:
plug = PowerStrip(host, token) device = ChuangMiPlugV1Switch(
device = XiaomiPowerStripSwitch(name, plug, device_info) name, plug, model, channel_usb)
devices.append(device) devices.append(device)
elif device_info.model in ['chuangmi.plug.m1',
'chuangmi.plug.v2']: elif model in ['qmi.powerstrip.v1',
from miio import Plug 'zimi.powerstrip.v2']:
plug = Plug(host, token) from miio import PowerStrip
device = XiaomiPlugGenericSwitch(name, plug, device_info) plug = PowerStrip(host, token)
devices.append(device) device = XiaomiPowerStripSwitch(name, plug, model)
else: devices.append(device)
_LOGGER.error( elif model in ['chuangmi.plug.m1',
'Unsupported device found! Please create an issue at ' 'chuangmi.plug.v2']:
'https://github.com/rytilahti/python-miio/issues ' from miio import Plug
'and provide the following data: %s', device_info.model) plug = Plug(host, token)
except DeviceException: device = XiaomiPlugGenericSwitch(name, plug, model)
raise PlatformNotReady devices.append(device)
else:
_LOGGER.error(
'Unsupported device found! Please create an issue at '
'https://github.com/rytilahti/python-miio/issues '
'and provide the following data: %s', model)
return False
async_add_devices(devices, update_before_add=True) async_add_devices(devices, update_before_add=True)
@ -92,17 +105,17 @@ def async_setup_platform(hass, config, async_add_devices, discovery_info=None):
class XiaomiPlugGenericSwitch(SwitchDevice): class XiaomiPlugGenericSwitch(SwitchDevice):
"""Representation of a Xiaomi Plug Generic.""" """Representation of a Xiaomi Plug Generic."""
def __init__(self, name, plug, device_info): def __init__(self, name, plug, model):
"""Initialize the plug switch.""" """Initialize the plug switch."""
self._name = name self._name = name
self._icon = 'mdi:power-socket' self._icon = 'mdi:power-socket'
self._device_info = device_info self._model = model
self._plug = plug self._plug = plug
self._state = None self._state = None
self._state_attrs = { self._state_attrs = {
ATTR_TEMPERATURE: None, ATTR_TEMPERATURE: None,
ATTR_MODEL: self._device_info.model, ATTR_MODEL: self._model,
} }
self._skip_update = False self._skip_update = False
@ -191,20 +204,21 @@ class XiaomiPlugGenericSwitch(SwitchDevice):
}) })
except DeviceException as ex: except DeviceException as ex:
self._state = None
_LOGGER.error("Got exception while fetching the state: %s", ex) _LOGGER.error("Got exception while fetching the state: %s", ex)
class XiaomiPowerStripSwitch(XiaomiPlugGenericSwitch, SwitchDevice): class XiaomiPowerStripSwitch(XiaomiPlugGenericSwitch, SwitchDevice):
"""Representation of a Xiaomi Power Strip.""" """Representation of a Xiaomi Power Strip."""
def __init__(self, name, plug, device_info): def __init__(self, name, plug, model):
"""Initialize the plug switch.""" """Initialize the plug switch."""
XiaomiPlugGenericSwitch.__init__(self, name, plug, device_info) XiaomiPlugGenericSwitch.__init__(self, name, plug, model)
self._state_attrs = { self._state_attrs = {
ATTR_TEMPERATURE: None, ATTR_TEMPERATURE: None,
ATTR_LOAD_POWER: None, ATTR_LOAD_POWER: None,
ATTR_MODEL: self._device_info.model, ATTR_MODEL: self._model,
} }
@asyncio.coroutine @asyncio.coroutine
@ -228,17 +242,18 @@ class XiaomiPowerStripSwitch(XiaomiPlugGenericSwitch, SwitchDevice):
}) })
except DeviceException as ex: except DeviceException as ex:
self._state = None
_LOGGER.error("Got exception while fetching the state: %s", ex) _LOGGER.error("Got exception while fetching the state: %s", ex)
class ChuangMiPlugV1Switch(XiaomiPlugGenericSwitch, SwitchDevice): class ChuangMiPlugV1Switch(XiaomiPlugGenericSwitch, SwitchDevice):
"""Representation of a Chuang Mi Plug V1.""" """Representation of a Chuang Mi Plug V1."""
def __init__(self, name, plug, device_info, channel_usb): def __init__(self, name, plug, model, channel_usb):
"""Initialize the plug switch.""" """Initialize the plug switch."""
name = name + ' USB' if channel_usb else name name = '{} USB'.format(name) if channel_usb else name
XiaomiPlugGenericSwitch.__init__(self, name, plug, device_info) XiaomiPlugGenericSwitch.__init__(self, name, plug, model)
self._channel_usb = channel_usb self._channel_usb = channel_usb
@asyncio.coroutine @asyncio.coroutine
@ -293,4 +308,5 @@ class ChuangMiPlugV1Switch(XiaomiPlugGenericSwitch, SwitchDevice):
}) })
except DeviceException as ex: except DeviceException as ex:
self._state = None
_LOGGER.error("Got exception while fetching the state: %s", ex) _LOGGER.error("Got exception while fetching the state: %s", ex)