5b32eea3d0
* Add support for the Quasar bidirectional charger to the Wallbox integration, including ability to control charger while discharging, set a negative charge rate and monitor discharged amount * Make code more generic in order to support other bidirectional models in the future * Updates to files to comply with HA formatting rules * Change const file to fix black check failure * Remove unnecessay loop in number entity
100 lines
3.4 KiB
Python
100 lines
3.4 KiB
Python
"""Home Assistant component for accessing the Wallbox Portal API. The number component allows control of charging current."""
|
|
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
from typing import Optional, cast
|
|
|
|
from homeassistant.components.number import NumberEntity, NumberEntityDescription
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|
|
|
from . import InvalidAuth, WallboxCoordinator, WallboxEntity
|
|
from .const import (
|
|
BIDIRECTIONAL_MODEL_PREFIXES,
|
|
CHARGER_DATA_KEY,
|
|
CHARGER_MAX_AVAILABLE_POWER_KEY,
|
|
CHARGER_MAX_CHARGING_CURRENT_KEY,
|
|
CHARGER_PART_NUMBER_KEY,
|
|
CHARGER_SERIAL_NUMBER_KEY,
|
|
DOMAIN,
|
|
)
|
|
|
|
|
|
@dataclass
|
|
class WallboxNumberEntityDescription(NumberEntityDescription):
|
|
"""Describes Wallbox number entity."""
|
|
|
|
|
|
NUMBER_TYPES: dict[str, WallboxNumberEntityDescription] = {
|
|
CHARGER_MAX_CHARGING_CURRENT_KEY: WallboxNumberEntityDescription(
|
|
key=CHARGER_MAX_CHARGING_CURRENT_KEY,
|
|
name="Max. Charging Current",
|
|
),
|
|
}
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
|
|
) -> None:
|
|
"""Create wallbox number entities in HASS."""
|
|
coordinator: WallboxCoordinator = hass.data[DOMAIN][entry.entry_id]
|
|
# Check if the user is authorized to change current, if so, add number component:
|
|
try:
|
|
await coordinator.async_set_charging_current(
|
|
coordinator.data[CHARGER_MAX_CHARGING_CURRENT_KEY]
|
|
)
|
|
except InvalidAuth:
|
|
return
|
|
|
|
async_add_entities(
|
|
[
|
|
WallboxNumber(coordinator, entry, description)
|
|
for ent in coordinator.data
|
|
if (description := NUMBER_TYPES.get(ent))
|
|
]
|
|
)
|
|
|
|
|
|
class WallboxNumber(WallboxEntity, NumberEntity):
|
|
"""Representation of the Wallbox portal."""
|
|
|
|
entity_description: WallboxNumberEntityDescription
|
|
|
|
def __init__(
|
|
self,
|
|
coordinator: WallboxCoordinator,
|
|
entry: ConfigEntry,
|
|
description: WallboxNumberEntityDescription,
|
|
) -> None:
|
|
"""Initialize a Wallbox number entity."""
|
|
super().__init__(coordinator)
|
|
self.entity_description = description
|
|
self._coordinator = coordinator
|
|
self._attr_name = f"{entry.title} {description.name}"
|
|
self._attr_unique_id = f"{description.key}-{coordinator.data[CHARGER_DATA_KEY][CHARGER_SERIAL_NUMBER_KEY]}"
|
|
self._is_bidirectional = (
|
|
coordinator.data[CHARGER_DATA_KEY][CHARGER_PART_NUMBER_KEY][0:3]
|
|
in BIDIRECTIONAL_MODEL_PREFIXES
|
|
)
|
|
|
|
@property
|
|
def native_max_value(self) -> float:
|
|
"""Return the maximum available current."""
|
|
return cast(float, self._coordinator.data[CHARGER_MAX_AVAILABLE_POWER_KEY])
|
|
|
|
@property
|
|
def native_min_value(self) -> float:
|
|
"""Return the minimum available current based on charger type - some chargers can discharge."""
|
|
return (self.max_value * -1) if self._is_bidirectional else 6
|
|
|
|
@property
|
|
def native_value(self) -> float | None:
|
|
"""Return the value of the entity."""
|
|
return cast(
|
|
Optional[float], self._coordinator.data[CHARGER_MAX_CHARGING_CURRENT_KEY]
|
|
)
|
|
|
|
async def async_set_native_value(self, value: float) -> None:
|
|
"""Set the value of the entity."""
|
|
await self._coordinator.async_set_charging_current(value)
|