脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → python Pexpect模块

python Pexpect模块的使用

 更新时间：2020年12月25日 09:36:37   作者：码道仕  
这篇文章主要介绍了python Pexpect模块的使用，帮助大家更好的理解和使用python，感兴趣的朋友可以了解下

Pexpect简介

在讲解Pexpect之前，我们需要先了解一下Expect这个脚本语言，它是由TCL语言实现的，主要用于人机交互式对话的自动化控制，可以用来完成ssh、ftp、telnet等命令行程序的自动化交互。Pexpect其实就是一个用Python语言实现的类Expect功能的模块，通过它就可以在Python中完成Expect所完成的功能。

Pexpect的基本工作流程，基本可以分为以下三个步骤：

  1. 首先用spawn来执行一个程序；
  2. 然后用expect方法来等待指定的关键字，这个关键字是被执行的程序打印到标准输出上面的；
  3. 最后当发现这个关键字以后，使用send/sendline方法发送字符串给这个程序。

通常在程序中第一步只需要做一次，第二步和第三步会不停的循环来完成整个工作。当然在Pexpect中还有很多其他方法，编写程序时可以根据自己的需求选择使用。

Pexpect API

spawn类

class spawn(SpawnBase):
  '''This is the main class interface for Pexpect. Use this class to start
  and control child applications. '''

  # This is purely informational now - changing it has no effect
  use_native_pty_fork = use_native_pty_fork
  
	def __init__(self, command, args=[], timeout=30, maxread=2000,
           searchwindowsize=None, logfile=None, cwd=None, env=None,
           ignore_sighup=False, echo=True, preexec_fn=None,
           encoding=None, codec_errors='strict', dimensions=None,
           use_poll=False):

通过spawn()方法用来执行一个程序，返回程序的操作句柄，后续就可以通过操作句柄来与这个程序进行交互了。

# 子程序退出时会引发pexpect.EOF异常，即如果捕捉到pexpect.EOF则说明子程序已退出
process = pexpect.spawn('ls -l')
process.expect(pexpect.EOF)
result = process.before.decode()
print(result)

command参数并不支持字符的特殊含义（比如管道符、通配符、重定向符等），在Linux系统中如果想使用这些符号的特殊含义就必须加上shell来运行。

# 示例一
process = pexpect.spawn('bash -c "ls -l | wc -l"')
process.expect(pexpect.EOF)

# 示例二
# 第一个参数为主程序，而args列表里的元素是主程序的参数
process = pexpect.spawn('bash', ['-c', 'ls -l | wc -l'])
process.expect(pexpect.EOF)

expect()方法

当使用spawn()方法启动了一个程序并返回程序控制句柄后，就可以使用expect()方法来等待指定的关键字了。关键字可以是字符串、正则表达式、EOF、TIMEOUT或者以上类型组成的列表，用来匹配子程序返回的结果。如果只提供字符串等非列表，则匹配成功后返回0，如果提供列表，则返回匹配成功的列表元素的索引，匹配失败会抛出异常。

process = pexpect.spawn('ls -l')
# 匹配expect字符
process.expect('expect')

before/after/match：当expect()匹配到关键字之后，系统会自动给这三个变量赋值，通过这三个变量可以获取子程序运行输出。

  • before：保存了到匹配到关键字为止，缓存里面已有的所有数据。也就是说如果缓存里缓存了100个字符的时候匹配到了关键字，那before就是除了匹配到的关键字之外的所有字符。
  • after：保存了匹配到了关键字。
  • match：保存的是匹配到的正则表达式的实例，和上面的after相比一个是匹配到的字符串，一个是匹配到的正则表达式实例。
process = pexpect.spawn('ls -l')
process.expect('expect')
print(process.before.decode())
print(process.after.decode())
print(process.match)

如果expect()过程中发生错误，那么before保存到目前为止缓存里的所有数据，after和match都是None。

如果没匹配成功则会抛出异常，可以通过匹配异常，让异常不在终端显示。

process = pexpect.spawn('ls -l')
# 返回0表示匹配成功，返回1和2表示匹配到了异常
index = process.expect(['expect', pexpect.EOF, pexpect.TIMEOUT])
print(index)

send()/sendline()方法

sendline()和send()的区别就是sendline()发送的是带回车符的字符串。

process = pexpect.spawn('nslookup')
process.expect('>')
process.sendline('www.baidu.com')
process.expect('>')
print(process.before.decode())
process.sendline('exit')

Pexpect还提供了很多其他方法，这里不再详细阐述，使用时可参考其官方文档。

