50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
"""Switches: флаги-свойства (economy/powerful/min_heat/...)."""
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from homeassistant.components.switch import SwitchEntity
|
|
|
|
from pyfglair import Prop
|
|
from pyfglair.templates import prop_info
|
|
|
|
from .const import SWITCH_PROPS
|
|
from .coordinator import FglairCoordinator, FglairRuntime
|
|
from .entity import FglairEntity
|
|
|
|
|
|
class FglairSwitch(FglairEntity, SwitchEntity):
|
|
def __init__(
|
|
self,
|
|
coordinator: FglairCoordinator,
|
|
prop: Prop,
|
|
key: str,
|
|
) -> None:
|
|
super().__init__(coordinator, prop.name.lower(), key)
|
|
self._prop = prop
|
|
|
|
@property
|
|
def is_on(self) -> bool:
|
|
return self.raw_value(self._prop) == 1
|
|
|
|
async def async_turn_on(self, **kwargs: Any) -> None:
|
|
await self.coordinator.async_write([(self._prop, 1)])
|
|
|
|
async def async_turn_off(self, **kwargs: Any) -> None:
|
|
await self.coordinator.async_write([(self._prop, 0)])
|
|
|
|
|
|
async def async_setup_entry(hass, entry, async_add_entities) -> None:
|
|
runtime: FglairRuntime = entry.runtime_data
|
|
coordinator = runtime.coordinator
|
|
template = runtime.client.template
|
|
|
|
entities = []
|
|
for prop, key in SWITCH_PROPS.items():
|
|
if prop_info(template, prop) is None:
|
|
continue
|
|
if not coordinator.features.enabled(key):
|
|
continue
|
|
entities.append(FglairSwitch(coordinator, prop, key))
|
|
async_add_entities(entities)
|