[python]使用pexpect模組進行批量scp

標籤：

#!/usr/bin/env python
# -*- coding: utf-8 -*-

#wangxiaofei

#awcloud自動化測試

import time,os
import threading
import sys

threads = []

username = ‘root‘
passwd = ‘www.awcloud.com‘
ssh_ip = [‘10.11.11.4‘,
‘10.11.11.5‘,
‘10.11.11.6‘,
‘10.11.11.7‘
]

try:
     import pexpect
except ImportError:
     print ‘Can not found pexpect module,Now install it......‘
     os.system(‘yum install pexpect -y‘)
     time.sleep(3)
     check_pexpect=os.popen(‘rpm -qa | grep pexpect‘)
     if ‘pexpect‘ in check_pexpect.read():
         print ‘check pexpect module is exists,going on...‘
     else:
         print ‘check pexpect module is not exists,exit...‘
         sys.exit()
else:
    import pexpect

def scp_file():
    i=0
    for i in range(len(ssh_ip)):
        try:
           scp_command=pexpect.spawn(‘scp ‘+scp_filename+‘ [email protected]‘+ssh_ip[i]+‘:/root/‘)
           expect_result = scp_command.expect([r‘assword:‘,r‘yes/no‘],timeout=30)
           if expect_result == 0:
               scp_command.sendline(passwd)

              #這句話真的很神奇，如果不加這句話，程式會執行，但是不執行copy，請各路神人解釋 

              scp_command.read()
           elif expect_result == 1:
               scp_command.sendline(‘yes‘)
               scp_command.expect(‘assword:‘,timeout=30)
               scp_command.sendline(passwd)
              #don‘t delete this code,if your do then,the program will be faill.
              scp_command.read()
              #important
          else:
              print ‘Unknow Result......‘
       except pexpect.EOF:
              print ‘Uploading Fail....‘
              print pexpect.EOF
       except pexpect.TIMEOUR:
              print ‘Uploading time out.....‘
    time.sleep(2)

 

if ‘__name__‘ == ‘__main__‘:

   scp_file()

[python]使用pexpect模組進行批量scp