python之pexpect實現自動交互的例子

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

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

運行程序

程序要求人的判斷和輸入

Expect 通過關鍵字匹配

根據關鍵字向程序發送符合的字符

基本使用流程

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

首先用 spawn 來執行一個程序

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

最后當發現這個關鍵字以后，根據關鍵字用 send 方法來發送字符串給這個程序

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

spawn() - 執行程序

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

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

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

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

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

正確的方式：

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

expect() - 關鍵字匹配

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

send() - 發送關鍵字

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

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

這個方法會返回發送字符的數量。

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

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

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

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

它也會返回發送的字符數量

sendcontrol() - 發送控制信號

sendcontrol() 向子程序發送控制字符，比如 <kbd>ctrl+C</kbd> 或者 <kbd>ctrl+D</kbd> 之類的，比如你要向子程序發送 <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() #等待系統提示符，用于等待命令執行結束

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

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