root
87 lines
2.3 KiB
Python
87 lines
2.3 KiB
Python
"""Button definitions."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
|
|
from homeassistant.components.button import (
|
|
ButtonDeviceClass,
|
|
ButtonEntityDescription,
|
|
ButtonEntity,
|
|
)
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import (
|
|
EntityCategory,
|
|
)
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from . import NordpoolPlanner, NordpoolPlannerEntity
|
|
from .const import (
|
|
CONF_END_TIME_ENTITY,
|
|
CONF_USED_TIME_RESET_ENTITY,
|
|
DOMAIN,
|
|
)
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
CONF_USED_TIME_RESET_ENTITY_DESCRIPTION = ButtonEntityDescription(
|
|
key=CONF_USED_TIME_RESET_ENTITY,
|
|
device_class=ButtonDeviceClass.RESTART,
|
|
entity_category=EntityCategory.DIAGNOSTIC
|
|
)
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities
|
|
):
|
|
"""Create action button entities for platform."""
|
|
|
|
planner: NordpoolPlanner = hass.data[DOMAIN][config_entry.entry_id]
|
|
entities = []
|
|
|
|
if config_entry.data.get(CONF_END_TIME_ENTITY):
|
|
entities.append(
|
|
NordpoolPlannerButton(
|
|
planner,
|
|
entity_description=CONF_USED_TIME_RESET_ENTITY_DESCRIPTION,
|
|
)
|
|
)
|
|
|
|
async_add_entities(entities)
|
|
return True
|
|
|
|
|
|
class NordpoolPlannerButton(NordpoolPlannerEntity, ButtonEntity):
|
|
"""Button config entity."""
|
|
|
|
def __init__(
|
|
self,
|
|
planner,
|
|
entity_description: ButtonEntityDescription,
|
|
) -> None:
|
|
"""Initialize the entity."""
|
|
super().__init__(planner)
|
|
self.entity_description = entity_description
|
|
self._attr_name = (
|
|
self._planner.name
|
|
+ " "
|
|
+ entity_description.key.replace("_entity", "").replace("_", " ")
|
|
)
|
|
self._attr_unique_id = (
|
|
("nordpool_planner_" + self._attr_name)
|
|
.lower()
|
|
.replace(".", "")
|
|
.replace(" ", "_")
|
|
)
|
|
|
|
async def async_added_to_hass(self) -> None:
|
|
"""Load the last known state when added to hass."""
|
|
await super().async_added_to_hass()
|
|
self._planner.register_input_entity_id(
|
|
self.entity_id, self.entity_description.key
|
|
)
|
|
|
|
def press(self) -> None:
|
|
"""Press the button."""
|
|
self._planner.low_hours = 0
|