import os
import sys
import paramiko

sys.stdout.reconfigure(encoding="utf-8")
pkey = paramiko.Ed25519Key.from_private_key_file(os.path.expanduser('~/.ssh/id_ed25519_snarfnet'))

client = paramiko.SSHClient()
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
try:
    client.connect("140.44.4.1", username="root", pkey=pkey, timeout=5)
    print("SUCCESS SSH to 140.44.4.1 with key")
    stdin, stdout, stderr = client.exec_command("uname -a")
    print(stdout.read().decode())
    client.close()
except Exception as e:
    print(f"Key failed: {e}")
    try:
        client.connect("140.44.4.1", username="root", password="Katzir476!", timeout=5)
        print("SUCCESS SSH to 140.44.4.1 with password")
        stdin, stdout, stderr = client.exec_command("uname -a")
        print(stdout.read().decode())
        client.close()
    except Exception as e2:
        print(f"Password failed: {e2}")