首页 > 代码库 > python实现动态更新远程机器列表的SSH登录脚本
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/shwhile [ 1 ]doecho "Choose host: "${ChooseList}read INPUT_VALUEcase "$INPUT_VALUE" in${SSHList} *) echo -e "\033[43;31m invalid params-$INPUT_VALUE \033[0m"; ;;esacdone
生成最终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 sshMachineListdef readFile(filename): f = open(filename) contents = ‘‘ for line in f: contents += line f.close() return contentsdef 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/shwhile [ 1 ]doecho "Choose host: "echo " 1. 127.0.0.0.1 ( 本地测试 ) " echo " 2. 1.1.1.1 ( 开发环境 ) " echo " 3. 2.2.2.2 ( 测试环境 ) " read INPUT_VALUEcase "$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"; ;;esacdone
使用:
当需要添加新的SSH机器时, 加入到 sshlist.txt , 以分号隔开 ip 地址 和 注释。 然后执行 python updatessh.py 即可生成最终用于登录的 ssh.sh 脚本。
为了少敲几个字符, 可以做个软连接: ln -s ssh.sh /usr/bin/s
python实现动态更新远程机器列表的SSH登录脚本
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。