Python模块之pexpect-linux学习

vivion32

# encoding: utf-8　　
#filename:python_rsync.py
　　

　　
__author__ = "ZhouLS"
　　

　　
#同要用来同步远程主机上的日志文件到本地。
　　

　　
import pexpect
　　
import os,sys
　　
import time
　　

　　
#远程主机地址
　　
Dir = "/usr/local/tomcat48080/webapps/WebApi_Develop/logs"
　　
#本地主机地址
　　
Ldir = "./logs/"
　　
#远程主机列表，主机：密码 #也可自己定义其他方式，
　　
hosts = ['192.168.20.20:redhat','192.168.20.22:redhat']
　　
#
　　
#定义一个同步文件的方法
　　

　　
def rsync(sfile):
　　
    #使用for循环取出远程主机和密码，并执行相应的远程命令，自动提交密码
　　
    for i in hosts:
　　
        hostname = i.split(':')[0]
　　
        passwd = i.split(':')[1]
　　
        sfile = os.path.join(Dir,sfile)
　　
        dfile = os.path.join(Ldir,hostname)
　　
        #cmd命令可以修改为自己的命令，自己发挥，我用的rsync命令同步文件的。
　　
        cmd = """rsync -azv root@%s:%s %s""" % (hostname,sfile,dfile)
　　
        print cmd
　　

　　
        #下面是自动交互程序
　　

　　
        ssh = pexpect.spawn(cmd, [], 86400)
　　
        try:
　　
            while True:#使用一个循环来进入交互式
　　
                i = ssh.expect(['assword: ', 'continue connecting (yes/no)?']) #接收交互的提示
　　
                if i == 0: #如果是assword:直接发送密码，并跳出循环
　　
                    ssh.sendline(passwd)
　　
                    break
　　
                elif i == 1:#如果提示的是yes/no)?直接发送yes并继续下一个循环
　　
                    ssh.sendline('yes')
　　
        except pexpect.EOF:
　　
            ssh.close
　　
        else:
　　
            ssh.expect(pexpect.EOF)
　　
            ssh.close()
　　

　　
        print "Done"
　　

　　
#测试程序
　　
if __name__ == "__main__":
　　
    rsync('command.log.2015-07-13')