From 8ac3ac49a87e0a9ce7edcad4993f04675718ebc5 Mon Sep 17 00:00:00 2001
From: vvzvlad <git@vvzvlad.xyz>
Date: Tue, 13 Aug 2024 04:18:53 +0300
Subject: [PATCH] add first version

---
 shutdown_ups.py      | 63 ++++++++++++++++++++++++++++++++++++++++++++
 ups_shutdown.service | 11 ++++++++
 2 files changed, 74 insertions(+)
 create mode 100644 shutdown_ups.py
 create mode 100644 ups_shutdown.service

diff --git a/shutdown_ups.py b/shutdown_ups.py
new file mode 100644
index 0000000..2ef7712
--- /dev/null
+++ b/shutdown_ups.py
@@ -0,0 +1,63 @@
+import subprocess
+import time
+import requests
+import socket
+import os
+
+ups_ip = "10.31.41.46"
+time_threshold = 4
+interval = 20
+
+def parse_time(time_str):
+    try:
+        time_parts = time_str.split(":")[-4:]
+        days = int(time_parts[0])
+        hours = int(time_parts[1])
+        minutes = int(time_parts[2])
+        seconds = float(time_parts[3])
+        total_minutes = (days * 24 * 60) + (hours * 60) + minutes + (int(seconds // 60))
+        return total_minutes
+    except:
+        return False
+
+def send_email(recipient, subject, body, sender):
+    email_message = f"Subject: {subject}\nFrom: {sender}\nTo: {recipient}\n\n{body}"
+    process = os.popen('/usr/sbin/sendmail -t', 'w')
+    process.write(email_message)
+    process.close()
+
+
+def parse_states(states_str):
+    if len(states_str) < 2: return "Invalid input" 
+    second_char = states_str[1]
+    if second_char == '0': return "on-line"
+    elif second_char == '1': return "on-battery"
+    else: return "Invalid input" 
+
+def run_command(command):
+    result = subprocess.run(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
+    return result.stdout.strip()
+
+while True:
+    oid_on_battery_time = "1.3.6.1.4.1.318.1.1.1.2.2.3.0"
+    result = run_command(f'/usr/bin/snmpget -v2c -O v -O q -c public {ups_ip} {oid_on_battery_time}')
+    time_on_battery = parse_time(result)
+
+    oid_states = "1.3.6.1.4.1.318.1.1.1.11.1.1.0"
+    result = run_command(f'/usr/bin/snmpget -v2c -O v -O q -c public {ups_ip} {oid_states}')
+    state = parse_states(result.strip('"'))
+
+    print(f"Time on battery: {time_on_battery} minutes, State: {state}")
+
+    if time_on_battery != False and time_on_battery < time_threshold:
+        if state == "on-battery":
+            shutdown_status = run_command('shutdown --show 2>&1')
+            if "No scheduled shutdown" in shutdown_status:
+                print("UPS is failing, shutting down in 60 seconds...")
+                send_email(f"{socket.gethostname()}@vvzvlad.xyz", "UPS is failing, shutting down in 60 seconds...", f"Proxmox node '{socket.gethostname()}': UPS is failing, shutting down in 60 seconds...", f"{socket.gethostname()}@vvzvlad.xyz")
+                run_command('sudo shutdown -h 1 "Shutting down due to UPS failure"')
+
+    if state == "on-battery":
+        time.sleep(5)
+    else:
+        time.sleep(interval)
diff --git a/ups_shutdown.service b/ups_shutdown.service
new file mode 100644
index 0000000..a4b8bfd
--- /dev/null
+++ b/ups_shutdown.service
@@ -0,0 +1,11 @@
+[Unit]
+Description=Service to manage UPS shutdown script
+After=network.target
+
+[Service]
+ExecStart=/usr/bin/python3 /usr/local/bin/shutdown_ups.py
+Restart=always
+RestartSec=5s
+
+[Install]
+WantedBy=multi-user.target
\ No newline at end of file