diff --git a/src/solaredge_modbus/__init__.py b/src/solaredge_modbus/__init__.py index f792fd1..0b7f977 100644 --- a/src/solaredge_modbus/__init__.py +++ b/src/solaredge_modbus/__init__.py @@ -443,11 +443,11 @@ def write(self, key, data): return self._write(self.registers[key], data) - def read_all(self, rtype=registerType.HOLDING): + def read_all(self, rtype=registerType.HOLDING, batches=None): registers = {k: v for k, v in self.registers.items() if (v[2] == rtype)} results = {} - for batch in range(1, len(registers)): + for batch in batches if batches else range(1, len(registers)): register_batch = {k: v for k, v in registers.items() if (v[7] == batch)} if not register_batch: