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

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

#!/bin/sh
while [ 1 ]
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

#!/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()

#!/bin/sh
while [ 1 ]
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
    1) 
        ssh qin.shuq@127.0.0.0.1
         ;; 
    2) 
        ssh qin.shuq@1.1.1.1
         ;; 
    3) 
        ssh qin.shuq@2.2.2.2
         ;; 

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