From 4c53a83a0ac93a56609289cce25778c333d1c269 Mon Sep 17 00:00:00 2001 From: stahlz Date: Fri, 3 Oct 2025 01:06:56 +0000 Subject: [PATCH] Add wifi.py --- wifi.py | 34 ++++++++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 wifi.py diff --git a/wifi.py b/wifi.py new file mode 100644 index 0000000..94a6bba --- /dev/null +++ b/wifi.py @@ -0,0 +1,34 @@ +#!/usr/bin/env python3 +import paramiko +import datetime + +# Device connection info +DEVICE_IP = "192.168.100.251" +USERNAME = "stahlz" +PASSWORD = "ZlNOBgEsbZ6ISXj0" +REBOOT_CMD = "sudo /sbin/reboot" +LOGFILE = "unifi-reboot.log" + +def log(message): + timestamp = datetime.datetime.now().strftime("[%Y-%m-%d %H:%M:%S]") + with open(LOGFILE, "a") as f: + f.write(f"{timestamp} {message}\n") + print(f"{timestamp} {message}") + +def reboot_device(): + log(f"Rebooting {DEVICE_IP} (user: {USERNAME})...") + try: + ssh = paramiko.SSHClient() + ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) + ssh.connect(DEVICE_IP, username=USERNAME, password=PASSWORD, timeout=10) + stdin, stdout, stderr = ssh.exec_command(REBOOT_CMD) + output = stdout.read().decode().strip() + ssh.close() + log(f"SUCCESS: {DEVICE_IP} reboot command sent.") + log(f"Output: {output}") + except Exception as e: + log(f"ERROR: Failed to reboot {DEVICE_IP}. Reason: {e}") + log(f"Error: {e}") + +if __name__ == "__main__": + reboot_device() \ No newline at end of file