Python模块之pexpect

2023-01-31 05:01:05 python pexpect 模块

pexpect是一个用来启动子程序并对其进行自动控制的 python 模块。

pexpect 可以用来和像 ssh、passwd、telnet 、rsync等命令行程序进行自动交互

下面是我用pexpect写的一个脚本,是用来执行远程主机命令,脚本是自动交互输入远程主机的密码的

# 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')


以上脚本是同步文件,也可以修改下执行远程命令。

也可以把主机列表变一下,写在一个配置文件中,也可以对主机进行分组,实现批量管理远程主机。

大多数的自动化运维工具在批量管理主机上也是大致相似的原理,所以可以根据自己的需要实现自己的远程管理工具

上面的脚本也可以实现多线程运行,这个就需要大家自己探索了,我这里已实现。

相关文章