67 lines
2.2 KiB
Python
67 lines
2.2 KiB
Python
"""Support for Xiaomi curtain."""
|
|
import logging
|
|
|
|
from homeassistant.components.cover import CoverDevice
|
|
from homeassistant.components.xiaomi import (PY_XIAOMI_GATEWAY, XiaomiDevice)
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
ATTR_CURTAIN_LEVEL = 'curtain_level'
|
|
|
|
|
|
def setup_platform(hass, config, add_devices, discovery_info=None):
|
|
"""Perform the setup for Xiaomi devices."""
|
|
devices = []
|
|
for (_, gateway) in hass.data[PY_XIAOMI_GATEWAY].gateways.items():
|
|
for device in gateway.devices['cover']:
|
|
model = device['model']
|
|
if model == 'curtain':
|
|
devices.append(XiaomiGenericCover(device, "Curtain",
|
|
{'status': 'status',
|
|
'pos': 'curtain_level'},
|
|
gateway))
|
|
add_devices(devices)
|
|
|
|
|
|
class XiaomiGenericCover(XiaomiDevice, CoverDevice):
|
|
"""Representation of a XiaomiPlug."""
|
|
|
|
def __init__(self, device, name, data_key, xiaomi_hub):
|
|
"""Initialize the XiaomiPlug."""
|
|
self._data_key = data_key
|
|
self._pos = 0
|
|
XiaomiDevice.__init__(self, device, name, xiaomi_hub)
|
|
|
|
@property
|
|
def current_cover_position(self):
|
|
"""Return the current position of the cover."""
|
|
return self._pos
|
|
|
|
@property
|
|
def is_closed(self):
|
|
"""Return if the cover is closed."""
|
|
return self.current_cover_position < 0
|
|
|
|
def close_cover(self, **kwargs):
|
|
"""Close the cover."""
|
|
self._write_to_hub(self._sid, self._data_key['status'], 'close')
|
|
|
|
def open_cover(self, **kwargs):
|
|
"""Open the cover."""
|
|
self._write_to_hub(self._sid, self._data_key['status'], 'open')
|
|
|
|
def stop_cover(self, **kwargs):
|
|
"""Stop the cover."""
|
|
self._write_to_hub(self._sid, self._data_key['status'], 'stop')
|
|
|
|
def set_cover_position(self, position, **kwargs):
|
|
"""Move the cover to a specific position."""
|
|
self._write_to_hub(self._sid, self._data_key['pos'], str(position))
|
|
|
|
def parse_data(self, data):
|
|
"""Parse data sent by gateway."""
|
|
if ATTR_CURTAIN_LEVEL in data:
|
|
self._pos = int(data[ATTR_CURTAIN_LEVEL])
|
|
return True
|
|
return False
|