import paramiko
import sys

sys.stdout.reconfigure(encoding='utf-8')

key_path = r"C:\Users\Snarf\.ssh\id_ed25519_snarfnet"
client = paramiko.SSHClient()
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())

client.connect("140.44.4.41", username="root", key_filename=key_path, timeout=10)

patch_cmd = """k3s kubectl -n cattle-system patch deployment cattle-cluster-agent --type='strategic' -p '{"spec":{"template":{"spec":{"hostAliases":[{"ip":"140.44.4.34","hostnames":["aerodrome.snarfnet.net"]}]}}}}'"""
stdin, stdout, stderr = client.exec_command(patch_cmd)
print("PATCH STDOUT:", stdout.read().decode('utf-8'))
print("PATCH STDERR:", stderr.read().decode('utf-8'))

client.exec_command("k3s kubectl -n cattle-system rollout status deployment/cattle-cluster-agent --timeout=60s")

stdin, stdout, stderr = client.exec_command("k3s kubectl -n cattle-system logs -l app=cattle-cluster-agent -c cluster-register --tail=15")
print("\nLOGS AFTER PATCH:")
print(stdout.read().decode('utf-8'))

client.close()