From 78e12d3b2a4a4d16988622bf3aa95d0305fb79d6 Mon Sep 17 00:00:00 2001
From: jwansek <eddie.atten.ea29@gmail.com>
Date: Sat, 3 Jan 2026 18:41:12 +0000
Subject: Added OCR and InfluxDB

---
 upsmartmonitor.py | 79 +++++++++++++++++++++++++++++++++++++++++++++++--------
 1 file changed, 68 insertions(+), 11 deletions(-)

(limited to 'upsmartmonitor.py')

diff --git a/upsmartmonitor.py b/upsmartmonitor.py
index a23bacb..360f375 100644
--- a/upsmartmonitor.py
+++ b/upsmartmonitor.py
@@ -1,9 +1,26 @@
+from influxdb_client import InfluxDBClient, Point, WritePrecision
+from influxdb_client.client.write_api import SYNCHRONOUS
+import dotenv
+import pytesseract
 import subprocess
 import tempfile
 import pyautogui
+import json
 import cv2
 import os
 
+ROIS = {
+    "input_voltage": ((38, 182, 123, 28), True),
+    "input_frequency": ((40, 238, 125, 31), True),
+    "output_voltage": ((510, 180, 125, 29), True),
+    "load_level": ((512, 240, 125, 28), True),
+    "battery_voltage": ((146, 308, 91, 31), True),
+    "temperature": ((356, 312, 92, 27), True),
+    "battery_capacity": ((566, 312, 90, 26), True),
+    "ups_status": ((304, 429, 87, 23), False),
+    "utility_status": ((523, 427, 121, 32), False)
+}
+
 def get_window_geometries():
     return [
         s.split()[:6] + [" ".join(s.split()[7:])] 
@@ -25,24 +42,64 @@ def focus_upsmart():
     pyautogui.moveTo(int(geom[2]), int(geom[3]), duration = 0.5)
     pyautogui.click()
 
+def parse_decimal(s):
+    s = s.replace(",", ".")
+    return float("".join([d for d in s if d.isdigit() or d == "."]))
+
+def read_image(im_path):
+    im = cv2.imread(im_path)
+    fields = {}
+    with tempfile.TemporaryDirectory() as td:
+        for field, r in ROIS.items():
+            r, decimal = r
+            impath = os.path.join(td, "%s.png" % field)
+            im_crop = im[r[1]:r[1] + r[3], r[0]: r[0] + r[2]]
+            cv2.imwrite(impath, im_crop)
+
+            ocr = pytesseract.image_to_string(im_crop).strip()
+            if decimal:
+                ocr = parse_decimal(ocr)
+            fields[field] = ocr
+
+    fields["on_battery"] = int(fields["utility_status"] != 'Utility Normal')
+    return fields
+
+def influx_write(fields):
+    influxc = InfluxDBClient(
+        url = "http://%s:8086" % INFLUXDB_HOST,
+        token = os.environ["DOCKER_INFLUXDB_INIT_ADMIN_TOKEN"],
+        org = os.environ["DOCKER_INFLUXDB_INIT_ORG"]
+    )
+    influxc.ping()
+
+    write_api = influxc.write_api(write_options = SYNCHRONOUS)
+    write_api.write(
+        os.environ["DOCKER_INFLUXDB_INIT_BUCKET"],
+        os.environ["DOCKER_INFLUXDB_INIT_ORG"],
+        [{
+            "measurement": "ups_status",
+            "fields": fields
+        }],
+        write_precision = WritePrecision.S
+    )
+
 def main():
     focus_upsmart()
 
     with tempfile.TemporaryDirectory() as td:
-        fp = os.path.join(td, "upsmart.jpg")
+        fp = os.path.join(td, "upsmart.png")
         subprocess.run(["gnome-screenshot", "--file=%s" % fp, "-w"])
 
-        im = cv2.imread(fp)
-        # im = cv2.cvtColor(im, cv2.COLOR_RGB2BGR)
-        print(fp)
-
-        cv2.imshow("im", im)
-        cv2.waitKey(0)
-        
-        # plt.imshow(img, cmap = 'gray', interpolation = 'bicubic')
-        # plt.xticks([]), plt.yticks([])  # to hide tick values on X and Y axis
-        # plt.show()
+        fields = read_image(fp)
+        print(json.dumps(fields, indent = 4))
+        influx_write(fields)
 
 
 if __name__ == "__main__":
+    env_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "config.env")
+    if os.path.exists(env_path):
+        import dotenv
+        dotenv.load_dotenv(dotenv_path = env_path)
+        INFLUXDB_HOST = "192.168.69.5"
+
     main()
\ No newline at end of file
-- 
cgit v1.2.3