From a01a674b1ed52e6643b7b46e4ba468b75dce85dd Mon Sep 17 00:00:00 2001
From: Sarah Chen <schen@galionsystems.com>
Date: Mon, 3 Nov 2025 14:22:05 -0600
Subject: [PATCH] add write_coil for pump control

---
 galion_modbus/poll.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/galion_modbus/poll.py b/galion_modbus/poll.py
index e25c2b6..8a33957 100644
--- a/galion_modbus/poll.py
+++ b/galion_modbus/poll.py
@@ -30,3 +30,13 @@ def poll_site(site_cfg):
                 tag = plc.get("tags", {}).get(str(i), f"{plc['host']}:r{i}")
                 results[tag] = val
     return results
+
+
+def write_coil(host, port=502, unit=1, address=0, value=True):
+    c = ModbusTcpClient(host, port=port)
+    c.connect()
+    try:
+        rr = c.write_coil(address, value, unit)
+        return not rr.isError()
+    finally:
+        c.close()