Python安装使用命令行交互模块pexpectpython之pexpect模块

Pexpect是一个纯Python模块,可以用来和ssh、ftp、passwd、telnet等命令行命令进行交互使用,在Linux系统下尤其好用,下面我们就来具体来看一下Python安装使用命令行交互模块pexpect的基础教程:

一、安装
1、安装easy_install工具

wget http://peak.telecommunity.com/dist/ez_setup.py

python ez_setup.py 安装easy_install工具(这个脚本会自动去官网搜索下载并安装)

python ez_setup.py -U setuptools

升级easy_install工具

2、安装pexpect

easy_install Pexpect

测试一下:

[root@OMS python]# python
Python 2.7.3rc1 default, Nov 7 2012, 15:03:45)
[GCC 4.1.2 20080704 Red Hat 4.1.2-48)] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> import pexpect
>>> import pxssh
>>>

ok已经安装完成。

二、基本用法
1.run)函数
run功能相对简单,只能实现简单交互

复制代码 代码如下:

runcommand,timeout=-1,withexitstatus=False,events=None,extra_args=None, logfile=None, cwd=None, env=None)

run运行命令,然后返回结果,与os.system类似.
示例:

pexpect.run'ls -la')
# 返回值输出,退出状态)
command_output, exitstatus) = pexpect.run'ls -l /bin', withexitstatus=1)

2.spawn类
spawn功能比run强大,可以实现更复杂交互

class spawn 
  __init__self, command, args=[], timeout=30, maxread=2000, searchwindowsize=None, logfile=None, cwd=None, env=None)

timeout指定交互是等待的超时值;
maxread设置read buffer大小. 每次pexpect尝试从TTYTeletype终端)从读取的最大字节数;

searchwindowsize 指定了从输入缓冲区中进行模式匹配的位置,默认从开始匹配;

logfile 参数指定日志的记录位置。
工作过程:

# 第一步与终端建立连接
child = pexpect.spawn'scp foo user@example.com:.')
# 第二步等待终端返回特定内容
child.expect'Password:')
# 第三步根据返回内容发送命令进行交互
child.sendlinemypassword)

3.pxssh类
pxssh是pexpect的派生类,用于建立ssh连接,比pexpect好用。

login) 建立到目标机器的ssh连接;
logout) 释放该连接;
prompt) 等待提示符,通常用于等待命令执行结束。

三、实例
写一个脚本给远程服务器发送命令,并返回结果。
脚本内容:

#!/usr/bin/python 
#2013-01-16 by larry 
import pexpect 
def loginport,user,passwd,ip,command): 
  child=pexpect.spawn'ssh -p%s %s@%s "%s"' %port,user,ip,command)) 
  o='' 
  try: 
    i=child.expect['[Pp]assword:','continue connecting yes/no)?']) 
    if i == 0: 
      child.sendlinepasswd) 
    elif i == 1: 
      child.sendline'yes') 
    else: 
      pass 
  except pexpect.EOF: 
    child.close) 
  else: 
    o=child.read) 
    child.expectpexpect.EOF) 
    child.close) 
  return o 
 
hosts=file'hosts.list','r') 
for line in hosts.readlines): 
  host=line.strip"\n") 
  if host: 
    ip,port,user,passwd,commands= host.split":") 
    for command in commands.split","): 
      print "+++++++++++++++ %s run:%s ++++++++++++" % ip,command), 
      print loginport,user,passwd,ip,command)  
hosts.close)

使用方法:

python scripts.py

host.list文件内容如下:

192.168.0.21:22999:root:123456:cat /etc/redhat-release,df -Th,whoami
192.168.0.21:22999:root:123456:cat /etc/redhat-release,df -Th,whoami

返回结果:

+++++++++++++++ 192.168.0.21 run:cat /etc/redhat-release ++++++++++++
Red Hat Enterprise Linux Server release 4
+++++++++++++++ 192.168.0.21 run:df -Th ++++++++++++
文件系统   类型  容量 已用 可用 已用% 挂载点
/dev/cciss/c0d0p6
ext3  5.9G 4.4G 1.2G 80% /
/dev/cciss/c0d0p7
ext3  426G 362G  43G 90% /opt
/dev/cciss/c0d0p5
ext3  5.9G 540M 5.0G 10% /var
/dev/cciss/c0d0p3
ext3  5.9G 4.1G 1.5G 74% /usr
/dev/cciss/c0d0p1
ext3  487M  17M 445M  4% /boot
tmpfs    tmpfs  4.0G   0 4.0G  0% /dev/shm
+++++++++++++++ 192.168.0.21 run:whoami ++++++++++++
root

更多Python安装使用命令行交互模块pexpect相关文章请关注PHP中文网!

Published by

风君子

独自遨游何稽首 揭天掀地慰生平