python之pexpect實現(xiàn)自動交互的例子

Pexpect 是 Expect 語言的一個 Python 實現(xiàn)，是一個用來啟動子程序，并使用正則表達式對程序輸出做出特定響應，以此實現(xiàn)與其自動交互的 Python 模塊。 Pexpect 的使用范圍很廣，可以用來實現(xiàn)與 ssh、ftp 、telnet 等程序的自動交互；可以用來自動復制軟件安裝包并在不同機器自動安裝；還可以用來實現(xiàn)軟件測試中與命令行交互的自動化。

整體來說大致的流程包括：

運行程序

程序要求人的判斷和輸入

Expect 通過關(guān)鍵字匹配

根據(jù)關(guān)鍵字向程序發(fā)送符合的字符

基本使用流程

pexpect 的使用說來說去，就是圍繞3個關(guān)鍵命令做操作：

首先用 spawn 來執(zhí)行一個程序

然后用 expect 來等待指定的關(guān)鍵字，這個關(guān)鍵字是被執(zhí)行的程序打印到標準輸出上面的

最后當發(fā)現(xiàn)這個關(guān)鍵字以后，根據(jù)關(guān)鍵字用 send 方法來發(fā)送字符串給這個程序

第一步只需要做一次，但在程序中會不停的循環(huán)第二、三步來一步一步的完成整個工作。掌握這個概念之后 pexpect 的使用就很容易了。當然 pexpect 不會只有這 3 個方法，實際上還有很多外圍的其他方法，

spawn() - 執(zhí)行程序

spawn() 方法用來執(zhí)行一個程序，它返回這個程序的操作句柄，以后可以通過操作這個句柄來對這個程序進行操作，比如：

process = pexpect.spawn('ftp sw-tftp')

上面 spawn() 中的字符串就是要執(zhí)行的程序，這里我們打開一個到 sw-tftp 服務器的 ftp 連接。 spawn() 中的第一個元素就是要執(zhí)行的命令，除此之外還可以指定一些其他參數(shù)，比如： pexpect.spawn('ftp sw-tftp', timeout=60) 就指定了超時時間，這些具體的會在后面講解。

process 就是 spawn() 的程序操作句柄了，之后對這個程序的所有操作都是基于這個句柄的，所以它可以說是最重要的部分。

注意: spawn() ，或者說 pexpect 并不會轉(zhuǎn)譯任何特殊字符 比如 | * 字符在Linux的shell中有特殊含義，但是在 pexpect 中不會轉(zhuǎn)譯它們，如果在 linux 系統(tǒng)中想使用這些符號的正確含義就必須加上 shell 來運行，這是很容易犯的一個錯誤。

正確的方式：

process = pexpect.spawn('/bin/bash –c "ls –l | grep LOG > log_list.txt"')

expect() - 關(guān)鍵字匹配

當 spawn() 啟動了一個程序并返回程序控制句柄后，就可以用 expect() 方法來等待指定的關(guān)鍵字了。它最后會返回 0 表示匹配到了所需的關(guān)鍵字，如果后面的匹配關(guān)鍵字是一個列表的話，就會返回一個數(shù)字表示匹配到了列表中第幾個關(guān)鍵字，從 0 開始計算。

send() - 發(fā)送關(guān)鍵字

send() 作為3個關(guān)鍵操作之一，用來向程序發(fā)送指定的字符串，它的使用沒什么特殊的地方，比如：

process.expect("ftp>")
process.send("by\n")

這個方法會返回發(fā)送字符的數(shù)量。

sendline() - 發(fā)送帶回車符的字符串

sendline() 和 send() 唯一的區(qū)別就是在發(fā)送的字符串后面加上了回車換行符，這也使它們用在了不同的地方：

只需要發(fā)送字符就可以的話用send()

如果發(fā)送字符后還要回車的話，就用 sendline()

它也會返回發(fā)送的字符數(shù)量

sendcontrol() - 發(fā)送控制信號

sendcontrol() 向子程序發(fā)送控制字符，比如 <kbd>ctrl+C</kbd> 或者 <kbd>ctrl+D</kbd> 之類的，比如你要向子程序發(fā)送 <kbd>ctrl+G</kbd>，那么就這樣寫：

process.sendcontrol('g')

簡單實例：

command = 'ssh '+username+'@'+host 
child = pexpect.spawn(command) 
ret = child.expect([pexpect.TIMEOUT,'Are you sure you want to continue connecting','[P|p]assword']+PROMPT) 
if ret == 0: 
  print('[-] Error Connecting') 
  return 
if ret == 1: 
  child.sendline('yes') 
  ret = child.expect([pexpect.TIMEOUT,'[p|P]assword']) 
  if ret == 0: 
    print('[-] Error Connecting') 
    return 
  if ret == 1: 
    send_command(password) 
    return 
if ret == 2: 
  send_command(password) 
  return 
return child

注：針對ssh遠程登錄，pexpect又派生出了pxssh類，在ssh會話操作上再做一層封裝

其常用方法是：

login() #建立ssh連接

logout() #斷開ssh連接

prompt() #等待系統(tǒng)提示符，用于等待命令執(zhí)行結(jié)束

詳情請戳：https://www.jb51.net/article/156232.htm

以上這篇python之pexpect實現(xiàn)自動交互的例子就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持億速云。