add current and voltage sensors

bigbadblunt · Oct 22, 2021 · fcc48c4 · fcc48c4
1 parent 7fbbf98
commit fcc48c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/custom_components/lightwave2/sensor.py b/custom_components/lightwave2/sensor.py
@@ -3,7 +3,7 @@
 from homeassistant.components.sensor import  STATE_CLASS_MEASUREMENT, STATE_CLASS_TOTAL_INCREASING, SensorEntity, SensorEntityDescription
 from homeassistant.const import (POWER_WATT, ENERGY_WATT_HOUR, DEVICE_CLASS_POWER, DEVICE_CLASS_ENERGY, 
     DEVICE_CLASS_SIGNAL_STRENGTH, SIGNAL_STRENGTH_DECIBELS_MILLIWATT, PERCENTAGE, DEVICE_CLASS_BATTERY,
-    DEVICE_CLASS_TIMESTAMP)
+    DEVICE_CLASS_TIMESTAMP, ELECTRIC_POTENTIAL_VOLT, ELECTRIC_CURRENT_AMPERE, DEVICE_CLASS_CURRENT, DEVICE_CLASS_VOLTAGE)
 from homeassistant.core import callback
 
 DEPENDENCIES = ['lightwave2']