simulate modbus from registry dumps

Author: HotNoob

tools/modbus_server_sim.py

@@ -1,16 +1,41 @@
 ''' simulate modbus tcp server for testing ppg '''
+import json
 import sys
-from modbus_tk import modbus_tcp, hooks, utils
-from modbus_tk.defines import HOLDING_REGISTERS
+
+from modbus_tk import hooks, modbus_tcp, utils
+from modbus_tk.defines import HOLDING_REGISTERS, READ_INPUT_REGISTERS
+
 
 def on_write_request(request):
     print(f"Write request: {request}")
 
 
 server = modbus_tcp.TcpServer(address="0.0.0.0", port=5020)
 slave = server.add_slave(1)
-slave.add_block('0', HOLDING_REGISTERS, 0, 100)  # 100 registers
-slave.set_values('0', 40, [1] * (55 - 40 + 1)) #regiters 40-55 set to 1. for emulating hdhk_16ch_ac_module
+
+#load registries
+input_save_path = "input_registry.json"
+holding_save_path = "holding_registry.json"
+
+#load previous scan if enabled and exists
+with open(input_save_path, "r") as file:
+    input_registry = json.load(file)
+
+with open(holding_save_path, "r") as file:
+    holding_registry = json.load(file)
+
+# Convert keys to integers
+input_registry = {int(key): value for key, value in input_registry.items()}
+holding_registry = {int(key): value for key, value in holding_registry.items()}
+
+slave.add_block('INPUT', READ_INPUT_REGISTERS, min(input_registry.keys()), max(input_registry.keys()))  # 100 registers
+slave.add_block('HOLDING', HOLDING_REGISTERS,  min(holding_registry.keys()), max(holding_registry.keys()))  # 100 registers
+
+for address, value in input_registry.items():
+    slave.set_values('INPUT', address, [value])
+
+for address, value in holding_registry.items():
+    slave.set_values('HOLDING', address, [value])
 
 server.start()
 print("Modbus server is running on port 5020...")