goodwe 0.4.6__tar.gz → 0.4.8__tar.gz

{goodwe-0.4.6/goodwe.egg-info → goodwe-0.4.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: goodwe
-Version: 0.4.6
+Version: 0.4.8
 Summary: Read data from GoodWe inverter via local network
 Home-page: https://github.com/marcelblijleven/goodwe
 Author: Martin Letenay, Marcel Blijleven

goodwe-0.4.8/VERSION

@@ -0,0 +1 @@
+0.4.8

{goodwe-0.4.6 → goodwe-0.4.8}/goodwe/dt.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 import logging
 from typing import Tuple
 
-from .exceptions import InverterError, RequestRejectedException
+from .exceptions import InverterError, RequestFailedException, RequestRejectedException
 from .inverter import Inverter
 from .inverter import OperationMode
 from .inverter import SensorKind as Kind
@@ -35,6 +35,12 @@ class DT(Inverter):
         Calculated("ppv3",
                    lambda data: round(read_voltage(data, 30107) * read_current(data, 30108)),
                    "PV3 Power", "W", Kind.PV),
+        # ppv1 + ppv2 + ppv3
+        Calculated("ppv",
+                   lambda data: (round(read_voltage(data, 30103) * read_current(data, 30104))) + (round(
+                       read_voltage(data, 30105) * read_current(data, 30106))) + (round(
+                       read_voltage(data, 30107) * read_current(data, 30108))),
+                   "PV Power", "W", Kind.PV),
         # Voltage("vpv4", 14, "PV4 Voltage", Kind.PV),
         # Current("ipv4", 16, "PV4 Current", Kind.PV),
         # Voltage("vpv5", 14, "PV5 Voltage", Kind.PV),
@@ -63,7 +69,7 @@ class DT(Inverter):
                    lambda data: round(read_voltage(data, 30120) * read_current(data, 30123)),
                    "On-grid L3 Power", "W", Kind.AC),
         # 30127 reserved
-        Power("ppv", 30128, "PV Power", Kind.PV),
+        PowerS("total_inverter_power", 30128, "Total Power", Kind.AC),
         Integer("work_mode", 30129, "Work Mode code"),
         Enum2("work_mode_label", 30129, WORK_MODES, "Work Mode"),
         Long("error_codes", 30130, "Error Codes"),
@@ -72,7 +78,7 @@ class DT(Inverter):
         Reactive4("reactive_power", 30135, "Reactive Power", Kind.AC),
         # 30137 reserved
         # 30138 reserved
-        # 30139 reserved
+        Decimal("power_factor", 30139, 1000, "Power Factor", "", Kind.GRID),
         # 30140 reserved
         Temp("temperature", 30141, "Inverter Temperature", Kind.AC),
         # 30142 reserved
@@ -107,6 +113,12 @@ class DT(Inverter):
         # 30172 reserved
     )
 
+    # Inverter's meter data
+    # Modbus registers from offset 0x75f4 (30196)
+    __all_sensors_meter: Tuple[Sensor, ...] = (
+        PowerS("active_power", 30196, "Active Power", Kind.GRID),
+    )
+
     # Modbus registers of inverter settings, offsets are modbus register addresses
     __all_settings: Tuple[Sensor, ...] = (
         Timestamp("time", 40313, "Inverter time"),
@@ -133,9 +145,12 @@ class DT(Inverter):
     def __init__(self, host: str, port: int, comm_addr: int = 0, timeout: int = 1, retries: int = 3):
         super().__init__(host, port, comm_addr if comm_addr else 0x7f, timeout, retries)
         self._READ_DEVICE_VERSION_INFO: ProtocolCommand = self._read_command(0x7531, 0x0028)
-        self._READ_DEVICE_RUNNING_DATA: ProtocolCommand = self._read_command(0x7594, 0x0049)
+        self._READ_RUNNING_DATA: ProtocolCommand = self._read_command(0x7594, 0x0049)
+        self._READ_METER_DATA: ProtocolCommand = self._read_command(0x75f4, 0x01)
         self._sensors = self.__all_sensors
+        self._sensors_meter = self.__all_sensors_meter
         self._settings: dict[str, Sensor] = {s.id_: s for s in self.__all_settings}
+        self._has_meter: bool = True
 
     @staticmethod
     def _single_phase_only(s: Sensor) -> bool:
@@ -154,10 +169,13 @@ class DT(Inverter):
             self.model_name = response[22:32].decode("ascii").rstrip()
         except:
             print("No model name sent from the inverter.")
-        self.serial_number = self._decode(response[6:22])
-        self.dsp1_version = read_unsigned_int(response, 66)
-        self.dsp2_version = read_unsigned_int(response, 68)
-        self.arm_version = read_unsigned_int(response, 70)
+        # Modbus registers from 30001 - 30040
+        self.serial_number = self._decode(response[6:22])  # 30004 - 30012
+        self.dsp1_version = read_unsigned_int(response, 66)  # 30034
+        self.dsp2_version = read_unsigned_int(response, 68)  # 30035
+        self.arm_version = read_unsigned_int(response, 70)  # 30036
+        self.dsp_svn_version = read_unsigned_int(response, 72)  # 35037
+        self.arm_svn_version = read_unsigned_int(response, 74)  # 35038
         self.firmware = "{}.{}.{:02x}".format(self.dsp1_version, self.dsp2_version, self.arm_version)
 
         if is_single_phase(self):
@@ -176,8 +194,17 @@ class DT(Inverter):
         pass
 
     async def read_runtime_data(self) -> Dict[str, Any]:
-        response = await self._read_from_socket(self._READ_DEVICE_RUNNING_DATA)
+        response = await self._read_from_socket(self._READ_RUNNING_DATA)
         data = self._map_response(response, self._sensors)
+
+        if self._has_meter:
+            try:
+                response = await self._read_from_socket(self._READ_METER_DATA)
+                data.update(self._map_response(response, self._sensors_meter))
+            except (RequestRejectedException, RequestFailedException):
+                logger.info("Meter values not supported, disabling further attempts.")
+                self._has_meter = False
+
         return data
 
     async def read_setting(self, setting_id: str) -> Any:
@@ -257,7 +284,10 @@ class DT(Inverter):
         raise InverterError("Operation not supported, inverter has no batteries.")
 
     def sensors(self) -> Tuple[Sensor, ...]:
-        return self._sensors
+        result = self._sensors
+        if self._has_meter:
+            result = result + self._sensors_meter
+        return result
 
     def settings(self) -> Tuple[Sensor, ...]:
         return tuple(self._settings.values())

{goodwe-0.4.6 → goodwe-0.4.8}/goodwe/et.py

@@ -257,7 +257,8 @@ class ET(Inverter):
         Apparent4("meter_apparent_power_total", 36041, "Meter Apparent Power Total", Kind.GRID),
         Integer("meter_type", 36043, "Meter Type", "", Kind.GRID),  # (0: Single phase, 1: 3P3W, 2: 3P4W, 3: HomeKit)
         Integer("meter_sw_version", 36044, "Meter Software Version", "", Kind.GRID),
-        # Sensors added in some ARM fw update, read when flag _has_meter_extended is on
+
+        # Sensors added in some ARM fw update (or platform 745/753), read when flag _has_meter_extended is on
         Power4S("meter2_active_power", 36045, "Meter 2 Active Power", Kind.GRID),
         Float("meter2_e_total_exp", 36047, 1000, "Meter 2 Total Energy (export)", "kWh", Kind.GRID),
         Float("meter2_e_total_imp", 36049, 1000, "Meter 2 Total Energy (import)", "kWh", Kind.GRID),
@@ -268,6 +269,15 @@ class ET(Inverter):
         Current("meter_current1", 36055, "Meter L1 Current", Kind.GRID),
         Current("meter_current2", 36056, "Meter L2 Current", Kind.GRID),
         Current("meter_current3", 36057, "Meter L3 Current", Kind.GRID),
+
+        Energy8("meter_e_total_exp1", 36092, "Meter Total Energy (export) L1", Kind.GRID),
+        Energy8("meter_e_total_exp2", 36096, "Meter Total Energy (export) L2", Kind.GRID),
+        Energy8("meter_e_total_exp3", 36100, "Meter Total Energy (export) L3", Kind.GRID),
+        Energy8("meter_e_total_exp", 36104, "Meter Total Energy (export)", Kind.GRID),
+        Energy8("meter_e_total_imp1", 36108, "Meter Total Energy (import) L1", Kind.GRID),
+        Energy8("meter_e_total_imp2", 36112, "Meter Total Energy (import) L2", Kind.GRID),
+        Energy8("meter_e_total_imp3", 36116, "Meter Total Energy (import) L3", Kind.GRID),
+        Energy8("meter_e_total_imp", 36120, "Meter Total Energy (import)", Kind.GRID),
     )
 
     # Inverter's MPPT data
@@ -464,6 +474,7 @@ class ET(Inverter):
         self._READ_RUNNING_DATA: ProtocolCommand = self._read_command(0x891c, 0x007d)
         self._READ_METER_DATA: ProtocolCommand = self._read_command(0x8ca0, 0x2d)
         self._READ_METER_DATA_EXTENDED: ProtocolCommand = self._read_command(0x8ca0, 0x3a)
+        self._READ_METER_DATA_EXTENDED2: ProtocolCommand = self._read_command(0x8ca0, 0x7d)
         self._READ_BATTERY_INFO: ProtocolCommand = self._read_command(0x9088, 0x0018)
         self._READ_BATTERY2_INFO: ProtocolCommand = self._read_command(0x9858, 0x0016)
         self._READ_MPPT_DATA: ProtocolCommand = self._read_command(0x89e5, 0x3d)
@@ -472,6 +483,7 @@ class ET(Inverter):
         self._has_battery: bool = True
         self._has_battery2: bool = False
         self._has_meter_extended: bool = False
+        self._has_meter_extended2: bool = False
         self._has_mppt: bool = False
         self._sensors = self.__all_sensors
         self._sensors_battery = self.__all_sensors_battery
@@ -490,6 +502,11 @@ class ET(Inverter):
         """Filter to exclude extended meter sensors"""
         return s.offset < 36045
 
+    @staticmethod
+    def _not_extended_meter2(s: Sensor) -> bool:
+        """Filter to exclude extended meter sensors"""
+        return s.offset < 36058
+
     async def read_device_info(self):
         response = await self._read_from_socket(self._READ_DEVICE_VERSION_INFO)
         response = response.response_data()
@@ -520,9 +537,10 @@ class ET(Inverter):
         if is_2_battery(self) or self.rated_power >= 25000:
             self._has_battery2 = True
 
-        if self.rated_power >= 15000:
+        if is_745_platform(self) or self.rated_power >= 15000:
             self._has_mppt = True
             self._has_meter_extended = True
+            self._has_meter_extended2 = True
         else:
             self._sensors_meter = tuple(filter(self._not_extended_meter, self._sensors_meter))
 
@@ -577,7 +595,21 @@ class ET(Inverter):
                 else:
                     raise ex
 
-        if self._has_meter_extended:
+        if self._has_meter_extended2:
+            try:
+                response = await self._read_from_socket(self._READ_METER_DATA_EXTENDED2)
+                data.update(self._map_response(response, self._sensors_meter))
+            except RequestRejectedException as ex:
+                if ex.message == ILLEGAL_DATA_ADDRESS:
+                    logger.info("Extended meter values not supported, disabling further attempts.")
+                    self._has_meter_extended2 = False
+                    self._sensors_meter = tuple(filter(self._not_extended_meter2, self._sensors_meter))
+                    response = await self._read_from_socket(self._READ_METER_DATA_EXTENDED)
+                    data.update(
+                        self._map_response(response, self._sensors_meter))
+                else:
+                    raise ex
+        elif self._has_meter_extended:
             try:
                 response = await self._read_from_socket(self._READ_METER_DATA_EXTENDED)
                 data.update(self._map_response(response, self._sensors_meter))

{goodwe-0.4.6 → goodwe-0.4.8}/goodwe/model.py

@@ -48,3 +48,7 @@ def is_2_battery(inverter: Inverter) -> bool:
 def is_745_platform(inverter: Inverter) -> bool:
     return any(model in inverter.serial_number for model in PLATFORM_745_LV_MODELS) or any(
         model in inverter.serial_number for model in PLATFORM_745_HV_MODELS)
+
+
+def is_753_platform(inverter: Inverter) -> bool:
+    return any(model in inverter.serial_number for model in PLATFORM_753_MODELS)

{goodwe-0.4.6 → goodwe-0.4.8}/goodwe/protocol.py

@@ -37,7 +37,7 @@ class InverterProtocol:
         self._timer: asyncio.TimerHandle | None = None
         self.timeout: int = timeout
         self.retries: int = retries
-        self.keep_alive: bool = True
+        self.keep_alive: bool = False
         self.protocol: asyncio.Protocol | None = None
         self.response_future: Future | None = None
         self.command: ProtocolCommand | None = None
@@ -62,6 +62,24 @@ class InverterProtocol:
             self._close_transport()
             return self._lock
 
+    def _max_retries_reached(self) -> Future:
+        logger.debug("Max number of retries (%d) reached, request %s failed.", self.retries, self.command)
+        self._close_transport()
+        self.response_future = asyncio.get_running_loop().create_future()
+        self.response_future.set_exception(MaxRetriesException)
+        return self.response_future
+
+    def _close_transport(self) -> None:
+        if self._transport:
+            try:
+                self._transport.close()
+            except RuntimeError:
+                logger.debug("Failed to close transport.")
+            self._transport = None
+        # Cancel Future on connection lost
+        if self.response_future and not self.response_future.done():
+            self.response_future.cancel()
+
     async def close(self) -> None:
         """Close the underlying transport/connection."""
         raise NotImplementedError()
@@ -133,15 +151,16 @@ class UdpInverterProtocol(InverterProtocol, asyncio.DatagramProtocol):
                 self._partial_missing = 0
             if self.command.validator(data):
                 logger.debug("Received: %s", data.hex())
+                self._retry = 0
                 self.response_future.set_result(data)
             else:
                 logger.debug("Received invalid response: %s", data.hex())
-                asyncio.get_running_loop().call_soon(self._retry_mechanism)
+                asyncio.get_running_loop().call_soon(self._timeout_mechanism)
         except PartialResponseException as ex:
             logger.debug("Received response fragment (%d of %d): %s", ex.length, ex.expected, data.hex())
             self._partial_data = data
             self._partial_missing = ex.expected - ex.length
-            self._timer = asyncio.get_running_loop().call_later(self.timeout, self._retry_mechanism)
+            self._timer = asyncio.get_running_loop().call_later(self.timeout, self._timeout_mechanism)
         except asyncio.InvalidStateError:
             logger.debug("Response already handled: %s", data.hex())
         except RequestRejectedException as ex:
@@ -158,13 +177,28 @@ class UdpInverterProtocol(InverterProtocol, asyncio.DatagramProtocol):
 
     async def send_request(self, command: ProtocolCommand) -> Future:
         """Send message via transport"""
-        async with self._ensure_lock():
+        await self._ensure_lock().acquire()
+        try:
             await self._connect()
             response_future = asyncio.get_running_loop().create_future()
-            self._retry = 0
             self._send_request(command, response_future)
             await response_future
             return response_future
