pymodbus context manager - manage TCP connection

README.md

A simple context manager to manage the connection to a TCP Modbus slave using pymodbus.

pymodbus_context_manager.py

from pymodbus.client.sync import ModbusTcpClient

class ModbusConnection:
    def __init__(self, host, port):
        self.host = host
        self.port = port

    def __enter__(self):
        self.client = ModbusTcpClient(host=self.host, port=self.port)
        self.client.connect
        return self.client

    def __exit__(self, exc_type, exc_val, exc_tb):
        if self.client:
            self.client.close()
 

with ModbusConnection(host=..., port=...) as client:
    ...