update_tpus.sh

update_tpus.sh

#!/bin/bash

if ! test -f /etc/hosts.bak; then
    sudo cp /etc/hosts /etc/hosts.bak
fi

sudo cp /etc/hosts.bak /etc/hosts

YAML_INFO="
`gcloud compute tpus tpu-vm list --zone=us-central1-f --format=yaml`
`gcloud compute tpus tpu-vm list --zone=europe-west4-a --format=yaml`
"

NAMES=(`echo "$YAML_INFO" | yq eval ".name" | cut -d "/" -f6 | grep -v "\-\-\-"`)
EXTERNAL_IPS=(`echo "$YAML_INFO" | yq eval ".networkEndpoints[0].accessConfig.externalIp" | grep -v "\-\-\-"`)

for index in ${!NAMES[*]}; do
  echo "${EXTERNAL_IPS[$index]} ${NAMES[$index]}" | sudo tee -a /etc/hosts
  ssh-keygen -R ${NAMES[$index]}
done