实用自动化运维Python脚本分享

#!/usr/bin/python

# -*- coding: UTF-8 -*-

import paramiko

import sys

import threading

#Copy local file to remote server.

def sshclient_scp(hostname, port, username, password, local_path, remote_path):

 t = paramiko.Transport((hostname, port))

 t.connect(username=username, password=password) # 登录远程服务器

 sftp = paramiko.SFTPClient.from_transport(t) # sftp传输协议

 sftp.put(local_path, remote_path)

 t.close()

def sshclient_scp_get(hostname, port, username, password, remote_path, local_path):

 t = paramiko.Transport((hostname, port))

 t.connect(username=username, password=password) # 登录远程服务器

 sftp = paramiko.SFTPClient.from_transport(t) # sftp传输协议

 sftp.get(remote_path, local_path)

 t.close()

def sshclient_execmd(hostname, port, username, password, execmd):

 paramiko.util.log_to_file("paramiko.log")

 s = paramiko.SSHClient()

 s.set_missing_host_key_policy(paramiko.AutoAddPolicy())

 s.connect(hostname=hostname, port=port, username=username, password=password)

 stdin, stdout, stderr = s.exec_command(execmd)

 stdin.write("Y") # Generally speaking, the first connection, need a simple interaction.

 line=stdout.read()

 s.close()

 print (hostname+":")

 print line

try:

 file_name = sys.argv[1]

 cmd= sys.argv[2]

except IndexError:

 print 'Wrong params!'

 print 'Usage :'

 print '  batch.py "$OS_LIST_FILE" "$BATCH_EXECUTE_CMD"'

 print 'cat oslist.txt:'

 print '192.168.0.1,22,oracle,passwd1'

 print '192.168.0.2,22,oracle,passwd1'

 print '192.168.0.3,24,oracle,passwd1'

 print 'Format is :'

 print 'IPADDR,SSHPORT,USERNAME,PASSWORD'

 print 'Examples of usage:'

 print './batch.py "/root/workspace/oslist.txt" "df -h"'

 sys.exit()

#file_name = sys.argv[1]

#cmd= sys.argv[2]

#maintenance_osinfo

with open(file_name) as file_object:

 for line in file_object:

  splits_str = line.rstrip().split(',')

  a=threading.Thread(target=sshclient_execmd,args=(splits_str[0],int(splits_str[1]),splits_str[2],splits_str[3],cmd))

  a.start()

  #print sshclient_execmd(splits_str[0],int(splits_str[1]),splits_str[2],splits_str[3],cmd)

#  print sshclient_scp(splits_str[0], int(splits_str[1]), splits_str[2], splits_str[3], file_name, splits_str[4]+file_name)