Pexpect簡(jiǎn)介

在講解Pexpect之前，我們需要先了解一下Expect這個(gè)腳本語(yǔ)言，它是由TCL語(yǔ)言實(shí)現(xiàn)的，主要用于人機(jī)交互式對(duì)話的自動(dòng)化控制，可以用來(lái)完成ssh、ftp、telnet等命令行程序的自動(dòng)化交互。Pexpect其實(shí)就是一個(gè)用Python語(yǔ)言實(shí)現(xiàn)的類Expect功能的模塊，通過(guò)它就可以在Python中完成Expect所完成的功能。

Pexpect的基本工作流程，基本可以分為以下三個(gè)步驟：

首先用spawn來(lái)執(zhí)行一個(gè)程序； 然后用expect方法來(lái)等待指定的關(guān)鍵字，這個(gè)關(guān)鍵字是被執(zhí)行的程序打印到標(biāo)準(zhǔn)輸出上面的； 最后當(dāng)發(fā)現(xiàn)這個(gè)關(guān)鍵字以后，使用send/sendline方法發(fā)送字符串給這個(gè)程序。

通常在程序中第一步只需要做一次，第二步和第三步會(huì)不停的循環(huán)來(lái)完成整個(gè)工作。當(dāng)然在Pexpect中還有很多其他方法，編寫程序時(shí)可以根據(jù)自己的需求選擇使用。

Pexpect APIspawn類

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):

通過(guò)spawn()方法用來(lái)執(zhí)行一個(gè)程序，返回程序的操作句柄，后續(xù)就可以通過(guò)操作句柄來(lái)與這個(gè)程序進(jìn)行交互了。

# 子程序退出時(shí)會(huì)引發(fā)pexpect.EOF異常，即如果捕捉到pexpect.EOF則說(shuō)明子程序已退出process = pexpect.spawn(’ls -l’)process.expect(pexpect.EOF)result = process.before.decode()print(result)

command參數(shù)并不支持字符的特殊含義（比如管道符、通配符、重定向符等），在Linux系統(tǒng)中如果想使用這些符號(hào)的特殊含義就必須加上shell來(lái)運(yùn)行。

# 示例一process = pexpect.spawn(’bash -c 'ls -l | wc -l'’)process.expect(pexpect.EOF)# 示例二# 第一個(gè)參數(shù)為主程序，而args列表里的元素是主程序的參數(shù)process = pexpect.spawn(’bash’, [’-c’, ’ls -l | wc -l’])process.expect(pexpect.EOF)expect()方法

當(dāng)使用spawn()方法啟動(dòng)了一個(gè)程序并返回程序控制句柄后，就可以使用expect()方法來(lái)等待指定的關(guān)鍵字了。關(guān)鍵字可以是字符串、正則表達(dá)式、EOF、TIMEOUT或者以上類型組成的列表，用來(lái)匹配子程序返回的結(jié)果。如果只提供字符串等非列表，則匹配成功后返回0，如果提供列表，則返回匹配成功的列表元素的索引，匹配失敗會(huì)拋出異常。

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

before/after/match：當(dāng)expect()匹配到關(guān)鍵字之后，系統(tǒng)會(huì)自動(dòng)給這三個(gè)變量賦值，通過(guò)這三個(gè)變量可以獲取子程序運(yùn)行輸出。

before：保存了到匹配到關(guān)鍵字為止，緩存里面已有的所有數(shù)據(jù)。也就是說(shuō)如果緩存里緩存了100個(gè)字符的時(shí)候匹配到了關(guān)鍵字，那before就是除了匹配到的關(guān)鍵字之外的所有字符。 after：保存了匹配到了關(guān)鍵字。 match：保存的是匹配到的正則表達(dá)式的實(shí)例，和上面的after相比一個(gè)是匹配到的字符串，一個(gè)是匹配到的正則表達(dá)式實(shí)例。

process = pexpect.spawn(’ls -l’)process.expect(’expect’)print(process.before.decode())print(process.after.decode())print(process.match)

如果expect()過(guò)程中發(fā)生錯(cuò)誤，那么before保存到目前為止緩存里的所有數(shù)據(jù)，after和match都是None。

如果沒匹配成功則會(huì)拋出異常，可以通過(guò)匹配異常，讓異常不在終端顯示。

process = pexpect.spawn(’ls -l’)# 返回0表示匹配成功，返回1和2表示匹配到了異常index = process.expect([’expect’, pexpect.EOF, pexpect.TIMEOUT])print(index)send()/sendline()方法

sendline()和send()的區(qū)別就是sendline()發(fā)送的是帶回車符的字符串。

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

Pexpect還提供了很多其他方法，這里不再詳細(xì)闡述，使用時(shí)可參考其官方文檔。

interact()方法

interact()表示將終端控制權(quán)交給用戶（或者說(shuō)將標(biāo)準(zhǔn)輸入交給用戶）。通常情況下Pexpect會(huì)接管所有的輸入和輸出，如果需要用戶介入完成部分工作的時(shí)候，interact()就派上用場(chǎng)了。

# 讓出控制權(quán)給用戶process.interact()# 通過(guò)設(shè)置escape_character的值定義返回碼，當(dāng)用戶輸入此值后，會(huì)將控制權(quán)重新交給pexpectprocess.interact(escape_character=’x1d’, input_filter=None, output_filter=None)應(yīng)用示例

接下來(lái)通過(guò)SSH連接遠(yuǎn)程服務(wù)器的示例來(lái)體驗(yàn)下Pexpect的使用方法。

#!/usr/bin/env python3.6#-*- coding:utf-8 -*-import pexpectdef 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模塊的使用的詳細(xì)內(nèi)容，更多關(guān)于python Pexpect模塊的資料請(qǐng)關(guān)注好吧啦網(wǎng)其它相關(guān)文章！