Python Pexpect库的简单使用方法

2023-08-29 13:00:05 326

简介

最近需要远程操作一个服务器并执行该服务器上的一个python脚本，查到可以使用Pexpect这个库。记录一下。

什么是Pexpect？Pexpect能够产生子应用程序，并控制他们，并能够通过期望模式对子应用的输出做出反应。Pexpect允许你的脚本产生子应用、控制他们像一个人类在输入命令一样。

Pexpect使用在自动交互的应用，例如SSH、SFTP、PASSWD、TELNET。它可以被应用在使用自动设置脚本为不同的服务器自动地重复的安装软件包。也可以被应用在自动的软件测试。

Pexpect的主要特点是需要Python的基本库pty，这个库只有在类Unix系统上才有

Pexpect关于SSH的使用

注：测试，我们直接用虚拟机本机ssh本机

环境

1.win10物理机
2.VmwareCentos虚拟机
3.Xshell
4.虚拟机python安装pexpect:pipinstallpexpect

在虚拟机创建一个python文件

#-*-coding:UTF-8-*-
importpexpect
#定义ssh连接
defssh(user,host,password,command):
#创建子应用，命令是ssh-lroot127.0.0.1python/home/python/test.py
child=pexpect.spawn('ssh-l%s%s%s'%(user,host,command))
#期待开启的子程序的显示，子程序的不同显示会匹配到不同key然后我们定义不同的操作
#0：连接超时
#1：ssh有时候提示你是否确认连接
#2：提示输入密码
#3：匹配到#号，表示命令已经执行完毕。没用到
i=child.expect([pexpect.TIMEOUT,'Areyousureyouwanttocontinueconnecting','password:',r"([^-]>|#)"])
#如果登录超时，renturnnone
ifi==0:#Timeout
print"Timeout"
returnNone
#提示是否确认连接
ifi==1:
child.sendline('yes')#我们输入yes
child.expect('password:')#输入yes后子程序应该提示输入密码，我们再次期待password
i=child.expect([pexpect.TIMEOUT,'password:'])
#超时
ifi==0:#Timeout
returnNone
#不考虑其他情况，走到此处时，要么timeout已经return，要么等待输入密码
#输入密码
child.sendline(password)
#返回子程序
returnchild
if__name__=='__main__':
try:
#配置数据
host='127.0.0.1'
user="root"
password='********'
command='python/home/python/test.py'
#command="ls-l"
child=ssh(user,host,password,command)
#这句是将子程序的命令行拉到末端
test=child.expect(pexpect.EOF)
#child中before就是我们要的数据，有时候还会在after中
printchild.before
printchild.after
exceptException,e:
printstr(e)
#最终的显示结果是test.py中打印的hahaha结果，
[root@localhostpython]#pythontest_pexpect.py

hahaha

我们尝试一下开两个虚拟机的情况

上面的代码只需要更改ipuserpassword即可

#ip192.168.233.133
#userroot
#在另一台虚拟机的相同位置创建/home/pyhton/test.py内容如下
if__name__=="__main__":
print"anothervirualmachinehahaha"
#打印结果
[root@localhostpython]#pythontest3.py

anothervirualmachinehahaha

Pexpect关于SFTP的使用

与ssh相同，就是使用python在当前机器上输入sftpip然后期望结果，输入密码，并发送get下载文件即可。

注：使用的时候发现一点注意：在每次执行sendline之前都需要重新期望一下当前的sftp>，或者在每次输入sendline之后重新期望一下sftp>。也就是期望到这行，否则输入的命令都没有反应，我理解是远程连接的服务器有输出时候当前的位置可能不在sftp>这里所以在sendline的任何东西都是无意义的。如果这个解释不对望高人指点一下，

#--*--coding:utf-8--*--
importpexpect
importos
importtime
defsftp(ip,password,command):
#创建子应用
child=pexpect.spawn("sftp%s"%(ip))
i=child.expect([pexpect.TIMEOUT,'password:'])
#超时
ifi==0:
print"Timeout"
returnNone
#准备输入密码
ifi==1:
#输入密码
child.sendline(password)
j=child.expect([pexpect.TIMEOUT,'sftp>'])
#超时
ifj==0:
print"Timeout"
returnNone
#匹配到进入sftp命令模式
ifj==1:
print'Beforesftpgetcommand'
printchild.before
print"-----------------"
#发送命令
child.sendline(command)
child.expect(['sftp>'])
print"Aftersftpgetcommand"
printchild.before
print"-----------------"
child.sendline("bye")
#child.expect(['sftp>'])
print"Aftersftpbye"
printchild.before
print"-----------------"
printchild.after
returnchild
if__name__=='__main__':
ip="192.168.233.133"
command="get/home/python/test.txt"
password="********"
child=sftp(ip,password,command)
printchild.before
printchild.after
ifos.path.exists("./test.txt"):
print"Makesuretransfersuccessfully"
else:
print"Cannotfindthetransferfile"

#----------------------------结果-----------------------------------------------
'''
Beforesftpgetcommand

Connectedto192.168.233.133.

-----------------
Aftersftpgetcommand
get/home/python/test.txt
Fetching/home/python/test.txttotest.txt
/home/python/test.txt100%7325.2KB/s00:00

-----------------
Aftersftpbye
get/home/python/test.txt
Fetching/home/python/test.txttotest.txt
/home/python/test.txt100%7325.2KB/s00:00

-----------------
sftp>
get/home/python/test.txt
Fetching/home/python/test.txttotest.txt
/home/python/test.txt100%7325.2KB/s00:00

sftp>
Makesuretransfersuccessfully
'''

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持毛票票。

Python Pexpect库的简单使用方法

热门推荐

随机推荐