+        except asyncio.CancelledError:
+            if self._retry < self.retries:
+                self._retry += 1
+                if self._lock and self._lock.locked():
+                    self._lock.release()
+                if not self.keep_alive:
+                    self._close_transport()
+                return await self.send_request(command)
+            else:
+                return self._max_retries_reached()
+        finally:
+            if self._lock and self._lock.locked():
+                self._lock.release()
+            if not self.keep_alive:
+                self._close_transport()
 
     def _send_request(self, command: ProtocolCommand, response_future: Future) -> None:
         """Send message via transport"""
@@ -178,32 +212,19 @@ class UdpInverterProtocol(InverterProtocol, asyncio.DatagramProtocol):
         else:
             logger.debug("Sending: %s", self.command)
         self._transport.sendto(payload)
-        self._timer = asyncio.get_running_loop().call_later(self.timeout, self._retry_mechanism)
+        self._timer = asyncio.get_running_loop().call_later(self.timeout, self._timeout_mechanism)
 
-    def _retry_mechanism(self) -> None:
-        """Retry mechanism to prevent hanging transport"""
-        if self.response_future.done():
+    def _timeout_mechanism(self) -> None:
+        """Timeout mechanism to prevent hanging transport"""
+        if self.response_future and self.response_future.done():
             logger.debug("Response already received.")
-        elif self._retry < self.retries:
+            self._retry = 0
+        else:
             if self._timer:
                 logger.debug("Failed to receive response to %s in time (%ds).", self.command, self.timeout)
-            self._retry += 1
-            self._send_request(self.command, self.response_future)
-        else:
-            logger.debug("Max number of retries (%d) reached, request %s failed.", self.retries, self.command)
-            self.response_future.set_exception(MaxRetriesException)
-            self._close_transport()
-
-    def _close_transport(self) -> None:
-        if self._transport:
-            try:
-                self._transport.close()
-            except RuntimeError:
-                logger.debug("Failed to close transport.")
-            self._transport = None
-        # Cancel Future on connection close
-        if self.response_future and not self.response_future.done():
-            self.response_future.cancel()
+                self._timer = None
+            if self.response_future and not self.response_future.done():
+                self.response_future.cancel()
 
     async def close(self):
         self._close_transport()
@@ -358,24 +379,6 @@ class TcpInverterProtocol(InverterProtocol, asyncio.Protocol):
                 self._timer = None
             self._close_transport()
 
-    def _max_retries_reached(self) -> Future:
-        logger.debug("Max number of retries (%d) reached, request %s failed.", self.retries, self.command)
-        self._close_transport()
-        self.response_future = asyncio.get_running_loop().create_future()
-        self.response_future.set_exception(MaxRetriesException)
-        return self.response_future
-
-    def _close_transport(self) -> None:
-        if self._transport:
-            try:
-                self._transport.close()
-            except RuntimeError:
-                logger.debug("Failed to close transport.")
-            self._transport = None
-        # Cancel Future on connection lost
-        if self.response_future and not self.response_future.done():
-            self.response_future.cancel()
-
     async def close(self):
         await self._ensure_lock().acquire()
         try:

{goodwe-0.4.6 → goodwe-0.4.8}/goodwe/sensor.py

@@ -197,6 +197,17 @@ class Energy4(Sensor):
         return float(value) / 10 if value is not None else None
 
 
+class Energy8(Sensor):
+    """Sensor representing energy [kWh] value encoded in 8 bytes"""
+
+    def __init__(self, id_: str, offset: int, name: str, kind: Optional[SensorKind]):
+        super().__init__(id_, offset, name, 8, "kWh", kind)
+
+    def read_value(self, data: ProtocolResponse):
+        value = read_bytes8(data)
+        return float(value) / 100 if value is not None else None
+
+
 class Apparent(Sensor):
     """Sensor representing apparent power [VA] value encoded in 2 bytes"""
 
@@ -840,6 +851,14 @@ def read_bytes4_signed(buffer: ProtocolResponse, offset: int = None) -> int:
     return int.from_bytes(buffer.read(4), byteorder="big", signed=True)
 
 
+def read_bytes8(buffer: ProtocolResponse, offset: int = None, undef: int = None) -> int:
+    """Retrieve 8 byte (unsigned int) value from buffer"""
+    if offset is not None:
+        buffer.seek(offset)
+    value = int.from_bytes(buffer.read(8), byteorder="big", signed=False)
+    return undef if value == 0xffffffffffffffff else value
+
+
 def read_decimal2(buffer: ProtocolResponse, scale: int, offset: int = None) -> float:
     """Retrieve 2 byte (signed float) value from buffer"""
     if offset is not None:

