大貓咪的新玩具大貓咪的新玩具
文章 系列 標籤 分類 連結
大貓咪的新玩具大貓咪的新玩具

目錄

廣告 AD

目錄

Pexpect：自動化的神奇工具，讓自動化變得更簡單

  收錄於 Python
     約 1360 字   預計閱讀 6 分鐘    次閱讀         CC BY-NC 4.0   
Pexpect 是一個設計用來自動化與交互式應用程式（如ssh、ftp、telnet等）互動的工具。透過模擬人類使用終端的方式，Pexpect 能自動輸入指令並回應應用程式的輸出，從而大幅提高任務的自動化和效率

當初看到 Pexpect 就覺得應該蠻好玩的

這次測試下來也真的蠻好用的

廣告 AD

Pexpect 讓你可以透過 Python 跟其他應用程式互動，省去了自己使用 subprocess 和處理輸入輸出的操作，只要透過 library 就可以與之互動，接下來就介紹一些基本的用法。

安裝上直接使用 pip install 就可以了。

shell

pip install pexpect

一般的 Linux 環境下，直接使用 pexpect.spawn。

python

import pexpect

child = pexpect.spawn('ssh -T [email protected]')

目前的版本支援 Windows，但由於 pexpect.spawn 底層使用 Linux 的 pseudoterminals，在 Windows 使用 pexpect.spawn 會出現以下的錯誤，Windows 則是可以使用 pexpect.popen_spawn.PopenSpawn 來運行。

AttributeError: module ‘pexpect’ has no attribute ‘spawn’

Pexcept Document - Pexcept on Windows

python

import pexpect.popen_spawn

child = pexpect.popen_spawn.PopenSpawn('ssh -T [email protected]')

我們可以用 expect 來等待應用程式回傳給定的字串，給定的字串可以使用正規表達式

並在之後透過 before 和 after 來獲取應用程式的字串

  • before: 回傳 expect 配對到的字串之前的字串
  • after: 回傳 expect 配對到的字串

下面我們用檢測是否能正常用 ssh 登入 github 的指令當作範例，正常輸出為：

text

Hi bigcat! You've successfully authenticated, but GitHub does not provide shell access.

我們抓取 success 的這個字串，你會看到 before 印出了 success 之前的字串，而 after 則是 success。

python

import pexpect

child = pexpect.spawn('ssh -T [email protected]')
child.expect("success")
print(f"Before: {child.before.decode()}")
print(f"After: {child.after.decode()}")

text

Before: Hi bigcat! You've 
After: success

緊接著我們繼續抓取剩餘字串，透過正規表達式，我們抓取了逗號(,)開頭，句點(.)結尾的字串，before 包含著 success 之後到逗號之前的字串，而 after 就是我們要抓取的字串。

python

import pexpect

child = pexpect.spawn('ssh -T [email protected]')
child.expect("success")
print(f"Before: {child.before.decode()}")
print(f"After: {child.after.decode()}")

child.expect(r",.+\.")
print(f"Before: {child.before.decode()}")
print(f"After: {child.after.decode()}")

text

Before: Hi bigcat! You've 
After: success
Before: fully authenticated
After: , but GitHub does not provide shell access.

如果我們要直接抓取到 EOF 的所有字串，可以直接傳入 pexpect.EOF，則 after 就會是 pexpect.EOF 的變數，before 存著所有 EOF 之前的字串，包含換行，因此輸出才會空一行。

python

import pexpect

child = pexpect.spawn('ssh -T [email protected]')
child.expect(pexpect.EOF)
print(f"Before: {child.before.decode()}")
print(f"After: {child.after}")

text

Before: Hi bigcat! You've successfully authenticated, but GitHub does not provide shell access.

After: <class 'pexpect.exceptions.EOF'>

我們也可以設定 timeout 來設定等待的最長時間，單位為秒，超過會拋出 pexpect.TIMEOUT。

python

import pexpect

child = pexpect.spawn('ssh -T [email protected]')
try:
    child.expect(pexpect.EOF, timeout=10)
except pexpect.TIMEOUT:
    print("Timeout")

text

Timeout

另外，我們也可以傳入 list，裡面是各種可能會出現的字串、正規表達式，或是 pexpect.EOF 和 pexpect.TIMEOUT，最後看輸出是符合哪一個，回傳符合的 index，我們可以透過這個 index 來知道應用程式目前的狀態，並依據狀態做出相對的處理。

