Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -137,6 +137,7 @@ If your device is unlisted but working, please open a pull request to update the
* KP105
* KP115
* KP125
* KP125M
* KP401
* EP10

Expand Down
29 changes: 25 additions & 4 deletions kasa/tapo/tapoplug.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,8 @@

from ..credentials import Credentials
from ..emeterstatus import EmeterStatus
from ..smartdevice import DeviceType
from ..modules import Emeter
from ..smartdevice import DeviceType, requires_update
from .tapodevice import TapoDevice

_LOGGER = logging.getLogger(__name__)
Expand All @@ -24,6 +25,15 @@ def __init__(
) -> None:
super().__init__(host, port=port, credentials=credentials, timeout=timeout)
self._device_type = DeviceType.Plug
self.modules: Dict[str, Any] = {}
self.emeter_type = "emeter"
self.modules["emeter"] = Emeter(self, self.emeter_type)
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think we need to rethink the whole modules system for "tapo" devices separately from the legacy protocol, but I suppose this is fine for now.


@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):
"""Call the device endpoint and update the device data."""
Expand Down Expand Up @@ -52,17 +62,24 @@ def state_information(self) -> Dict[str, Any]:
@property
def emeter_realtime(self) -> EmeterStatus:
"""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
def emeter_today(self) -> Optional[float]:
"""Get the emeter value for today."""
return None
return self._convert_energy_data(self._energy.get("today_energy"), 1 / 1000)

@property
def emeter_this_month(self) -> Optional[float]:
"""Get the emeter value for this month."""
return None
return self._convert_energy_data(self._energy.get("month_energy"), 1 / 1000)

@property
def on_since(self) -> Optional[datetime]:
Expand All @@ -71,3 +88,7 @@ def on_since(self) -> Optional[datetime]:
return None
on_time = cast(float, self._info.get("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