interact()方法

interact()表示将终端控制权交给用户（或者说将标准输入交给用户）。通常情况下Pexpect会接管所有的输入和输出，如果需要用户介入完成部分工作的时候，interact()就派上用场了。

# 让出控制权给用户
process.interact()
# 通过设置escape_character的值定义返回码，当用户输入此值后，会将控制权重新交给pexpect
process.interact(escape_character='\x1d', input_filter=None, output_filter=None)

应用示例

接下来通过SSH连接远程服务器的示例来体验下Pexpect的使用方法。

#!/usr/bin/env python3.6
#-*- coding:utf-8 -*-

import pexpect

def main(server):
  command = 'ssh -p %s %s@%s' % (server['port'], server['username'], server['hostname'])
  process = pexpect.spawn(command, timeout=30)
  print(f'命令: {command}')
  expect_list = [
    'yes/no',
    'password:',
    pexpect.EOF,
    pexpect.TIMEOUT,
  ]
  index = process.expect(expect_list)
  print(f'匹配到: {index} => {expect_list[index]}')
  if index == 0: 
    process.sendline("yes")
    expect_list = [
      'password:',
      pexpect.EOF,
      pexpect.TIMEOUT,
    ]
    index = process.expect(expect_list)
    print(f'匹配到: {index} => {expect_list[index]}')
    if index == 0:
      process.sendline(server['password'])
      process.interact()
    else:
      print('EOF or TIMEOUT')
  elif index == 1:
    process.sendline(server['password'])
    process.interact()
  else:
    print('EOF or TIMEOUT')
  
if __name__ == '__main__':
  server = {
    'hostname': '192.168.1.100',
    'port': '22',
    'username': 'admin',
    'password': 'ABuklhsfnVyxI',
  }
  main(server)

以上就是python Pexpect模块的使用的详细内容，更多关于python Pexpect模块的资料请关注脚本之家其它相关文章！

您可能感兴趣的文章:

相关文章

  • python实现远程控制电脑

    python实现远程控制电脑

    这篇文章主要为大家详细介绍了python实现远程控制电脑，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
    2019-05-05
  • Python优化代码的技巧分享

    Python优化代码的技巧分享

    Python 是 一种用着很爽的语言,Python也有着很多特性和技巧,可以帮助我们编写更高效、更优雅、更易维护的代码,下面小编为大家整理了一些常用的Python代码优化的简单小技巧,希望对大家有所帮助
    2023-09-09
  • 详解Python中的三元运算

    详解Python中的三元运算

    本文详细介绍了Python中的三元运算符的使用方法和技巧，三元运算也称三目运算，在Python中，三元运算也称为条件表达式，感兴趣的小伙伴可以参考阅读
    2023-04-04
  • python实现bitmap数据结构详解

    python实现bitmap数据结构详解

    bitmap是很常用的数据结构，比如用于Bloom Filter中,下面是使用python实现bitmap数据结构的代码讲解,需要的朋友可以参考下
    2014-02-02
  • python列表删除和多重循环退出原理详解

    python列表删除和多重循环退出原理详解

    这篇文章主要介绍了python列表删除和多重循环退出原理详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-03-03
  • Pygame实现小球躲避实例代码

    Pygame实现小球躲避实例代码

    大家好，本篇文章主要讲的是Pygame实现小球躲避实例代码，感兴趣的同学赶快来看一看吧，对你有帮助的话记得收藏一下，方便下次浏览
    2021-12-12
  • python代码实现TSNE降维数据可视化教程

    python代码实现TSNE降维数据可视化教程

    今天小编就为大家分享一篇python代码实现TSNE降维数据可视化教程，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2020-02-02
  • 详解Pandas与openpyxl库的超强结合

    详解Pandas与openpyxl库的超强结合

    Pandas绝对是Python中处理Excel最快、最好用的库，但是使用 openpyxl 的一些优势是能够轻松地使用样式、条件格式等自定义电子表格，感兴趣的可以了解一下
    2021-09-09
  • python使用openpyxl实现对excel表格相对路径的超链接的创建方式

    python使用openpyxl实现对excel表格相对路径的超链接的创建方式

    这篇文章主要介绍了python使用openpyxl实现对excel表格相对路径的超链接的创建方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
    2024-03-03
  • PyCharm 光标变成黑块的解决方式

    PyCharm 光标变成黑块的解决方式

    这篇文章主要介绍了PyCharm 光标变成黑块的解决方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2021-02-02

最新评论