import paramiko
import time

ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect('140.44.4.11', username='root', key_filename=r'C:\Users\Snarf\.ssh\id_ed25519_snarfnet', timeout=10)

def run(cmd):
    print(f"=== {cmd} ===")
    _, out, err = ssh.exec_command(cmd)
    res = out.read().decode().strip()
    errors = err.read().decode().strip()
    if res:
        print(res)
    if errors:
        print("ERR:", errors)

print("[*] Upgrading Grafana to 11.6.16 on Aerodrome...")
run("kubectl set image deployment/grafana grafana=docker.io/grafana/grafana:11.6.16 -n default")
run("kubectl rollout status deployment/grafana -n default --timeout=180s")
run("kubectl get pods -n default -l app.kubernetes.io/name=grafana -o wide")
ssh.close()