""" Simple platform to locally control Tuya-based cover devices. Sample config yaml: cover: - platform: localtuya #REQUIRED host: 192.168.0.123 #REQUIRED local_key: 1234567891234567 #REQUIRED device_id: 123456789123456789abcd #REQUIRED name: cover_guests #REQUIRED friendly_name: Cover guests #REQUIRED protocol_version: 3.3 #REQUIRED id: 1 #OPTIONAL icon: mdi:blinds #OPTIONAL open_cmd: open #OPTIONAL, default is 'on' close_cmd: close #OPTIONAL, default is 'off' stop_cmd: stop #OPTIONAL, default is 'stop' """ import logging from time import time, sleep import voluptuous as vol from homeassistant.components.cover import ( CoverEntity, DOMAIN, PLATFORM_SCHEMA, SUPPORT_CLOSE, SUPPORT_OPEN, SUPPORT_STOP, SUPPORT_SET_POSITION, ) from homeassistant.const import ( CONF_ID, CONF_FRIENDLY_NAME, ) import homeassistant.helpers.config_validation as cv from . import ( BASE_PLATFORM_SCHEMA, TuyaDevice, LocalTuyaEntity, prepare_setup_entities, import_from_yaml, ) from .const import ( CONF_OPEN_CMD, CONF_CLOSE_CMD, CONF_STOP_CMD, ) from .const import CONF_OPEN_CMD, CONF_CLOSE_CMD, CONF_STOP_CMD _LOGGER = logging.getLogger(__name__) DEFAULT_OPEN_CMD = "on" DEFAULT_CLOSE_CMD = "off" DEFAULT_STOP_CMD = "stop" PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(BASE_PLATFORM_SCHEMA).extend( { vol.Optional(CONF_OPEN_CMD, default=DEFAULT_OPEN_CMD): cv.string, vol.Optional(CONF_CLOSE_CMD, default=DEFAULT_CLOSE_CMD): cv.string, vol.Optional(CONF_STOP_CMD, default=DEFAULT_STOP_CMD): cv.string, } ) def flow_schema(dps): """Return schema used in config flow.""" return { vol.Optional(CONF_OPEN_CMD, default=DEFAULT_OPEN_CMD): str, vol.Optional(CONF_CLOSE_CMD, default=DEFAULT_CLOSE_CMD): str, vol.Optional(CONF_STOP_CMD, default=DEFAULT_STOP_CMD): str, } async def async_setup_entry(hass, config_entry, async_add_entities): """Setup a Tuya cover based on a config entry.""" tuyainterface, entities_to_setup = prepare_setup_entities( config_entry, DOMAIN ) if not entities_to_setup: return covers = [] for device_config in entities_to_setup: covers.append( LocaltuyaCover( TuyaDevice(tuyainterface, config_entry.data[CONF_FRIENDLY_NAME]), config_entry, device_config[CONF_ID], ) ) async_add_entities(covers, True) def setup_platform(hass, config, add_devices, discovery_info=None): """Set up of the Tuya cover.""" return import_from_yaml(hass, config, DOMAIN) class LocaltuyaCover(LocalTuyaEntity, CoverEntity): """Tuya cover devices.""" def __init__( self, device, config_entry, switchid, **kwargs, ): #_LOGGER.info("running def __init__ of LocaltuyaCover(CoverEntity) with self=%s device=%s name=%s friendly_name=%s icon=%s switchid=%s open_cmd=%s close_cmd=%s stop_cmd=%s", self, device, name, friendly_name, icon, switchid, open_cmd, close_cmd, stop_cmd) super().__init__(device, config_entry, switchid, **kwargs) self._state = None self._position = 50 print( "Initialized tuya cover [{}] with switch status [{}] and state [{}]".format( self.name, self._status, self._state ) ) @property def available(self): """Return if device is available or not.""" return self._available @property def supported_features(self): """Flag supported features.""" supported_features = ( SUPPORT_OPEN | SUPPORT_CLOSE | SUPPORT_STOP | SUPPORT_SET_POSITION ) return supported_features @property def current_cover_position(self): # self.update() # state = self._state # _LOGGER.info("curr_pos() : %i", self._position) # print('curr_pos() : state [{}]'.format(state)) return self._position @property def is_opening(self): # self.update() state = self._state # print('is_opening() : state [{}]'.format(state)) if state == "on": return True return False @property def is_closing(self): # self.update() state = self._state # print('is_closing() : state [{}]'.format(state)) if state == "off": return True return False @property def is_closed(self): """Return if the cover is closed or not.""" # _LOGGER.info("running is_closed from cover") # self.update() state = self._state # print('is_closed() : state [{}]'.format(state)) if state == "off": return False if state == "on": return True return None def set_cover_position(self, **kwargs): # _LOGGER.info("running set_cover_position from cover") """Move the cover to a specific position.""" newpos = float(kwargs["position"]) # _LOGGER.info("Set new pos: %f", newpos) currpos = self.current_cover_position posdiff = abs(newpos - currpos) # 25 sec corrisponde alla chiusura/apertura completa mydelay = posdiff / 2.0 if newpos > currpos: # _LOGGER.info("Opening to %f: delay %f", newpos, mydelay ) self.open_cover() else: # _LOGGER.info("Closing to %f: delay %f", newpos, mydelay ) self.close_cover() sleep(mydelay) self.stop_cover() self._position = 50 # newpos def open_cover(self, **kwargs): """Open the cover.""" _LOGGER.debug("Launching command %s to cover ", self._config[CONF_OPEN_CMD]) self._device.set_dps(self._config[CONF_OPEN_CMD], self._dps_id) def close_cover(self, **kwargs): """Close cover.""" _LOGGER.debug("Launching command %s to cover ", self._config[CONF_CLOSE_CMD]) self._device.set_dps(self._config[CONF_CLOSE_CMD], self._dps_id) def stop_cover(self, **kwargs): """Stop the cover.""" _LOGGER.debug("Laudebugching command %s to cover ", self._config[CONF_STOP_CMD]) self._device.set_dps(self._config[CONF_STOP_CMD], self._dps_id) def status_updated(self): """Device status was updated. """ self._state = self.dps(self._dps_id)