Apply suggestions from code review

openWB · Oct 15, 2024 · f847603 · f847603
1 parent 21ffe4e
commit f847603
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/packages/modules/devices/growatt/growatt/bat.py b/packages/modules/devices/growatt/growatt/bat.py
@@ -2,6 +2,7 @@
 from typing import Dict, Union
 
 from dataclass_utils import dataclass_from_dict
+from modules.common.abstract_device import AbstractBat
 from modules.common.component_state import BatState
 from modules.common.component_type import ComponentDescriptor
 from modules.common.fault_state import ComponentInfo, FaultState
@@ -11,7 +12,7 @@
 from modules.devices.growatt.growatt.version import GrowattVersion
 
 
-class GrowattBat:
+class GrowattBat(AbstractBat):
     def __init__(self,
                  component_config: Union[Dict, GrowattBatSetup],
                  modbus_id: int,

diff --git a/packages/modules/devices/growatt/growatt/counter.py b/packages/modules/devices/growatt/growatt/counter.py
@@ -2,6 +2,7 @@
 from typing import Dict, Union
 
 from dataclass_utils import dataclass_from_dict
+from modules.common.abstract_device import AbstractCounter
 from modules.common.component_state import CounterState
 from modules.common.component_type import ComponentDescriptor
 from modules.common.fault_state import ComponentInfo, FaultState
@@ -11,7 +12,7 @@
 from modules.devices.growatt.growatt.version import GrowattVersion
 
 
-class GrowattCounter:
+class GrowattCounter(AbstractCounter):
     def __init__(self,
                  component_config: Union[Dict, GrowattCounterSetup],
                  modbus_id: int,

diff --git a/packages/modules/devices/growatt/growatt/inverter.py b/packages/modules/devices/growatt/growatt/inverter.py
@@ -2,6 +2,7 @@
 from typing import Dict, Union
 
 from dataclass_utils import dataclass_from_dict
+from modules.common.abstract_device import AbstractInverter
 from modules.common.component_state import InverterState
 from modules.common.component_type import ComponentDescriptor
 from modules.common.fault_state import ComponentInfo, FaultState
@@ -11,7 +12,7 @@
 from modules.devices.growatt.growatt.version import GrowattVersion
 
 
-class GrowattInverter:
+class GrowattInverter(AbstractInverter):
     def __init__(self,
                  component_config: Union[Dict, GrowattInverterSetup],
                  modbus_id: int,