Simple python wrapper to connect to Azure VM via SSH

azure-ssh.py

#!/usr/bin/python
 
import sys, subprocess, json, os
 
if __name__ == "__main__":
 
    
    key_path='[PATH_TO_PRIVATE_KEY]'
 
    proc = subprocess.Popen("azure vm list --json", shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    out = proc.communicate()[0]
    vms = json.loads(out)
 
 
    if len(vms) == 0:
        print "Error: No VMs exist"
        sys.exit(0)
 
    if len(sys.argv) < 2:
        print "Usage: ./az-connect [HOSTNAME] [USERNAME]"
        sys.exit(0)
 
    if len(sys.argv) == 2:
        username = os.environ['USER']
    else:
        username = sys.argv[2]
 
    for vm in vms:
 
        if vm['VMName'] == sys.argv[1]:
            
            cmd = "ssh -o UserKnownHostsFile=/dev/null -o CheckHostIP=no -o StrictHostKeyChecking=no -i {0} -p {1} {2}@{3}".format(key_path, vm['Network']['Endpoints'][0]['Port'], username, vm['DNSName'])
            print "Running: {0}".format(cmd)
            sys.exit(subprocess.call(cmd, shell=True))