-
-
Notifications
You must be signed in to change notification settings - Fork 13
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add target time and percentage inputs (#25)
* Moved max charge resume code into common api_client * Refactor entity loading * Switch housekeeping * Added target time and percentage inputs
- Loading branch information
Showing
6 changed files
with
197 additions
and
60 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,74 @@ | ||
| from __future__ import annotations | ||
| import asyncio | ||
| from homeassistant.components.number import NumberEntity, NumberDeviceClass | ||
| from homeassistant.helpers.entity import generate_entity_id | ||
| from homeassistant.core import callback, HomeAssistant | ||
| from .const import DOMAIN, DATA_CLIENT, DATA_COORDINATORS, COORDINATOR_CHARGESESSIONS, COORDINATOR_ACCOUNTINFO | ||
|
|
||
|
|
||
| async def async_setup_entry( | ||
| hass: HomeAssistant, | ||
| config_entry: config_entries.ConfigEntry, | ||
| async_add_entities | ||
| ): | ||
| """Setup switches and configure coordinator.""" | ||
| coordinators = hass.data[DOMAIN][DATA_COORDINATORS] | ||
|
|
||
| coordinator = coordinators[COORDINATOR_CHARGESESSIONS] | ||
| client = hass.data[DOMAIN][DATA_CLIENT] | ||
|
|
||
| numbers = [TargetPercentNumber(coordinator, hass, client)] | ||
|
|
||
| async_add_entities(numbers, update_before_add=True) | ||
|
|
||
|
|
||
| class TargetPercentNumber(NumberEntity): | ||
| """Target percentage sensor.""" | ||
| _attr_name = "Target Percentage" | ||
| _attr_device_class = NumberDeviceClass.BATTERY | ||
| _attr_suggested_display_precision = 0 | ||
|
|
||
| def __init__(self, coordinator, hass: HomeAssistant, client): | ||
| self.coordinator = coordinator | ||
|
|
||
| self._client = client | ||
|
|
||
| self._state = 0 | ||
| self._last_updated = None | ||
| self._attributes = {} | ||
|
|
||
| self.entity_id = generate_entity_id( | ||
| "number.{}", "ohme_target_percent", hass=hass) | ||
|
|
||
| self._attr_device_info = client.get_device_info() | ||
|
|
||
| @property | ||
| def unique_id(self): | ||
| """The unique ID of the switch.""" | ||
| return self._client.get_unique_id("target_percent") | ||
|
|
||
| async def async_set_native_value(self, value: float) -> None: | ||
| """Update the current value.""" | ||
| await self._client.async_apply_charge_rule(target_percent=int(value)) | ||
|
|
||
| await asyncio.sleep(1) | ||
| await self.coordinator.async_refresh() | ||
|
|
||
| @property | ||
| def icon(self): | ||
| """Icon of the sensor.""" | ||
| return "mdi:battery-heart" | ||
|
|
||
| @property | ||
| def native_value(self): | ||
| """Get value from data returned from API by coordinator""" | ||
| if self.coordinator.data and self.coordinator.data['appliedRule']: | ||
| target = round( | ||
| self.coordinator.data['appliedRule']['targetPercent']) | ||
|
|
||
| if target == 0: | ||
| return self._state | ||
|
|
||
| self._state = target | ||
| return self._state | ||
| return None |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,74 @@ | ||
| from __future__ import annotations | ||
| import asyncio | ||
| import logging | ||
| from homeassistant.components.time import TimeEntity | ||
| from homeassistant.helpers.entity import generate_entity_id | ||
| from homeassistant.core import callback, HomeAssistant | ||
| from .const import DOMAIN, DATA_CLIENT, DATA_COORDINATORS, COORDINATOR_CHARGESESSIONS, COORDINATOR_ACCOUNTINFO | ||
| from datetime import time as dt_time | ||
|
|
||
| _LOGGER = logging.getLogger(__name__) | ||
|
|
||
|
|
||
| async def async_setup_entry( | ||
| hass: HomeAssistant, | ||
| config_entry: config_entries.ConfigEntry, | ||
| async_add_entities | ||
| ): | ||
| """Setup switches and configure coordinator.""" | ||
| coordinators = hass.data[DOMAIN][DATA_COORDINATORS] | ||
|
|
||
| coordinator = coordinators[COORDINATOR_CHARGESESSIONS] | ||
| client = hass.data[DOMAIN][DATA_CLIENT] | ||
|
|
||
| numbers = [TargetTime(coordinator, hass, client)] | ||
|
|
||
| async_add_entities(numbers, update_before_add=True) | ||
|
|
||
|
|
||
| class TargetTime(TimeEntity): | ||
| """Target time sensor.""" | ||
| _attr_name = "Target Time" | ||
|
|
||
| def __init__(self, coordinator, hass: HomeAssistant, client): | ||
| self.coordinator = coordinator | ||
|
|
||
| self._client = client | ||
|
|
||
| self._state = 0 | ||
| self._last_updated = None | ||
| self._attributes = {} | ||
|
|
||
| self.entity_id = generate_entity_id( | ||
| "number.{}", "ohme_target_time", hass=hass) | ||
|
|
||
| self._attr_device_info = client.get_device_info() | ||
|
|
||
| @property | ||
| def unique_id(self): | ||
| """The unique ID of the switch.""" | ||
| return self._client.get_unique_id("target_time") | ||
|
|
||
| async def async_set_value(self, value: dt_time) -> None: | ||
| """Update the current value.""" | ||
| await self._client.async_apply_charge_rule(target_time=(int(value.hour), int(value.minute))) | ||
|
|
||
| await asyncio.sleep(1) | ||
| await self.coordinator.async_refresh() | ||
|
|
||
| @property | ||
| def icon(self): | ||
| """Icon of the sensor.""" | ||
| return "mdi:alarm-check" | ||
|
|
||
| @property | ||
| def native_value(self): | ||
| """Get value from data returned from API by coordinator""" | ||
| if self.coordinator.data and self.coordinator.data['appliedRule']: | ||
| target = self.coordinator.data['appliedRule']['targetTime'] | ||
| return dt_time( | ||
| hour=target // 3600, | ||
| minute=(target % 3600) // 60, | ||
| second=0 | ||
| ) | ||
| return None |