import paramiko
import sys
import time

sys.stdout.reconfigure(encoding='utf-8')
c = paramiko.SSHClient()
c.set_missing_host_key_policy(paramiko.AutoAddPolicy())
c.connect('140.44.4.41', username='root', key_filename=r'C:\Users\Snarf\.ssh\id_ed25519_snarfnet', timeout=10)

def run(cmd):
    print(f"\n>>> {cmd}")
    _, o, e = c.exec_command(cmd)
    res = o.read().decode().strip()
    err = e.read().decode().strip()
    if res:
        print(res)
    if err:
        print("[stderr]", err)
    return res

run("kubectl rollout restart deployment/homepage-2-1744068158 -n infrastructure")
run("kubectl rollout status deployment/homepage-2-1744068158 -n infrastructure --timeout=60s")

c.close()