"""Support for HomematicIP Cloud switches.""" import logging from homeassistant.components.homematicip_cloud import ( DOMAIN as HMIPC_DOMAIN, HMIPC_HAPID, HomematicipGenericDevice) from homeassistant.components.homematicip_cloud.device import ( ATTR_GROUP_MEMBER_UNREACHABLE) from homeassistant.components.switch import SwitchDevice DEPENDENCIES = ['homematicip_cloud'] _LOGGER = logging.getLogger(__name__) ATTR_POWER_CONSUMPTION = 'power_consumption' ATTR_ENERGIE_COUNTER = 'energie_counter' ATTR_PROFILE_MODE = 'profile_mode' async def async_setup_platform( hass, config, async_add_entities, discovery_info=None): """Set up the HomematicIP Cloud switch devices.""" pass async def async_setup_entry(hass, config_entry, async_add_entities): """Set up the HomematicIP switch from a config entry.""" from homematicip.aio.device import ( AsyncPlugableSwitch, AsyncPlugableSwitchMeasuring, AsyncBrandSwitchMeasuring, AsyncFullFlushSwitchMeasuring, AsyncOpenCollector8Module, ) from homematicip.aio.group import AsyncSwitchingGroup home = hass.data[HMIPC_DOMAIN][config_entry.data[HMIPC_HAPID]].home devices = [] for device in home.devices: if isinstance(device, AsyncBrandSwitchMeasuring): # BrandSwitchMeasuring inherits PlugableSwitchMeasuring # This device is implemented in the light platform and will # not be added in the switch platform pass elif isinstance(device, (AsyncPlugableSwitchMeasuring, AsyncFullFlushSwitchMeasuring)): devices.append(HomematicipSwitchMeasuring(home, device)) elif isinstance(device, AsyncPlugableSwitch): devices.append(HomematicipSwitch(home, device)) elif isinstance(device, AsyncOpenCollector8Module): for channel in range(1, 9): devices.append(HomematicipMultiSwitch(home, device, channel)) for group in home.groups: if isinstance(group, AsyncSwitchingGroup): devices.append( HomematicipGroupSwitch(home, group)) if devices: async_add_entities(devices) class HomematicipSwitch(HomematicipGenericDevice, SwitchDevice): """representation of a HomematicIP Cloud switch device.""" def __init__(self, home, device): """Initialize the switch device.""" super().__init__(home, device) @property def is_on(self): """Return true if device is on.""" return self._device.on async def async_turn_on(self, **kwargs): """Turn the device on.""" await self._device.turn_on() async def async_turn_off(self, **kwargs): """Turn the device off.""" await self._device.turn_off() class HomematicipGroupSwitch(HomematicipGenericDevice, SwitchDevice): """representation of a HomematicIP switching group.""" def __init__(self, home, device, post='Group'): """Initialize switching group.""" device.modelType = 'HmIP-{}'.format(post) super().__init__(home, device, post) @property def is_on(self): """Return true if group is on.""" return self._device.on @property def available(self): """Switch-Group available.""" # A switch-group must be available, and should not be affected by the # individual availability of group members. # This allows switching even when individual group members # are not available. return True @property def device_state_attributes(self): """Return the state attributes of the switch-group.""" attr = {} if self._device.unreach: attr[ATTR_GROUP_MEMBER_UNREACHABLE] = True return attr async def async_turn_on(self, **kwargs): """Turn the group on.""" await self._device.turn_on() async def async_turn_off(self, **kwargs): """Turn the group off.""" await self._device.turn_off() class HomematicipSwitchMeasuring(HomematicipSwitch): """Representation of a HomematicIP measuring switch device.""" @property def current_power_w(self): """Return the current power usage in W.""" return self._device.currentPowerConsumption @property def today_energy_kwh(self): """Return the today total energy usage in kWh.""" if self._device.energyCounter is None: return 0 return round(self._device.energyCounter) class HomematicipMultiSwitch(HomematicipGenericDevice, SwitchDevice): """Representation of a HomematicIP Cloud multi switch device.""" def __init__(self, home, device, channel): """Initialize the multi switch device.""" self.channel = channel super().__init__(home, device, 'Channel{}'.format(channel)) @property def unique_id(self): """Return a unique ID.""" return "{}_{}_{}".format(self.__class__.__name__, self.post, self._device.id) @property def is_on(self): """Return true if device is on.""" return self._device.functionalChannels[self.channel].on async def async_turn_on(self, **kwargs): """Turn the device on.""" await self._device.turn_on(self.channel) async def async_turn_off(self, **kwargs): """Turn the device off.""" await self._device.turn_off(self.channel)