Kasa KP125M basic emeter support (#560)

* Add KP125M basic emeter support.

* Reduce diff.

* PR Comments
This commit is contained in:
Steven Bytnar 2023-12-03 08:41:46 -06:00 committed by GitHub
parent a6b7d73d79
commit bfd1d6ae0a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 26 additions and 4 deletions

View File

@ -137,6 +137,7 @@ If your device is unlisted but working, please open a pull request to update the
* KP105 * KP105
* KP115 * KP115
* KP125 * KP125
* KP125M
* KP401 * KP401
* EP10 * EP10

View File

@ -5,7 +5,8 @@ from typing import Any, Dict, Optional, cast
from ..credentials import Credentials from ..credentials import Credentials
from ..emeterstatus import EmeterStatus from ..emeterstatus import EmeterStatus
from ..smartdevice import DeviceType from ..modules import Emeter
from ..smartdevice import DeviceType, requires_update
from .tapodevice import TapoDevice from .tapodevice import TapoDevice
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
@ -24,6 +25,15 @@ class TapoPlug(TapoDevice):
) -> None: ) -> None:
super().__init__(host, port=port, credentials=credentials, timeout=timeout) super().__init__(host, port=port, credentials=credentials, timeout=timeout)
self._device_type = DeviceType.Plug self._device_type = DeviceType.Plug
self.modules: Dict[str, Any] = {}
self.emeter_type = "emeter"
self.modules["emeter"] = Emeter(self, self.emeter_type)
@property # type: ignore
@requires_update
def has_emeter(self) -> bool:
"""Return that the plug has an emeter."""
return True
async def update(self, update_children: bool = True): async def update(self, update_children: bool = True):
"""Call the device endpoint and update the device data.""" """Call the device endpoint and update the device data."""
@ -52,17 +62,24 @@ class TapoPlug(TapoDevice):
@property @property
def emeter_realtime(self) -> EmeterStatus: def emeter_realtime(self) -> EmeterStatus:
"""Get the emeter status.""" """Get the emeter status."""
return EmeterStatus({"power_mw": self._energy.get("current_power")}) return EmeterStatus(
{
"power_mw": self._energy.get("current_power"),
"total": self._convert_energy_data(
self._energy.get("today_energy"), 1 / 1000
),
}
)
@property @property
def emeter_today(self) -> Optional[float]: def emeter_today(self) -> Optional[float]:
"""Get the emeter value for today.""" """Get the emeter value for today."""
return None return self._convert_energy_data(self._energy.get("today_energy"), 1 / 1000)
@property @property
def emeter_this_month(self) -> Optional[float]: def emeter_this_month(self) -> Optional[float]:
"""Get the emeter value for this month.""" """Get the emeter value for this month."""
return None return self._convert_energy_data(self._energy.get("month_energy"), 1 / 1000)
@property @property
def on_since(self) -> Optional[datetime]: def on_since(self) -> Optional[datetime]:
@ -71,3 +88,7 @@ class TapoPlug(TapoDevice):
return None return None
on_time = cast(float, self._info.get("on_time")) on_time = cast(float, self._info.get("on_time"))
return datetime.now().replace(microsecond=0) - timedelta(seconds=on_time) return datetime.now().replace(microsecond=0) - timedelta(seconds=on_time)
def _convert_energy_data(self, data, scale) -> Optional[float]:
"""Return adjusted emeter information."""
return data if not data else data * scale