{goodwe-0.4.6 → goodwe-0.4.8/goodwe.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: goodwe
-Version: 0.4.6
+Version: 0.4.8
 Summary: Read data from GoodWe inverter via local network
 Home-page: https://github.com/marcelblijleven/goodwe
 Author: Martin Letenay, Marcel Blijleven

{goodwe-0.4.6 → goodwe-0.4.8}/tests/test_dt.py

@@ -4,16 +4,17 @@ from datetime import datetime
 from unittest import TestCase
 
 from goodwe.dt import DT
-from goodwe.exceptions import RequestFailedException
+from goodwe.exceptions import RequestFailedException, RequestRejectedException
+from goodwe.modbus import ILLEGAL_DATA_ADDRESS
 from goodwe.protocol import ProtocolCommand, ProtocolResponse
 
 
 class DtMock(TestCase, DT):
 
-    def __init__(self, methodName='runTest'):
+    def __init__(self, methodName='runTest', port=8899):
         TestCase.__init__(self, methodName)
-        DT.__init__(self, "localhost", 8899)
-        self.sensor_map = {s.id_: s.unit for s in self.sensors()}
+        DT.__init__(self, "localhost", port)
+        self.sensor_map = {s.id_: s for s in self.sensors()}
         self._mock_responses = {}
 
     def mock_response(self, command: ProtocolCommand, filename: str):
@@ -24,6 +25,10 @@ class DtMock(TestCase, DT):
         root_dir = os.path.dirname(os.path.abspath(__file__))
         filename = self._mock_responses.get(command)
         if filename is not None:
+            if ILLEGAL_DATA_ADDRESS == filename:
+                raise RequestRejectedException(ILLEGAL_DATA_ADDRESS)
+            if 'NO RESPONSE' == filename:
+                raise RequestFailedException()
             with open(root_dir + '/sample/dt/' + filename, 'r') as f:
                 response = bytes.fromhex(f.read())
                 if not command.validator(response):
@@ -33,10 +38,11 @@ class DtMock(TestCase, DT):
             self.request = command.request
             return ProtocolResponse(bytes.fromhex("aa557f00010203040506070809"), command)
 
-    def assertSensor(self, sensor, expected_value, expected_unit, data):
-        self.assertEqual(expected_value, data.get(sensor))
-        self.assertEqual(expected_unit, self.sensor_map.get(sensor))
-        self.sensor_map.pop(sensor)
+    def assertSensor(self, sensor_name, expected_value, expected_unit, data):
+        self.assertEqual(expected_value, data.get(sensor_name))
+        sensor = self.sensor_map.get(sensor_name);
+        self.assertEqual(expected_unit, sensor.unit)
+        self.sensor_map.pop(sensor_name)
 
     @classmethod
     def setUpClass(cls):
@@ -47,12 +53,15 @@ class GW6000_DT_Test(DtMock):
 
     def __init__(self, methodName='runTest'):
         DtMock.__init__(self, methodName)
-        self.mock_response(self._READ_DEVICE_RUNNING_DATA, 'GW6000-DT_running_data.hex')
+        self.mock_response(self._READ_RUNNING_DATA, 'GW6000-DT_running_data.hex')
+        self.mock_response(self._READ_METER_DATA, ILLEGAL_DATA_ADDRESS)
 
     def test_GW6000_DT_runtime_data(self):
         self.loop.run_until_complete(self.read_device_info())
         data = self.loop.run_until_complete(self.read_runtime_data())
-        self.assertEqual(40, len(data))
+        self.assertEqual(42, len(data))
+
+        self.sensor_map = {s.id_: s for s in self.sensors()}
 
         self.assertSensor('timestamp', datetime.strptime('2021-08-31 12:03:02', '%Y-%m-%d %H:%M:%S'), '', data)
         self.assertSensor('vpv1', 320.8, 'V', data)
@@ -61,9 +70,7 @@ class GW6000_DT_Test(DtMock):
         self.assertSensor('vpv2', 324.1, 'V', data)
         self.assertSensor('ipv2', 3.2, 'A', data)
         self.assertSensor('ppv2', 1037, 'W', data)
-        self.assertSensor('vpv3', None, 'V', data)
-        self.assertSensor('ipv3', None, 'A', data)
-        self.assertSensor('ppv3', None, 'W', data)
+        self.assertSensor('ppv', 2031, 'W', data)
         self.assertSensor('vline1', 0, 'V', data)
         self.assertSensor('vline2', 0, 'V', data)
         self.assertSensor('vline3', 0, 'V', data)
@@ -79,13 +86,14 @@ class GW6000_DT_Test(DtMock):
         self.assertSensor('pgrid1', 609, 'W', data)
         self.assertSensor('pgrid2', 597, 'W', data)
         self.assertSensor('pgrid3', 624, 'W', data)
-        self.assertSensor('ppv', 1835, 'W', data)
+        self.assertSensor('total_inverter_power', 1835, 'W', data)
         self.assertSensor('work_mode', 1, '', data)
         self.assertSensor('work_mode_label', 'Normal', '', data)
         self.assertSensor('error_codes', 0, '', data)
         self.assertSensor('warning_code', 0, '', data)
         self.assertSensor("apparent_power", -1, "VA", data),
         self.assertSensor("reactive_power", -1, "var", data),
+        self.assertSensor("power_factor", 0.0, "", data),
         self.assertSensor('temperature', 41.3, 'C', data)
         self.assertSensor('e_day', 6.0, 'kWh', data)
         self.assertSensor('e_total', 13350.2, 'kWh', data)
@@ -120,8 +128,9 @@ class GW8K_DT_Test(DtMock):
 
     def __init__(self, methodName='runTest'):
         DtMock.__init__(self, methodName)
-        self.mock_response(self._READ_DEVICE_RUNNING_DATA, 'GW8K-DT_running_data.hex')
         self.mock_response(self._READ_DEVICE_VERSION_INFO, 'GW8K-DT_device_info.hex')
+        self.mock_response(self._READ_RUNNING_DATA, 'GW8K-DT_running_data.hex')
+        self.mock_response(self._READ_METER_DATA, ILLEGAL_DATA_ADDRESS)
 
     def test_GW8K_DT_device_info(self):
         self.loop.run_until_complete(self.read_device_info())
@@ -129,13 +138,15 @@ class GW8K_DT_Test(DtMock):
         self.assertEqual('00000DTS00000000', self.serial_number)
         self.assertEqual(1010, self.dsp1_version)
         self.assertEqual(1010, self.dsp2_version)
+        self.assertEqual(728, self.dsp_svn_version)
         self.assertEqual(8, self.arm_version)
+        self.assertEqual(49, self.arm_svn_version)
         self.assertEqual('1010.1010.08', self.firmware)
 
     def test_GW8K_DT_runtime_data(self):
         self.loop.run_until_complete(self.read_device_info())
         data = self.loop.run_until_complete(self.read_runtime_data())
-        self.assertEqual(40, len(data))
+        self.assertEqual(42, len(data))
 
         self.assertSensor('timestamp', datetime.strptime('2021-08-24 16:43:27', '%Y-%m-%d %H:%M:%S'), '', data)
         self.assertSensor('vpv1', 275.5, 'V', data)
@@ -144,6 +155,7 @@ class GW8K_DT_Test(DtMock):
         self.assertSensor('vpv2', 510.8, 'V', data)
         self.assertSensor('ipv2', 0.8, 'A', data)
         self.assertSensor('ppv2', 409, 'W', data)
+        self.assertSensor('ppv', 574, 'W', data)
         self.assertSensor('vline1', 413.7, 'V', data)
         self.assertSensor('vline2', 413.0, 'V', data)
         self.assertSensor('vline3', 408.0, 'V', data)
@@ -159,13 +171,14 @@ class GW8K_DT_Test(DtMock):
         self.assertSensor('pgrid1', 237, 'W', data)
         self.assertSensor('pgrid2', 240, 'W', data)
         self.assertSensor('pgrid3', 235, 'W', data)
-        self.assertSensor('ppv', 643, 'W', data)
+        self.assertSensor('total_inverter_power', 643, 'W', data)
         self.assertSensor('work_mode', 1, '', data)
         self.assertSensor('work_mode_label', 'Normal', '', data)
         self.assertSensor('error_codes', 0, '', data)
         self.assertSensor('warning_code', 0, '', data)
         self.assertSensor("apparent_power", 0, "VA", data),
         self.assertSensor("reactive_power", 0, "var", data),
+        self.assertSensor("power_factor", 0.0, "", data),
         self.assertSensor('temperature', 45.3, 'C', data)
         self.assertSensor('e_day', None, 'kWh', data)
         self.assertSensor('e_total', None, 'kWh', data)
@@ -193,13 +206,14 @@ class GW5000D_NS_Test(DtMock):
 
     def __init__(self, methodName='runTest'):
         DtMock.__init__(self, methodName)
-        self.mock_response(self._READ_DEVICE_RUNNING_DATA, 'GW5000D-NS_running_data.hex')
         self.mock_response(self._READ_DEVICE_VERSION_INFO, 'Mock_device_info.hex')
+        self.mock_response(self._READ_RUNNING_DATA, 'GW5000D-NS_running_data.hex')
+        self.mock_response(self._READ_METER_DATA, ILLEGAL_DATA_ADDRESS)
 
     def test_GW5000D_NS_runtime_data(self):
         self.loop.run_until_complete(self.read_device_info())
         data = self.loop.run_until_complete(self.read_runtime_data())
-        self.assertEqual(30, len(data))
+        self.assertEqual(32, len(data))
 
         self.assertSensor('timestamp', datetime.strptime('2021-09-06 06:56:01', '%Y-%m-%d %H:%M:%S'), '', data)
         self.assertSensor('vpv1', 224.4, 'V', data)
@@ -208,18 +222,20 @@ class GW5000D_NS_Test(DtMock):
         self.assertSensor('vpv2', 291.8, 'V', data)
         self.assertSensor('ipv2', 0, 'A', data)
         self.assertSensor('ppv2', 0, 'W', data)
+        self.assertSensor('ppv', 0, 'W', data)
         self.assertSensor('vline1', 0, 'V', data)
         self.assertSensor('vgrid1', 240.5, 'V', data)
         self.assertSensor('igrid1', 0.0, 'A', data)
         self.assertSensor('fgrid1', 49.97, 'Hz', data)
         self.assertSensor('pgrid1', 0, 'W', data)
-        self.assertSensor('ppv', 0, 'W', data)
+        self.assertSensor('total_inverter_power', 0, 'W', data)
         self.assertSensor('work_mode', 0, '', data)
         self.assertSensor('work_mode_label', 'Wait Mode', '', data)
         self.assertSensor('error_codes', 0, '', data)
         self.assertSensor('warning_code', 0, '', data)
         self.assertSensor("apparent_power", -1, "VA", data),
         self.assertSensor("reactive_power", -1, "var", data),
+        self.assertSensor("power_factor", -0.001, "", data),
         self.assertSensor('temperature', 1.4, 'C', data)
         self.assertSensor('e_day', 0, 'kWh', data)
         self.assertSensor('e_total', 881.7, 'kWh', data)
@@ -247,8 +263,9 @@ class GW5000_MS_Test(DtMock):
 
     def __init__(self, methodName='runTest'):
         DtMock.__init__(self, methodName)
-        self.mock_response(self._READ_DEVICE_RUNNING_DATA, 'GW5000-MS_running_data.hex')
         self.mock_response(self._READ_DEVICE_VERSION_INFO, 'GW5000-MS_device_info.hex')
+        self.mock_response(self._READ_RUNNING_DATA, 'GW5000-MS_running_data.hex')
+        self.mock_response(self._READ_METER_DATA, ILLEGAL_DATA_ADDRESS)
 
     def test_GW6000_MS_device_info(self):
         self.loop.run_until_complete(self.read_device_info())
@@ -256,13 +273,15 @@ class GW5000_MS_Test(DtMock):
         self.assertEqual('00000MSU00000000', self.serial_number)
         self.assertEqual(12, self.dsp1_version)
         self.assertEqual(12, self.dsp2_version)
+        self.assertEqual(65535, self.dsp_svn_version)
         self.assertEqual(16, self.arm_version)
+        self.assertEqual(271, self.arm_svn_version)
         self.assertEqual('12.12.10', self.firmware)
 
     def test_GW5000_MS_runtime_data(self):
         self.loop.run_until_complete(self.read_device_info())
         data = self.loop.run_until_complete(self.read_runtime_data())
-        self.assertEqual(33, len(data))
+        self.assertEqual(35, len(data))
 
         self.assertSensor('timestamp', datetime.strptime('2021-10-15 09:03:12', '%Y-%m-%d %H:%M:%S'), '', data)
         self.assertSensor('vpv1', 319.6, 'V', data)
@@ -274,18 +293,20 @@ class GW5000_MS_Test(DtMock):
         self.assertSensor('vpv3', 143.2, 'V', data)
         self.assertSensor('ipv3', 0.4, 'A', data)
         self.assertSensor('ppv3', 57, 'W', data)
+        self.assertSensor('ppv', 165, 'W', data)
         self.assertSensor('vline1', 0, 'V', data)
         self.assertSensor('vgrid1', 240.1, 'V', data)
         self.assertSensor('igrid1', 0.9, 'A', data)
         self.assertSensor('fgrid1', 49.98, 'Hz', data)
         self.assertSensor('pgrid1', 216, 'W', data)
-        self.assertSensor('ppv', 295, 'W', data)
+        self.assertSensor('total_inverter_power', 295, 'W', data)
         self.assertSensor('work_mode', 1, '', data)
         self.assertSensor('work_mode_label', 'Normal', '', data)
         self.assertSensor('error_codes', 0, '', data)
         self.assertSensor('warning_code', 0, '', data)
         self.assertSensor("apparent_power", -1, "VA", data),
         self.assertSensor("reactive_power", -1, "var", data),
+        self.assertSensor("power_factor", -0.001, "", data),
         self.assertSensor('temperature', 10.7, 'C', data)
         self.assertSensor('e_day', 0.4, 'kWh', data)
         self.assertSensor('e_total', 6.8, 'kWh', data)
@@ -304,7 +325,8 @@ class GW10K_MS_30_Test(DtMock):
     def __init__(self, methodName='runTest'):
         DtMock.__init__(self, methodName)
         self.mock_response(self._READ_DEVICE_VERSION_INFO, 'GW10K-MS-30_device_info.hex')
-        self.mock_response(self._READ_DEVICE_RUNNING_DATA, 'GW10K-MS-30_running_data.hex')
+        self.mock_response(self._READ_RUNNING_DATA, 'GW10K-MS-30_running_data.hex')
+        self.mock_response(self._READ_METER_DATA, ILLEGAL_DATA_ADDRESS)
 
     def test_GW10K_MS_30_device_info(self):
         self.loop.run_until_complete(self.read_device_info())
@@ -312,13 +334,15 @@ class GW10K_MS_30_Test(DtMock):
         self.assertEqual('5010KMSC000W0000', self.serial_number)
         self.assertEqual(0, self.dsp1_version)
         self.assertEqual(0, self.dsp2_version)
+        self.assertEqual(504, self.dsp_svn_version)
         self.assertEqual(2, self.arm_version)
+        self.assertEqual(13, self.arm_svn_version)
         self.assertEqual('0.0.02', self.firmware)
 
     def test_GW10K_MS_30_runtime_data(self):
         self.loop.run_until_complete(self.read_device_info())
         data = self.loop.run_until_complete(self.read_runtime_data())
-        self.assertEqual(33, len(data))
+        self.assertEqual(35, len(data))
 
         self.assertSensor('timestamp', datetime.strptime('2024-01-09 22:08:20', '%Y-%m-%d %H:%M:%S'), '', data)
         self.assertSensor('vpv1', 0.0, 'V', data)
@@ -330,18 +354,20 @@ class GW10K_MS_30_Test(DtMock):
         self.assertSensor('vpv3', 0.0, 'V', data)
         self.assertSensor('ipv3', 0.0, 'A', data)
         self.assertSensor('ppv3', 0, 'W', data)
+        self.assertSensor('ppv', 0, 'W', data)
         self.assertSensor('vline1', 0.0, 'V', data)
         self.assertSensor('vgrid1', 236.2, 'V', data)
         self.assertSensor('igrid1', 0.0, 'A', data)
         self.assertSensor('fgrid1', 50.0, 'Hz', data)
         self.assertSensor('pgrid1', 0, 'W', data)
-        self.assertSensor('ppv', 0, 'W', data)
+        self.assertSensor('total_inverter_power', 0, 'W', data)
         self.assertSensor('work_mode', 0, '', data)
         self.assertSensor('work_mode_label', 'Wait Mode', '', data)
         self.assertSensor('error_codes', 0, '', data)
         self.assertSensor('warning_code', 0, '', data)
         self.assertSensor("apparent_power", 0, "VA", data),
         self.assertSensor("reactive_power", 0, "var", data),
+        self.assertSensor("power_factor", 0.0, "", data),
         self.assertSensor('temperature', 24.3, 'C', data)
         self.assertSensor('e_day', 71.8, 'kWh', data)
         self.assertSensor('e_total', 3433.4, 'kWh', data)
@@ -355,12 +381,69 @@ class GW10K_MS_30_Test(DtMock):
         self.assertSensor('derating_mode_label', '', '', data)
 
 
+class GW10K_MS_TCP_Test(DtMock):
+
+    def __init__(self, methodName='runTest'):
+        DtMock.__init__(self, methodName, 502)
+        self.mock_response(self._READ_RUNNING_DATA, 'GW10K-MS-30_tcp_running_data.hex')
+        self.mock_response(self._READ_METER_DATA, ILLEGAL_DATA_ADDRESS)
+
+    def test_GW10K_MS_TCP_runtime_data(self):
+        self.loop.run_until_complete(self.read_device_info())
+        data = self.loop.run_until_complete(self.read_runtime_data())
+        self.assertEqual(42, len(data))
+
+        self.assertSensor('timestamp', datetime.strptime('2024-06-02 09:07:17', '%Y-%m-%d %H:%M:%S'), '', data)
+        self.assertSensor('vpv1', 400.6, 'V', data)
+        self.assertSensor('ipv1', 6.9, 'A', data)
+        self.assertSensor('ppv1', 2764, 'W', data)
+        self.assertSensor('vpv2', 364.0, 'V', data)
+        self.assertSensor('ipv2', 3.6, 'A', data)
+        self.assertSensor('ppv2', 1310, 'W', data)
+        self.assertSensor('ppv', 6143, 'W', data)
+        self.assertSensor('vline1', 0, 'V', data)
+        self.assertSensor('vline2', 0, 'V', data)
+        self.assertSensor('vline3', 0, 'V', data)
+        self.assertSensor('vgrid1', 241.1, 'V', data)
+        self.assertSensor('vgrid2', 0, 'V', data)
+        self.assertSensor('vgrid3', 0, 'V', data)
+        self.assertSensor('igrid1', 24.7, 'A', data)
+        self.assertSensor('igrid2', 0, 'A', data)
+        self.assertSensor('igrid3', 0, 'A', data)
+        self.assertSensor('fgrid1', 49.98, 'Hz', data)
+        self.assertSensor('fgrid2', -0.01, 'Hz', data)
+        self.assertSensor('fgrid3', -0.01, 'Hz', data)
+        self.assertSensor('pgrid1', 5955, 'W', data)
+        self.assertSensor('pgrid2', 0, 'W', data)
+        self.assertSensor('pgrid3', 0, 'W', data)
+        self.assertSensor('total_inverter_power', 5914, 'W', data)
+        self.assertSensor('work_mode', 1, '', data)
+        self.assertSensor('work_mode_label', 'Normal', '', data)
+        self.assertSensor('error_codes', 0, '', data)
+        self.assertSensor('warning_code', 0, '', data)
+        self.assertSensor('apparent_power', 5957, 'VA', data)
+        self.assertSensor('reactive_power', -6, 'var', data)
+        self.assertSensor("power_factor", 0.999, "", data),
+        self.assertSensor('temperature', 36.0, 'C', data)
+        self.assertSensor('e_day', 4.3, 'kWh', data)
+        self.assertSensor('e_total', 998.2, 'kWh', data)
+        self.assertSensor('h_total', 246, 'h', data)
+        self.assertSensor('safety_country', 32, '', data)
+        self.assertSensor('safety_country_label', '50Hz 230Vac Default', '', data)
+        self.assertSensor('funbit', 0, '', data)
+        self.assertSensor('vbus', 397.3, 'V', data)
+        self.assertSensor('vnbus', 0, 'V', data)
+        self.assertSensor('derating_mode', 0, '', data)
+        self.assertSensor('derating_mode_label', '', '', data)
+
+
 class GW20KAU_DT_Test(DtMock):
 
     def __init__(self, methodName='runTest'):
         DtMock.__init__(self, methodName)
-        self.mock_response(self._READ_DEVICE_RUNNING_DATA, 'GW20KAU-DT_running_data.hex')
         self.mock_response(self._READ_DEVICE_VERSION_INFO, 'GW20KAU-DT_device_info.hex')
+        self.mock_response(self._READ_RUNNING_DATA, 'GW20KAU-DT_running_data.hex')
+        self.mock_response(self._READ_METER_DATA, ILLEGAL_DATA_ADDRESS)
 
     def test_GW20KAU_DT_device_info(self):
         self.loop.run_until_complete(self.read_device_info())
@@ -368,13 +451,15 @@ class GW20KAU_DT_Test(DtMock):
         self.assertEqual('0000KDTA00000000', self.serial_number)
         self.assertEqual(15, self.dsp1_version)
         self.assertEqual(15, self.dsp2_version)
+        self.assertEqual(1099, self.dsp_svn_version)
         self.assertEqual(16, self.arm_version)
+        self.assertEqual(187, self.arm_svn_version)
         self.assertEqual('15.15.10', self.firmware)
 
     def test_GW20KAU_DT_runtime_data(self):
         self.loop.run_until_complete(self.read_device_info())
         data = self.loop.run_until_complete(self.read_runtime_data())
-        self.assertEqual(40, len(data))
+        self.assertEqual(42, len(data))
 
         self.assertSensor('timestamp', datetime.strptime('2022-10-21 19:23:42', '%Y-%m-%d %H:%M:%S'), '', data)
         self.assertSensor('vpv1', 390.5, 'V', data)
@@ -383,6 +468,7 @@ class GW20KAU_DT_Test(DtMock):
         self.assertSensor('vpv2', 351.6, 'V', data)
         self.assertSensor('ipv2', 7.1, 'A', data)
         self.assertSensor('ppv2', 2496, 'W', data)
+        self.assertSensor('ppv', 5151, 'W', data)
         self.assertSensor('vline1', 388.5, 'V', data)
         self.assertSensor('vline2', 391.7, 'V', data)
         self.assertSensor('vline3', 394.5, 'V', data)
@@ -398,13 +484,14 @@ class GW20KAU_DT_Test(DtMock):
         self.assertSensor('pgrid1', 1628, 'W', data)
         self.assertSensor('pgrid2', 1655, 'W', data)
         self.assertSensor('pgrid3', 1621, 'W', data)
-        self.assertSensor('ppv', 4957, 'W', data)
+        self.assertSensor('total_inverter_power', 4957, 'W', data)
         self.assertSensor('work_mode', 1, '', data)
         self.assertSensor('work_mode_label', 'Normal', '', data)
         self.assertSensor('error_codes', 0, '', data)
         self.assertSensor('warning_code', 0, '', data)
         self.assertSensor("apparent_power", 0, "VA", data),
         self.assertSensor("reactive_power", 205, "var", data),
+        self.assertSensor("power_factor", 0.999, "", data),
         self.assertSensor('temperature', 36.4, 'C', data)
         self.assertSensor('e_day', 19.8, 'kWh', data)
         self.assertSensor('e_total', 4304.8, 'kWh', data)
@@ -422,8 +509,9 @@ class GW17K_DT_Test(DtMock):
 
     def __init__(self, methodName='runTest'):
         DtMock.__init__(self, methodName)
-        self.mock_response(self._READ_DEVICE_RUNNING_DATA, 'GW17K-DT_running_data.hex')
         self.mock_response(self._READ_DEVICE_VERSION_INFO, 'GW17K-DT_device_info.hex')
+        self.mock_response(self._READ_RUNNING_DATA, 'GW17K-DT_running_data.hex')
+        self.mock_response(self._READ_METER_DATA, ILLEGAL_DATA_ADDRESS)
 
     def test_GW20KAU_DT_device_info(self):
         self.loop.run_until_complete(self.read_device_info())
@@ -431,13 +519,15 @@ class GW17K_DT_Test(DtMock):
         self.assertEqual('5017KDTT00BW0000', self.serial_number)
         self.assertEqual(12, self.dsp1_version)
         self.assertEqual(12, self.dsp2_version)
+        self.assertEqual(931, self.dsp_svn_version)
         self.assertEqual(13, self.arm_version)
+        self.assertEqual(130, self.arm_svn_version)
         self.assertEqual('12.12.0d', self.firmware)
 
     def test_GW20KAU_DT_runtime_data(self):
         self.loop.run_until_complete(self.read_device_info())
         data = self.loop.run_until_complete(self.read_runtime_data())
-        self.assertEqual(40, len(data))
+        self.assertEqual(42, len(data))
 
         self.assertSensor('timestamp', datetime.strptime('2024-05-20 10:35:55', '%Y-%m-%d %H:%M:%S'), '', data)
         self.assertSensor('vpv1', 540.0, 'V', data)
@@ -446,6 +536,7 @@ class GW17K_DT_Test(DtMock):
         self.assertSensor('vpv2', 475.5, 'V', data)
         self.assertSensor('ipv2', 14.8, 'A', data)
         self.assertSensor('ppv2', 7037, 'W', data)
+        self.assertSensor('ppv', 12707, 'W', data)
         self.assertSensor('vline1', 413.0, 'V', data)
         self.assertSensor('vline2', 411.5, 'V', data)
         self.assertSensor('vline3', 409.5, 'V', data)
@@ -461,13 +552,14 @@ class GW17K_DT_Test(DtMock):
         self.assertSensor('pgrid1', 4166, 'W', data)
         self.assertSensor('pgrid2', 4170, 'W', data)
         self.assertSensor('pgrid3', 4153, 'W', data)
-        self.assertSensor('ppv', 12470, 'W', data)
+        self.assertSensor('total_inverter_power', 12470, 'W', data)
         self.assertSensor('work_mode', 1, '', data)
         self.assertSensor('work_mode_label', 'Normal', '', data)
         self.assertSensor('error_codes', 0, '', data)
         self.assertSensor('warning_code', 0, '', data)
         self.assertSensor('apparent_power', 0, 'VA', data)
         self.assertSensor('reactive_power', 0, 'var', data)
+        self.assertSensor("power_factor", 0.0, "", data),
         self.assertSensor('temperature', 45.7, 'C', data)
         self.assertSensor('e_day', 29.3, 'kWh', data)
         self.assertSensor('e_total', 29984.4, 'kWh', data)

{goodwe-0.4.6 → goodwe-0.4.8}/tests/test_et.py

@@ -15,7 +15,7 @@ class EtMock(TestCase, ET):
     def __init__(self, methodName='runTest'):
         TestCase.__init__(self, methodName)
         ET.__init__(self, "localhost", 8899)
-        self.sensor_map = {s.id_: s.unit for s in self.sensors()}
+        self.sensor_map = {s.id_: s for s in self.sensors()}
         self._mock_responses = {}
         self._list_of_requests = []
 
@@ -41,10 +41,11 @@ class EtMock(TestCase, ET):
             self._list_of_requests.append(command.request)
             return ProtocolResponse(bytes.fromhex("aa55f700010203040506070809"), command)
 
-    def assertSensor(self, sensor, expected_value, expected_unit, data):
-        self.assertEqual(expected_value, data.get(sensor))
-        self.assertEqual(expected_unit, self.sensor_map.get(sensor))
-        self.sensor_map.pop(sensor)
+    def assertSensor(self, sensor_name, expected_value, expected_unit, data):
+        self.assertEqual(expected_value, data.get(sensor_name))
+        sensor = self.sensor_map.get(sensor_name);
+        self.assertEqual(expected_unit, sensor.unit)
+        self.sensor_map.pop(sensor_name)
 
     @classmethod
     def setUpClass(cls):
@@ -81,13 +82,15 @@ class GW10K_ET_Test(EtMock):
     def test_GW10K_ET_runtime_data(self):
         # Reset sensors
         self.loop.run_until_complete(self.read_device_info())
-        self.sensor_map = {s.id_: s.unit for s in self.sensors()}
+        self.sensor_map = {s.id_: s for s in self.sensors()}
 
         data = self.loop.run_until_complete(self.read_runtime_data())
         self.assertEqual(145, len(data))
 
+        self.assertEqual(36015, self.sensor_map.get("meter_e_total_exp").offset)
+
         # for sensor in self.sensors():
-        #   print(f"self.assertSensor('{sensor.id_}', {data[sensor.id_]}, '{self.sensor_map.get(sensor.id_)}', data)")
+        #   print(f"self.assertSensor('{sensor.id_}', {data[sensor.id_]}, '{self.sensor_map.get(sensor.id_).unit}', data)")
 
         self.assertSensor('timestamp', datetime.strptime('2021-08-22 11:11:12', '%Y-%m-%d %H:%M:%S'), '', data)
         self.assertSensor('vpv1', 332.6, 'V', data)
@@ -386,7 +389,7 @@ class GW10K_ET_fw1023_Test(EtMock):
     def test_GW10K_ET_runtime_data_fw1023(self):
         # Reset sensors
         self.loop.run_until_complete(self.read_device_info())
-        self.sensor_map = {s.id_: s.unit for s in self.sensors()}
+        self.sensor_map = {s.id_: s for s in self.sensors()}
 
         data = self.loop.run_until_complete(self.read_runtime_data())
         self.assertEqual(145, len(data))
@@ -596,7 +599,7 @@ class GEH10_1U_10_Test(EtMock):
     def test_GEH10_1U_10_runtime_data(self):
         # Reset sensors
         self.loop.run_until_complete(self.read_device_info())
-        self.sensor_map = {s.id_: s.unit for s in self.sensors()}
+        self.sensor_map = {s.id_: s for s in self.sensors()}
 
         data = self.loop.run_until_complete(self.read_runtime_data())
         self.assertEqual(125, len(data))
@@ -760,6 +763,7 @@ class GW25K_ET_Test(EtMock):
         EtMock.__init__(self, methodName)
         self.mock_response(self._READ_DEVICE_VERSION_INFO, 'GW25K-ET_device_info.hex')
         self.mock_response(self._READ_RUNNING_DATA, 'GW25K-ET_running_data.hex')
+        self.mock_response(self._READ_METER_DATA_EXTENDED2, ILLEGAL_DATA_ADDRESS)
         self.mock_response(self._READ_METER_DATA_EXTENDED, 'GW25K-ET_meter_data.hex')
         self.mock_response(self._READ_BATTERY_INFO, 'GW25K-ET_battery_info.hex')
         self.mock_response(self._READ_MPPT_DATA, 'GW25K-ET_mppt_data.hex')
@@ -782,11 +786,14 @@ class GW25K_ET_Test(EtMock):
     def test_GW25K_ET_runtime_data(self):
         # Reset sensors
         self.loop.run_until_complete(self.read_device_info())
-        self.sensor_map = {s.id_: s.unit for s in self.sensors()}
 
         data = self.loop.run_until_complete(self.read_runtime_data())
         self.assertEqual(237, len(data))
 
+        self.sensor_map = {s.id_: s for s in self.sensors()}
+
+        # self.assertEqual(36104, self.sensor_map.get("meter_e_total_exp").offset)
+
         self.assertSensor('timestamp', datetime.strptime('2023-12-03 14:07:07', '%Y-%m-%d %H:%M:%S'), '', data)
         self.assertSensor('vpv1', 737.9, 'V', data)
         self.assertSensor('ipv1', 1.4, 'A', data)
@@ -1036,6 +1043,7 @@ class GW29K9_ET_Test(EtMock):
         EtMock.__init__(self, methodName)
         self.mock_response(self._READ_DEVICE_VERSION_INFO, 'GW29K9-ET_device_info.hex')
         self.mock_response(self._READ_RUNNING_DATA, 'GW29K9-ET_running_data.hex')
+        self.mock_response(self._READ_METER_DATA_EXTENDED2, ILLEGAL_DATA_ADDRESS)
         self.mock_response(self._READ_METER_DATA_EXTENDED, 'GW29K9-ET_meter_data.hex')
         self.mock_response(self._READ_BATTERY_INFO, 'GW29K9-ET_battery_info.hex')
         self.mock_response(self._READ_BATTERY2_INFO, 'GW29K9-ET_battery2_info.hex')
@@ -1059,11 +1067,12 @@ class GW29K9_ET_Test(EtMock):
     def test_GW29K9_ET_runtime_data(self):
         # Reset sensors
         self.loop.run_until_complete(self.read_device_info())
-        self.sensor_map = {s.id_: s.unit for s in self.sensors()}
 
         data = self.loop.run_until_complete(self.read_runtime_data())
         self.assertEqual(211, len(data))
 
+        self.sensor_map = {s.id_: s for s in self.sensors()}
+
         self.assertSensor('timestamp', datetime.strptime('2024-01-17 14:49:14', '%Y-%m-%d %H:%M:%S'), '', data)
         self.assertSensor('vpv1', 682.9, 'V', data)
         self.assertSensor('ipv1', 1.5, 'A', data)
@@ -1206,32 +1215,6 @@ class GW29K9_ET_Test(EtMock):
         self.assertSensor('meter_current1', 4.6, 'A', data)
         self.assertSensor('meter_current2', 6.0, 'A', data)
         self.assertSensor('meter_current3', 13.6, 'A', data)
-        self.assertSensor('battery_bms', None, '', data)
-        self.assertSensor('battery_index', None, '', data)
-        self.assertSensor('battery_status', None, '', data)
-        self.assertSensor('battery_temperature', None, 'C', data)
-        self.assertSensor('battery_charge_limit', None, 'A', data)
-        self.assertSensor('battery_discharge_limit', None, 'A', data)
-        self.assertSensor('battery_error_l', None, '', data)
-        self.assertSensor('battery_soc', None, '%', data)
-        self.assertSensor('battery_soh', None, '%', data)
-        self.assertSensor('battery_modules', None, '', data)
-        self.assertSensor('battery_warning_l', None, '', data)
-        self.assertSensor('battery_protocol', None, '', data)
-        self.assertSensor('battery_error_h', None, '', data)
-        self.assertSensor('battery_error', None, '', data)
-        self.assertSensor('battery_warning_h', None, '', data)
-        self.assertSensor('battery_warning', None, '', data)
-        self.assertSensor('battery_sw_version', None, '', data)
-        self.assertSensor('battery_hw_version', None, '', data)
-        self.assertSensor('battery_max_cell_temp_id', None, '', data)
-        self.assertSensor('battery_min_cell_temp_id', None, '', data)
-        self.assertSensor('battery_max_cell_voltage_id', None, '', data)
-        self.assertSensor('battery_min_cell_voltage_id', None, '', data)
-        self.assertSensor('battery_max_cell_temp', None, 'C', data)
-        self.assertSensor('battery_min_cell_temp', None, 'C', data)
-        self.assertSensor('battery_max_cell_voltage', None, 'V', data)
-        self.assertSensor('battery_min_cell_voltage', None, 'V', data)
         self.assertSensor('battery2_status', 0, '', data)
         self.assertSensor('battery2_temperature', 0.0, 'C', data)
         self.assertSensor('battery2_charge_limit', 0, 'A', data)

{goodwe-0.4.6 → goodwe-0.4.8}/tests/test_protocol.py

@@ -36,14 +36,14 @@ class TestUDPClientProtocol(TestCase):
         mock_loop = mock.Mock()
         mock_get_event_loop.return_value = mock_loop
 
-        mock_retry_mechanism = mock.Mock()
-        self.protocol._retry_mechanism = mock_retry_mechanism
+        mock_timeout_mechanism = mock.Mock()
+        self.protocol._timeout_mechanism = mock_timeout_mechanism
         self.protocol.connection_made(transport)
         self.protocol._send_request(self.protocol.command, self.protocol.response_future)
 
         transport.sendto.assert_called_with(self.protocol.command.request)
         mock_get_event_loop.assert_called()
-        mock_loop.call_later.assert_called_with(1, mock_retry_mechanism)
+        mock_loop.call_later.assert_called_with(1, mock_timeout_mechanism)
 
     def test_connection_lost(self):
         self.protocol.response_future.done.return_value = True
@@ -59,41 +59,41 @@ class TestUDPClientProtocol(TestCase):
         self.protocol._transport = mock.Mock()
         self.protocol._send_request = mock.Mock()
         self.protocol.response_future.done.return_value = True
-        self.protocol._retry_mechanism()
+        self.protocol._timeout_mechanism()
 
         # self.protocol._transport.close.assert_called()
         self.protocol._send_request.assert_not_called()
 
-    @mock.patch('goodwe.protocol.asyncio.get_running_loop')
-    def test_retry_mechanism_two_retries(self, mock_get_event_loop):
-        def call_later(_: int, retry_func: Callable):
-            retry_func()
-
-        mock_loop = mock.Mock()
-        mock_get_event_loop.return_value = mock_loop
-        mock_loop.call_later = call_later
-
-        self.protocol._transport = mock.Mock()
-        self.protocol.response_future.done.side_effect = [False, False, True, False]
-        self.protocol._retry_mechanism()
-
-        # self.protocol._transport.close.assert_called()
-        self.assertEqual(self.protocol._retry, 2)
-
-    @mock.patch('goodwe.protocol.asyncio.get_running_loop')
-    def test_retry_mechanism_max_retries(self, mock_get_event_loop):
-        def call_later(_: int, retry_func: Callable):
-            retry_func()
-
-        mock_loop = mock.Mock()
-        mock_get_event_loop.return_value = mock_loop
-        mock_loop.call_later = call_later
-
-        self.protocol._transport = mock.Mock()
-        self.protocol.response_future.done.side_effect = [False, False, False, False, False]
-        self.protocol._retry_mechanism()
-        self.protocol.response_future.set_exception.assert_called_once_with(MaxRetriesException)
-        self.assertEqual(self.protocol._retry, 3)
+    # @mock.patch('goodwe.protocol.asyncio.get_running_loop')
+    # def test_retry_mechanism_two_retries(self, mock_get_event_loop):
+    #     def call_later(_: int, retry_func: Callable):
+    #         retry_func()
+    #
+    #     mock_loop = mock.Mock()
+    #     mock_get_event_loop.return_value = mock_loop
+    #     mock_loop.call_later = call_later
+    #
+    #     self.protocol._transport = mock.Mock()
+    #     self.protocol.response_future.done.side_effect = [False, False, True, False]
+    #     self.protocol._timeout_mechanism()
+    #
+    #     # self.protocol._transport.close.assert_called()
+    #     self.assertEqual(self.protocol._retry, 2)
+
+    # @mock.patch('goodwe.protocol.asyncio.get_running_loop')
+    # def test_retry_mechanism_max_retries(self, mock_get_event_loop):
+    #     def call_later(_: int, retry_func: Callable):
+    #         retry_func()
+    #
+    #     mock_loop = mock.Mock()
+    #     mock_get_event_loop.return_value = mock_loop
+    #     mock_loop.call_later = call_later
+    #
+    #     self.protocol._transport = mock.Mock()
+    #     self.protocol.response_future.done.side_effect = [False, False, False, False, False]
+    #     self.protocol._timeout_mechanism()
+    #     self.protocol.response_future.set_exception.assert_called_once_with(MaxRetriesException)
+    #     self.assertEqual(self.protocol._retry, 3)
 
     def test_modbus_rtu_read_command(self):
         command = ModbusRtuReadCommand(0xf7, 0x88b8, 0x0021)

{goodwe-0.4.6 → goodwe-0.4.8}/tests/test_sensor.py

@@ -169,6 +169,16 @@ class TestUtils(TestCase):
         data = MockResponse("ffffffff")
         self.assertIsNone(testee.read(data))
 
+    def test_energy8(self):
+        testee = Energy8("", 0, "", None)
+
+        data = MockResponse("0000000000015b41")
+        self.assertEqual(888.97, testee.read(data))
+        data = MockResponse("0000000000038E6C")
+        self.assertEqual(2330.68, testee.read(data))
+        data = MockResponse("ffffffffffffffff")
+        self.assertIsNone(testee.read(data))
+
     def test_temp(self):
         testee = Temp("", 0, "", None)
 

goodwe-0.4.6/VERSION

@@ -1 +0,0 @@
-0.4.6