Initial commit

atxctl/main.py

@@ -0,0 +1,110 @@
+#!/usr/bin/env python3
+
+from pyiArduinoI2Cexpander import *
+import os
+import argparse
+import time
+
+PROG_NAME = "ATX Control"
+PROG_VERSION = 'v' + open(os.path.join(os.path.dirname(os.path.realpath(__file__)), "VERSION")).readline()
+PROG_DESCRIPTION = ""
+PROG_EPILOG = ""
+
+EXP_ADDR = 0x08
+POWER_PIN = 0
+RESET_PIN = 1
+STATUS_PIN = 2
+
+PRESS_TIME_SEC = 1
+LONG_PRESS_TIME_SEC = 5
+
+def exp_init(exp_addr, power_pin, reset_pin, status_pin):
+    exp = pyiArduinoI2Cexpander(exp_addr)
+
+    try:
+        exp.pinMode(power_pin, OUTPUT, DIGITAL)
+        exp.digitalWrite(power_pin, LOW)
+
+        exp.pinMode(reset_pin, OUTPUT, DIGITAL)
+        exp.digitalWrite(reset_pin, LOW)
+
+        exp.pinMode(status_pin, INPUT, DIGITAL)
+        exp.pinPull(status_pin, PULL_DOWN)
+    except Exception as E:
+        print(f"Error initializing expander hat: {E}")
+        exit(1)
+
+    return exp
+
+
+def get_status(exp, status_pin):
+    print("Getting status...")
+    try:
+        status_raw = exp.digitalRead(status_pin)
+    except Exception as E:
+        print(f"Error reading status: {E}")
+        exit(1)
+
+    return status_raw
+
+
+def press_power(exp, power_pin):
+    print("Switching power...")
+    try:
+        exp.digitalWrite(power_pin, HIGH)
+        time.sleep(PRESS_TIME_SEC)
+        exp.digitalWrite(power_pin, LOW)
+    except Exception as E:
+        print(f"Error pressing power: {E}")
+        exit(1)
+
+
+def press_reset(exp, reset_pin):
+    print("Resetting...")
+    try:
+        exp.digitalWrite(reset_pin, HIGH)
+        time.sleep(PRESS_TIME_SEC)
+        exp.digitalWrite(reset_pin, LOW)
+    except Exception as E:
+        print(f"Error pressing reset: {E}")
+        exit(1)
+
+
+def power_off(exp, power_pin):
+    print("Powering off...")
+    try:
+        exp.digitalWrite(power_pin, HIGH)
+        time.sleep(LONG_PRESS_TIME_SEC)
+        exp.digitalWrite(power_pin, LOW)
+    except Exception as E:
+        print(f"Error powering off: {E}")
+        exit(1)
+
+
+def main():
+    parser = argparse.ArgumentParser(description="ATX Power Control")
+    parser.add_argument('-s', '--status', action='store_true', help='Get status of the host')
+    parser.add_argument('-p', '--press-power', action='store_true', help='Press power button')
+    parser.add_argument('-r', '--press-reset', action='store_true', help='Press reset button')
+    parser.add_argument('--power-off', action='store_true', help='Force power off (long press power button)')
+    args = parser.parse_args()
+
+    exp = exp_init(EXP_ADDR, POWER_PIN, RESET_PIN, STATUS_PIN)
+
+    if args.status:
+        if get_status(exp, STATUS_PIN):
+            print("Host is powered on")
+        else:
+            print("Host is powered off")
+    elif args.press_power:
+        press_power(exp, POWER_PIN)
+    elif args.press_reset:
+        press_reset(exp, RESET_PIN)
+    elif args.power_off:
+        power_off(exp, POWER_PIN)
+    else:
+        parser.print_help()
+
+
+if __name__ == "__main__":
+    main()