76 lines
2.3 KiB
Python
76 lines
2.3 KiB
Python
|
"""Support for interface with a Gree climate systems."""
|
||
|
|
from typing import Optional
|
||
|
|
|
||
|
|
from homeassistant.components.switch import DEVICE_CLASS_SWITCH, SwitchEntity
|
||
|
|
from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC
|
||
|
|
from homeassistant.helpers.update_coordinator import CoordinatorEntity
|
||
|
|
|
||
|
|
from .const import COORDINATOR, DOMAIN
|
||
|
|
|
||
|
|
|
||
|
|
async def async_setup_entry(hass, config_entry, async_add_entities):
|
||
|
|
"""Set up the Gree HVAC device from a config entry."""
|
||
|
|
async_add_entities(
|
||
|
|
[
|
||
|
|
GreeSwitchEntity(coordinator)
|
||
|
|
for coordinator in hass.data[DOMAIN][COORDINATOR]
|
||
|
|
]
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class GreeSwitchEntity(CoordinatorEntity, SwitchEntity):
|
||
|
|
"""Representation of a Gree HVAC device."""
|
||
|
|
|
||
|
|
def __init__(self, coordinator):
|
||
|
|
"""Initialize the Gree device."""
|
||
|
|
super().__init__(coordinator)
|
||
|
|
self._name = coordinator.device.device_info.name + " Panel Light"
|
||
|
|
self._mac = coordinator.device.device_info.mac
|
||
|
|
|
||
|
|
@property
|
||
|
|
def name(self) -> str:
|
||
|
|
"""Return the name of the device."""
|
||
|
|
return self._name
|
||
|
|
|
||
|
|
@property
|
||
|
|
def unique_id(self) -> str:
|
||
|
|
"""Return a unique id for the device."""
|
||
|
|
return f"{self._mac}-panel-light"
|
||
|
|
|
||
|
|
@property
|
||
|
|
def icon(self) -> Optional[str]:
|
||
|
|
"""Return the icon for the device."""
|
||
|
|
return "mdi:lightbulb"
|
||
|
|
|
||
|
|
@property
|
||
|
|
def device_info(self):
|
||
|
|
"""Return device specific attributes."""
|
||
|
|
return {
|
||
|
|
"name": self._name,
|
||
|
|
"identifiers": {(DOMAIN, self._mac)},
|
||
|
|
"manufacturer": "Gree",
|
||
|
|
"connections": {(CONNECTION_NETWORK_MAC, self._mac)},
|
||
|
|
}
|
||
|
|
|
||
|
|
@property
|
||
|
|
def device_class(self):
|
||
|
|
"""Return the class of this device, from component DEVICE_CLASSES."""
|
||
|
|
return DEVICE_CLASS_SWITCH
|
||
|
|
|
||
|
|
@property
|
||
|
|
def is_on(self) -> bool:
|
||
|
|
"""Return if the light is turned on."""
|
||
|
|
return self.coordinator.device.light
|
||
|
|
|
||
|
|
async def async_turn_on(self, **kwargs):
|
||
|
|
"""Turn the entity on."""
|
||
|
|
self.coordinator.device.light = True
|
||
|
|
await self.coordinator.push_state_update()
|
||
|
|
self.async_write_ha_state()
|
||
|
|
|
||
|
|
async def async_turn_off(self, **kwargs):
|
||
|
|
"""Turn the entity off."""
|
||
|
|
self.coordinator.device.light = False
|
||
|
|
await self.coordinator.push_state_update()
|
||
|
|
self.async_write_ha_state()
|