利用python實現批量自動化運維指令碼案例薦

本文為通過密碼或金鑰實現python批量自動化運維指令碼案例分享，是老男孩linux培訓 python課程教學案例內容，後續會分享多執行緒併發執行這個指令碼的更高階的功能(http://oldboy.blog.51cto.com/2561410/1221605)，敬請關注！

#!/usr/bin/env python
# this script comes from beijing oldboy trainning.
# e_mail:70271111@qq.com
# function: remote multi exec cmd by ssh.
# version:1.1
################################################
# oldboy trainning info.
# QQ 80042789 70271111
# site:http://www.etiantian.org
# blog:http://oldboy.blog.51cto.com
# oldboy trainning QQ group: 208160987 45039636
################################################
import paramiko
import socket
import sys,os
host = sys.argv[1]
user = sys.argv[2]
port = int(sys.argv[3])
auth_method = sys.argv[4]
if auth_method == `PASSWORD`:
    password = sys.argv[5]
    cmd = sys.argv[6]
else:
    cmd = sys.argv[5]
    s = paramiko.SSHClient()
    s.load_system_host_keys()
    s.set_missing_host_key_policy(paramiko.AutoAddPolicy())
#pkey_file = `/home/alex/.ssh/id_rsa`
#key = paramiko.RSAKey.from_private_key_file(pkey_file)
#s.connect(host,port,username,pkey=key,timeout=5)
try:
if auth_method == `SSH_KEY`:
    pkey_file = `/home/alex/.ssh/id_rsa`
    key = paramiko.RSAKey.from_private_key_file(pkey_file)
    s.connect(host,port,user,pkey=key,timeout=5)
elif auth_method == `PASSWORD`:
    s.connect(host,port,user,password,timeout=5)
    stdin,stdout,stderr = s.exec_command(cmd)
    cmd_result = stdout.read(),stderr.read()
print "--------Result from? %s -------" % host
for line in cmd_result:
    print line,
s.close()
except paramiko.AuthenticationException:
    print " 33[31;1m%s Authentication Error! 33[0m" % host
except socket.error:
    print " 33[31;1m%s Connection Refused! 33[0m" % host

執行命令：
python batch_ssh.py 10.0.0.8 oldboy 22 SSH_KEY  df
python batch_ssh.py 10.0.0.9 root   33 PASSWORD 123456 df