用python脚本执行ssh远程登录多台机器

缺乏、安全感 2022-04-23 04:24 221阅读 0赞

### [用python脚本执行ssh远程登录多台机器 ][python_ssh_] ###

分类： [ linux][linux] 2012-08-29 14:27  295人阅读 [评论][Link 1](0) 收藏 [举报][Link 2]

[远程登录][Link 3] [python][] [脚本][Link 4] [ssh][] [dst][] [command][]

### [Kids Return: \[*Python* \]一个用*ssh* 来远程*登录* 多台机器并执行命令的脚本][Kids Return_ _Python_ssh] ###

\[ *Python* \]一个用 *ssh* 来远程 *登录* 多台机器并执行命令的脚本  **....** 阅读器无法显示某些pdf文档的中文问题 · Ubuntu以及MacOS下使用街机 *模拟* 器Mame · \[ *Python* \] *Python* 3.0来了  **...**   
apc999.blogspot.com/2009/01/**python** \_19.html -  [网页快照][Link 5] - [类似结果][Link 6]

来源：apc999.blogspot.com/2009/01/**python** \_19.html（以上为GOOGLE搜索结果，因blogspot被墙了，故转载止此处，供己收藏，亦与众分享）

功能类似于multissh。事实上我也抄了这个名字//grin。  
要求安装了 [pexpect][] 这个包先。  
用法见usage:  
Usage: ./multissh.py -f cmdfile -l username -c cmd -n nodesfile -v -r  
execut cmd on remote hosts (all hosts in ./hosts.txt by default)  
\-v verbose  
\-r recording hosts on which mission succeeded and failed  
\-l username  
\-c cmd to be executed remotely  
\-n file containing the nodes  
\-f file conaining the cmd  
\-h show the usage  
就是指定一个文件比如nodes.txt以及命令以后，它可以自动 登录 到nodes.txt包含的节点里执行命令。可以在源文件里替换进你自己的密码，也可以使用公钥密钥 登录 不需输入密码。指定了v选项的话得到在远端每台主机上的详细输出。指定了r选项的话记录下那些节点成功那些节点失败。  
我前面的帖子里有关于ansi\_color的一个脚本，拿过来可以配合使用得到彩色输出

Python代码   [![收藏代码][icon_star.png]][icon_star.png 1]

1.    1. \#!/usr/bin/python   
2.    2. **import** sys    
3.    3. **import** os    
4.    4. **import** getopt    
5.    5. **import** pexpect    
6.    6. **try**:    
7.    7.    **from** ansi\_color **import** \* \#就是我前面帖子里关于ansi\_color的几个定义    
8.    8. **except** ImportError:    
9.    9.    **def** color\_str(s, \*args):    
10.  10.        **return** s    
11.  11.    fg\_green = None    
12.  12.    fg\_red = None    
13.  13.    fg\_blue = None    
14.  14. password="123456" \#替换成你自己的密码。    
15.  15. **def** do(cmds, dst, username, outfile):    
16.  16.    **global** verbose, is\_quiet, good\_hosts    
17.  17.    **print** "executing \\"%s\\""%(repr(cmds))    
18.  18.    **try**:    
19.  19.        prompt = "^.\*\$.\*\$:|\\$"    
20.  20.        hostname = dst    
21.  21.        sshcmd = '<b style="color: black; background-color: rgb(153, 255, 153);">ssh</b> %s'%(hostname)    
22.  22.        **if** username != None:    
23.  23.            sshcmd = sshcmd + " -l %s"%username    
24.  24.        s = pexpect.spawn(command=sshcmd, timeout=20)    
25.  25.        s.logfile\_read = outfile    
26.  26.        s.setecho(True)    
27.  27.        i = -1    
28.  28.        **while** (i<>0):    
29.  29.            i = s.expect(\[prompt,"Are you sure you want to continue connecting (yes/no)?","Password:"\])    
30.  30.            **if** i == 1:    
31.  31.                s.sendline("yes")    
32.  32.            **elif** i == 2:    
33.  33.                s.sendline(password)    
34.  34.        **for** cmd **in** cmds:         
35.  35.            s.sendline(cmd)    
36.  36.            s.expect(prompt)    
37.  37.        s.sendline("exit")    
38.  38.        s.close()    
39.  39.        **if** verbose:    
40.  40.            **print**    
41.  41.        **print** "\["\+color\_str("OK!", fg\_green)+"\]"    
42.  42.        **if** recording:    
43.  43.            **print**>>f\_good, hostname    
44.  44.            f\_good.flush()    
45.  45.        good\_hosts.append(hostname)    
46.  46.    **except** pexpect.ExceptionPexpect:    
47.  47.        **if** verbose:    
48.  48.            **print**    
49.  49.        **print** "\["\+color\_str("Fail!", fg\_red)+"\]"    
50.  50.        **if** recording:    
51.  51.            **print**>>f\_bad, hostname    
52.  52.            f\_bad.flush()    
53.  53.        bad\_hosts.append(hostname)    
54.  54. **def** print\_usage():    
55.  55.    **print** "Usage:\\t ./make\_do.py -f cmdfile -l username -c cmd -n nodesfile -v -r"    
56.  56.    **print** "execut cmd on remote hosts (all hosts in ./hosts.txt by default)"    
57.  57.    **print** "\\t-v verbose"    
58.  58.    **print** "\\t-r recording hosts on which mission succeeded and failed"    
59.  59.    **print** "\\t-l username"    
60.  60.    **print** "\\t-c cmd to be executed remotely"    
61.  61.    **print** "\\t-n file containing the nodes"    
62.  62.    **print** "\\t-f file conaining the cmd"    
63.  63.    **print** "\\t-h show the usage"    
64.  64.    sys.exit(-1)    
65.  65. **if** \_\_name\_\_ == "\_\_main\_\_":    
66.  66.    **try**:    
67.  67.        opts, args=getopt.getopt(sys.argv\[1:\], "l:f:n:c:vhr",\["login\_name", "cmdfile","nodesfile","command","help","verbose", "recording"\])    
68.  68.    **except** getopt.GetoptError, err:    
69.  69.        **print** str(err)    
70.  70.        print\_usage()    
71.  71.    **if** opts == \[\] **and** args == \[\]:    
72.  72.        print\_usage()    
73.  73.    hosts = None    
74.  74.    cmds = None    
75.  75.    outfile = open("/dev/null", "w")    
76.  76.    verbose = False    
77.  77.    username = None    
78.  78.    recording = False    
79.  79.    **for** o, ra **in** opts:    
80.  80.        a = ra.strip(" \\t\\n")    
81.  81.        **if** o **in** ("-h", "--help"):    
82.  82.            print\_usage()    
83.  83.        **elif** o **in** ("-n", "--nodesfile"):    
84.  84.            h = open(a, 'r')    
85.  85.            hosts = \[l.strip(" \\t\\n") **for** l **in** h\]    
86.  86.        **elif** o **in** ("-c", "--command"):    
87.  87.            cmds = \[a\]    
88.  88.        **elif** o **in** ("-f", "--cmdfile"):    
89.  89.            cmdfile =  open(a, "r")    
90.  90.            cmds = \[cmd.strip(' \\n') **for** cmd **in** cmdfile\]    
91.  91.        **elif** o **in** ("-v",  "--verbose"):    
92.  92.            outfile = sys.stdout    
93.  93.            verbose = True    
94.  94.        **elif** o **in** ("-r", "--recording"):    
95.  95.            recording = True    
96.  96.        **elif** o **in** ("-l", "--login\_name"):    
97.  97.            username = a    
98.  98.    **if** hosts **is** None:    
99.  99.        **print** "using default ./hosts.txt"    
100. 100.        h = open(os.path.join(os.path.expanduser("."), "hosts.txt"),'r')    
101. 101.        hosts = \[dst.strip(' \\n') **for** dst **in** h\]    
102. 102.    **if** cmds **is** None:    
103. 103.        **print** "-c or -f must specified"    
104. 104.        print\_usage()    
105. 105.    **if** recording:    
106. 106.        f\_good = open("good\_hosts.txt","w")    
107. 107.        f\_bad = open("bad\_hosts.txt","w")    
108. 108.    good\_hosts =\[\]    
109. 109.    bad\_hosts =\[\]    
110. 110.    **for** i **in** range(len(hosts)):    
111. 111.        dst = hosts\[i\]    
112. 112.        **print** "%d/%d: \["%(i+1, len(hosts))+ color\_str(dst, fg\_blue)+"\]"    
113. 113.        do(cmds, dst, username, outfile)    
114. 114.    **print** "%d hosts suceed!"%len(good\_hosts)    
115. 115.    outfile.close()    
116. 116.    h.close()

[python_ssh_]: http://blog.csdn.net/zouyongjin/article/details/7920098
[linux]: http://blog.csdn.net/zouyongjin/article/category/850936
[Link 1]: http://blog.csdn.net/zouyongjin/article/details/7920098#comments
[Link 2]: http://blog.csdn.net/zouyongjin/article/details/7920098#report
[Link 3]: http://blog.csdn.net/tag/details.html?tag=%e8%bf%9c%e7%a8%8b%e7%99%bb%e5%bd%95
[python]: http://blog.csdn.net/tag/details.html?tag=python
[Link 4]: http://blog.csdn.net/tag/details.html?tag=%e8%84%9a%e6%9c%ac
[ssh]: http://blog.csdn.net/tag/details.html?tag=ssh
[dst]: http://blog.csdn.net/tag/details.html?tag=dst
[command]: http://blog.csdn.net/tag/details.html?tag=command
[Kids Return_ _Python_ssh]: http://apc999.blogspot.com/2009/01/python_19.html
[Link 5]: http://203.208.35.132/search?q=cache:3l1o7qACYy0J:apc999.blogspot.com/2009/01/python_19.html+python+%E6%A8%A1%E6%8B%9Fssh%E7%99%BB%E5%BD%95&cd=2&hl=zh-CN&ct=clnk&gl=cn&st_usg=ALhdy28djmzAebTQmRP7YZD1dN7uLs_F5A
[Link 6]: http://www.google.cn/search?hl=zh-CN&newwindow=1&q=related:apc999.blogspot.com/2009/01/python_19.html
[pexpect]: http://www.noah.org/wiki/Pexpect
[icon_star.png]: http://tinypig.iteye.com/images/icon_star.png
[icon_star.png 1]: /images/20220323/27bc590dbc2b4e4d996672dd34545957.png