switch.py

Go to the documentation of this file.

"""Support for Powerwall Switches (V2 API only)."""
 
from typing import Any
 
from tesla_powerwall import GridStatus, IslandMode, PowerwallError
 
from homeassistant.components.switch import SwitchDeviceClass, SwitchEntity
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
from .entity import PowerWallEntity
from .models import PowerwallConfigEntry, PowerwallRuntimeData
 
OFF_GRID_STATUSES = {
    GridStatus.TRANSITION_TO_ISLAND,
    GridStatus.ISLANDED,
}
 
 
async def async_setup_entry(
    hass: HomeAssistant,
    entry: PowerwallConfigEntry,
    async_add_entities: AddEntitiesCallback,
) -> None:
    """Set up Powerwall switch platform from Powerwall resources."""
    async_add_entities([PowerwallOffGridEnabledEntity(entry.runtime_data)])
 
 
class PowerwallOffGridEnabledEntity(PowerWallEntity, SwitchEntity):
    """Representation of a Switch entity for Powerwall Off-grid operation."""
 
    _attr_translation_key = "off_grid_operation"
    _attr_entity_category = EntityCategory.CONFIG
    _attr_device_class = SwitchDeviceClass.SWITCH
 
    def __init__(self, powerwall_data: PowerwallRuntimeData) -> None:
        """Initialize powerwall entity and unique id."""
        super().__init__(powerwall_data)
        self._attr_unique_id_attr_unique_id = f"{self.base_unique_id}_off_grid_operation"
 
    @property
    def is_on(self) -> bool:
        """Return true if the powerwall is off-grid."""
        return self.coordinator.data.grid_status in OFF_GRID_STATUSES
 
    async def async_turn_on(self, **kwargs: Any) -> None:
        """Turn off-grid mode on."""
        await self._async_set_island_mode_async_set_island_mode(IslandMode.OFFGRID)
 
    async def async_turn_off(self, **kwargs: Any) -> None:
        """Turn off-grid mode off (return to on-grid usage)."""
        await self._async_set_island_mode_async_set_island_mode(IslandMode.ONGRID)
 
    async def _async_set_island_mode(self, island_mode: IslandMode) -> None:
        """Toggles off-grid mode using the island_mode argument."""
        try:
            await self.power_wallpower_wall.set_island_mode(island_mode)
        except PowerwallError as ex:
            raise HomeAssistantError(
                f"Setting off-grid operation to {island_mode} failed: {ex}"
            ) from ex
 
        self._attr_is_on_attr_is_on = island_mode == IslandMode.OFFGRID
        self.async_write_ha_stateasync_write_ha_state()
 
        await self.coordinator.async_request_refresh()