Fallback to get_current_power if get_energy_usage does not provide current_power (#1186)

This commit is contained in:
Fulch36 2024-10-25 19:24:43 +01:00 committed by GitHub
parent 1e0ca799bc
commit 8b95b7d557
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -28,6 +28,12 @@ class Energy(SmartModule, EnergyInterface):
"""Current power in watts."""
if (power := self.energy.get("current_power")) is not None:
return power / 1_000
# Fallback if get_energy_usage does not provide current_power,
# which can happen on some newer devices (e.g. P304M).
elif (
power := self.data.get("get_current_power").get("current_power")
) is not None:
return power
return None
@property
@ -105,3 +111,8 @@ class Energy(SmartModule, EnergyInterface):
async def get_monthly_stats(self, *, year=None, kwh=True) -> dict:
"""Return monthly stats for the given year."""
raise KasaException("Device does not support periodic statistics")
async def _check_supported(self):
"""Additional check to see if the module is supported by the device."""
# Energy module is not supported on P304M parent device
return "device_on" in self._device.sys_info