Merge pull request #456 from NatMarchand/feat/improvments

feat: improvments
jabesq-org · Oct 10, 2023 · 85f6330 · 85f6330
2 parents 1e7702c + 49dad0d
commit 85f6330
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/src/pyatmo/modules/legrand.py b/src/pyatmo/modules/legrand.py
@@ -4,6 +4,7 @@
 import logging
 
 from pyatmo.modules.module import (
+    BatteryMixin,
     ContactorMixin,
     Dimmer,
     EnergyMixin,
@@ -24,15 +25,15 @@
 # pylint: disable=R0901
 
 
-class NLG(FirmwareMixin, Module):
+class NLG(FirmwareMixin, OffloadMixin, WifiMixin, Module):
     """Legrand gateway."""
 
 
-class NLT(FirmwareMixin, Module):
+class NLT(FirmwareMixin, BatteryMixin, Module):
     """Legrand global remote control."""
 
 
-class NLP(Switch):
+class NLP(Switch, HistoryMixin, PowerMixin, OffloadMixin, Module):
     """Legrand plug."""
 
 
@@ -104,7 +105,7 @@ class NLPS(FirmwareMixin, PowerMixin, EnergyMixin, Module):
     """Legrand / BTicino smart load shedder."""
 
 
-class NLC(FirmwareMixin, SwitchMixin, Module):
+class NLC(FirmwareMixin, SwitchMixin, HistoryMixin, PowerMixin, OffloadMixin, Module):
     """Legrand / BTicino cable outlet."""
 
 

diff --git a/src/pyatmo/modules/module.py b/src/pyatmo/modules/module.py
@@ -607,11 +607,11 @@ async def async_update_measures(
         if start_time is None:
             start_time = end_time - days * 24 * 60 * 60
 
-        data_point = MeasureType.SUM_ENERGY_ELEC_BASIC.name
+        data_point = MeasureType.SUM_ENERGY_ELEC_BASIC.value
         params = {
             "device_id": self.bridge,
             "module_id": self.entity_id,
-            "scale": interval.name,
+            "scale": interval.value,
             "type": data_point,
             "date_begin": start_time,
             "date_end": end_time,