Update cli energy command to use energy module (#1252)

kasa/cli/usage.py

@@ -9,11 +9,9 @@ import asyncclick as click
 
 from kasa import (
     Device,
+    Module,
 )
-from kasa.iot import (
-    IotDevice,
-)
-from kasa.iot.iotstrip import IotStripPlug
+from kasa.interfaces import Energy
 from kasa.iot.modules import Usage
 
 from .common import (
@@ -49,42 +47,39 @@ async def energy(dev: Device, year, month, erase):
     Daily and monthly data provided in CSV format.
     """
     echo("[bold]== Emeter ==[/bold]")
-    if not dev.has_emeter:
-        error("Device has no emeter")
+    if not (energy := dev.modules.get(Module.Energy)):
+        error("Device has no energy module.")
         return
 
-    if (year or month or erase) and not isinstance(dev, IotDevice):
-        error("Device has no historical statistics")
+    if (year or month or erase) and not energy.supports(
+        Energy.ModuleFeature.PERIODIC_STATS
+    ):
+        error("Device does not support historical statistics")
         return
-    else:
-        dev = cast(IotDevice, dev)
 
     if erase:
         echo("Erasing emeter statistics..")
-        return await dev.erase_emeter_stats()
+        return await energy.erase_stats()
 
     if year:
         echo(f"== For year {year.year} ==")
         echo("Month, usage (kWh)")
-        usage_data = await dev.get_emeter_monthly(year=year.year)
+        usage_data = await energy.get_monthly_stats(year=year.year)
     elif month:
         echo(f"== For month {month.month} of {month.year} ==")
         echo("Day, usage (kWh)")
-        usage_data = await dev.get_emeter_daily(year=month.year, month=month.month)
+        usage_data = await energy.get_daily_stats(year=month.year, month=month.month)
     else:
         # Call with no argument outputs summary data and returns
-        if isinstance(dev, IotStripPlug):
-            emeter_status = await dev.get_emeter_realtime()
-        else:
-            emeter_status = dev.emeter_realtime
+        emeter_status = await energy.get_status()
 
         echo("Current: {} A".format(emeter_status["current"]))
         echo("Voltage: {} V".format(emeter_status["voltage"]))
         echo("Power: {} W".format(emeter_status["power"]))
         echo("Total consumption: {} kWh".format(emeter_status["total"]))
 
-        echo(f"Today: {dev.emeter_today} kWh")
-        echo(f"This month: {dev.emeter_this_month} kWh")
+        echo(f"Today: {energy.consumption_today} kWh")
+        echo(f"This month: {energy.consumption_this_month} kWh")
 
         return emeter_status