switch.py

Go to the documentation of this file.

"""Charge and Climate Control Support for the Nissan Leaf."""
 
from __future__ import annotations
 
import logging
from typing import Any
 
from homeassistant.components.switch import SwitchEntity
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
 
from . import LeafDataStore
from .const import DATA_CLIMATE, DATA_LEAF
from .entity import LeafEntity
 
_LOGGER = logging.getLogger(__name__)
 
 
def setup_platform(
    hass: HomeAssistant,
    config: ConfigType,
    add_entities: AddEntitiesCallback,
    discovery_info: DiscoveryInfoType | None = None,
) -> None:
    """Nissan Leaf switch platform setup."""
    if discovery_info is None:
        return
 
    entities: list[LeafEntity] = []
    for vin, datastore in hass.data[DATA_LEAF].items():
        _LOGGER.debug("Adding switch for vin=%s", vin)
        entities.append(LeafClimateSwitch(datastore))
 
    add_entities(entities, True)
 
 
class LeafClimateSwitch(LeafEntity, SwitchEntity):
    """Nissan Leaf Climate Control switch."""
 
    def __init__(self, car: LeafDataStore) -> None:
        """Set up climate control switch."""
        super().__init__(car)
        self._attr_unique_id_attr_unique_id = f"{self.car.leaf.vin.lower()}_climatecontrol"
 
    @property
    def name(self) -> str:
        """Switch name."""
        return f"{self.car.leaf.nickname} Climate Control"
 
    def log_registration(self) -> None:
        """Log registration."""
        _LOGGER.debug(
            "Registered LeafClimateSwitch integration with Home Assistant for VIN %s",
            self.carcar.leaf.vin,
        )
 
    @property
    def extra_state_attributes(self) -> dict[str, Any]:
        """Return climate control attributes."""
        attrs = super().extra_state_attributes
        attrs["updated_on"] = self.carcar.last_climate_response
        return attrs
 
    @property
    def is_on(self) -> bool:
        """Return true if climate control is on."""
        return bool(self.carcar.data[DATA_CLIMATE])
 
    async def async_turn_on(self, **kwargs: Any) -> None:
        """Turn on climate control."""
        if await self.carcar.async_set_climate(True):
            self.carcar.data[DATA_CLIMATE] = True
 
    async def async_turn_off(self, **kwargs: Any) -> None:
        """Turn off climate control."""
        if await self.carcar.async_set_climate(False):
            self.carcar.data[DATA_CLIMATE] = False