python实现动态更新远程机器列表的SSH登录脚本

在公司里， 常常要远程到很多机器上执行命令。机器列表会逐渐增多， 记忆这么多机器的IP或域名显然不是人脑所擅长的。因此， 需要保持一份SSH机器列表，从这些机器列表生成一个用于SSH到机器列表中机器的脚本， 执行该脚本就可以SSH到指定机器上。

必需文件：  sshlist.txt,  ssh_tpl.sh , updatessh.py  ;  输出文件：  ssh.sh

SSH 机器列表： sshlist.txt

127.0.0.0.1 ; 本地测试
1.1.1.1 ; 开发环境
2.2.2.2 ; 测试环境

SSH 脚本模板： ssh_tpl.sh

需要以机器列表生成的内容分别替换  ${ChooseList} 和 ${SSHList}

#!/bin/sh
while [  ]
do
echo "Choose host: "
${ChooseList}
read INPUT_VALUE
case "$INPUT_VALUE" in
${SSHList}
    *)
        echo -e "\033[43;31m invalid params-$INPUT_VALUE \033[0m";
        ;;
esac
done

生成最终SSH登录脚本的 python 程序： updatessh.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# -------------------------------------------------------------------------------
# Name:      updatessh.py
# Purpose:   reads ssh list from sshlist.txt , replaces contents of ssh_tpl.sh
#            and finally build a ssh.sh file to ssh specific machines
#
#
# Author:   qin.shuq
#
# Created:
# Output:   ssh.sh
#-------------------------------------------------------------------------------

sshTplFilename = "ssh_tpl.sh"
sshlistFilename = "sshlist.txt"
sshscriptFilename = "ssh.sh"
username = "qin.shuq"

def readSSHList():
    f = open(sshlistFilename)
    sshMachineList = []
    i = 1
    for line in f:
        sshIp, comment = tuple(line.split(';'))
        sshMachineList.append((i, sshIp.strip(), comment.strip()))
        i+=1
    f.close()
    return sshMachineList

def readFile(filename):
    f = open(filename)
    contents = ''
    for line in f:
        contents += line
    f.close()
    return contents

def readSSHTpl():
    return readFile(sshTplFilename)

def main():
    sshMachineList = readSSHList()
    chooseListContents = ''
    sshListContents = ''
    for (i, sshIp, comment) in sshMachineList:
        chooseListContents += "echo \" %d. %s ( %s ) \" \n" % (i, sshIp, comment)
        sshListContents += "\t%d) \n\t\tssh %s@%s\n \t\t;; \n" % (i, username, sshIp)

    sshtplContents = readSSHTpl()
    sshtplContents = sshtplContents.replace("${ChooseList}", chooseListContents).replace("${SSHList}", sshListContents)

    sshScriptFile = open(sshscriptFilename, 'w')
    sshScriptFile.write(sshtplContents)
    sshScriptFile.close()

if __name__ == '__main__':
    main()

最终生成的SSH登录脚本： ssh.sh

#!/bin/sh
while [  ]
do
echo "Choose host: "
echo " 1. 127.0.0.0.1 ( 本地测试 ) "
echo " 2. 1.1.1.1 ( 开发环境 ) "
echo " 3. 2.2.2.2 ( 测试环境 ) " 

read INPUT_VALUE
case "$INPUT_VALUE" in
    )
        ssh qin.shuq@127.0.0.0.
         ;;
    )
        ssh qin.shuq@1.1.1.1
         ;;
    )
        ssh qin.shuq@2.2.2.2
         ;; 

    *)
        echo -e "\033[43;31m invalid params-$INPUT_VALUE \033[0m";
        ;;
esac
done

使用：

当需要添加新的SSH机器时， 加入到 sshlist.txt ， 以分号隔开 ip 地址 和 注释。 然后执行 python updatessh.py 即可生成最终用于登录的 ssh.sh 脚本。

为了少敲几个字符， 可以做个软连接：  ln -s  ssh.sh  /usr/bin/s