python

import pexpect

child = pexpect.spawn('ssh -T [email protected]')
idx = child.expect(["success", "denied", pexpect.TIMEOUT], timeout=10)
match idx:
    case 0:
        print("Login Success")
    case 1:
        print("Login Failed")
    case 2:
        print("Connection Failed")

text

Login Success

最後，這裡有一點要提醒，在正規表達式的使用上，如果有使用到 * 和 +，因為 expect 會採用最小符合原則，因此只要字串有符合，則會馬上回傳，並不會看後面的字串，我們看例子比較好懂。

持續上面的例子，如果使用 expect(r".*") 則甚麼都抓不到。

python

import pexpect

child = pexpect.spawn('ssh -T [email protected]')
child.expect(r".*")
print(child.after.decode() == "")

text

True

send / sendline

expect 讓我們知道如何抓取應用程式的輸出，send 和 sendline 讓我們可以傳輸入給應用程式，達到跟應用程式互動的功能，sendline 就是多傳送換行的 send。

我們一樣用之前的例子，只是如果電腦上沒有儲存 github 的 fingerprint，則會詢問使用者是否要繼續連線，這時候就要輸入 yes 來繼續連線，我們使用 sendline 來達成這個動作。但以下的範例在 Windows 上測試會有問題，建議在 Linux 上執行。

python

import pexpect

child = pexpect.spawn('ssh -T [email protected]')
while True:
    idx = child.expect(["success", "denied", r"fingerprint.+\?", pexpect.TIMEOUT], timeout=10)
    match idx:
        case 0:
            print("Login Success")
            break
        case 1:
            print("Login Failed")
            break
        case 2:
            print("Store fingerprint")
            child.sendline("yes")
        case 3:
            print("Connection Failed")
            break

text

Store fingerprint
Login Success

有時候你會想看應用程式的輸出或是為了 debug，你可以使用透過設定 logfile 成 sys.out.buffer，這樣輸出就會顯示在 stdout 上面了。

python

import pexpect
import sys

child = pexpect.spawn('ssh -T [email protected]')
child.logfile = sys.stdout.buffer
child.expect("success")

text

Hi bigcat! You've successfully authenticated, but GitHub does not provide shell access.

也可以寫到檔案上，只要將 logfile 設定成檔案就好，記得要用 binary mode。

python

import pexpect

child = pexpect.spawn('ssh -T [email protected]')
child.logfile = open("output.log", "wb")
child.expect("success")
child.logfile.close()

Pexpect 是一個設計用來自動化與交互式應用程式（如ssh、ftp、telnet等）互動的工具。透過模擬人類使用終端的方式，Pexpect 能自動輸入指令並回應應用程式的輸出，從而大幅提高任務的自動化和效率

WAT：檢視 Python 物件，一款強大的 Debug 工具

Pexpect 是一個設計用來自動化與交互式應用程式（如ssh、ftp、telnet等）互動的工具。透過模擬人類使用終端的方式，Pexpect 能自動輸入指令並回應應用程式的輸出，從而大幅提高任務的自動化和效率

py-spy: 找出 Python 程式性能瓶頸

Pexpect 是一個設計用來自動化與交互式應用程式（如ssh、ftp、telnet等）互動的工具。透過模擬人類使用終端的方式，Pexpect 能自動輸入指令並回應應用程式的輸出，從而大幅提高任務的自動化和效率

加速大型語言模型：運用 ONNX 格式推理和 KV Cache 的使用

Pexpect 是一個設計用來自動化與交互式應用程式（如ssh、ftp、telnet等）互動的工具。透過模擬人類使用終端的方式，Pexpect 能自動輸入指令並回應應用程式的輸出，從而大幅提高任務的自動化和效率

掌握 Python Decorators：從基礎到進階的各種寫法

Pexpect 是一個設計用來自動化與交互式應用程式（如ssh、ftp、telnet等）互動的工具。透過模擬人類使用終端的方式，Pexpect 能自動輸入指令並回應應用程式的輸出，從而大幅提高任務的自動化和效率

使用 Intel NPU 加速推理 LLM Model - TinyLlama

廣告 AD
更新於 2025-01-04
CC BY-NC 4.0
閱讀原始文檔
 Python
 | 主頁