快捷導(dǎo)航

Python?命令行?prompt_toolkit?庫(kù)詳解

更新時(shí)間：2022年01月11日 08:31:44 作者：ITROOKIEIS

prompt_toolkit 是一個(gè)用于構(gòu)建強(qiáng)大交互式命令行的 Python 工具庫(kù)。接下來(lái)通過(guò)本文給大家介紹Python?命令行?prompt_toolkit?庫(kù)的相關(guān)知識(shí)，感興趣的朋友一起看看吧

一. 使用 Bash 下常用快捷鍵

想必很多開(kāi)發(fā)者在創(chuàng)建交互式命令行工具時(shí)，使用最多的還是 input 和 raw_input 。比如下面的代碼讀取用戶(hù)輸入數(shù)據(jù)，并進(jìn)行打印：

while True:
    # user_input = input('>')
    user_input = raw_input('>')
    print(user_input)
    if user_input.strip().lower() == 'exit':
        break

上述程序在 Linux 環(huán)境下運(yùn)行時(shí)，我們將無(wú)法使用任何的 Linux 快捷鍵，甚至在輸入錯(cuò)誤時(shí)，按退格刪除內(nèi)容都會(huì)出現(xiàn)問(wèn)題：

下面，我們使用 prompt_toolkit 模塊中的 prompt 函數(shù)改寫(xiě)上述程序：

from __future__ import print_function
from prompt_toolkit import prompt
while True:
    user_input = prompt(u'>>')
    print(user_input)

運(yùn)行新的程序，你會(huì)發(fā)現(xiàn)，不僅可以實(shí)現(xiàn)退格刪除，而且可以使用 Bash 下常用的快捷鍵：Ctrl + a 跳轉(zhuǎn)到開(kāi)頭、Ctrl + e 跳轉(zhuǎn)到末尾、Ctrl + k 刪除光標(biāo)到末尾的內(nèi)容。

二. 實(shí)現(xiàn)查找歷史命令

在 Bash 下，我們可以使用方向鍵中的 ↑ 和 ↓ 查看歷史輸入，或者使用 Ctrl + r 搜索歷史命令：

在 Python 打造的交互式命令行中，使用 prompt_toolkit.history 我們可以很容易實(shí)現(xiàn)查找歷史：

from __future__ import print_function
from __future__ import unicode_literals
from prompt_toolkit import prompt
from prompt_toolkit.history import FileHistory
 
while True:
    user_input = prompt('>>>', history=FileHistory('history.txt'))
    print(user_input)

運(yùn)行結(jié)果：

上述歷史輸入將被保存至當(dāng)前目錄下的 history.txt 文件中，后續(xù)就可以使用查看或搜索歷史命令了~

三. 根據(jù)歷史輸入自動(dòng)提示

在上面是示例中我們實(shí)現(xiàn)了查看或搜索歷史輸入的功能，其實(shí)我們還可以更加充分地利用 history.txt 中記載的歷史輸入，在用戶(hù)輸入時(shí)進(jìn)行提示。實(shí)現(xiàn)此功能只需要在調(diào)用 prompt 函數(shù)時(shí)指定 auto_suggest 的參數(shù)即可：

from __future__ import print_function
from __future__ import unicode_literals
from prompt_toolkit import prompt
from prompt_toolkit.history import FileHistory
from prompt_toolkit.auto_suggest import AutoSuggestFromHistory
 
while True:
    user_input = prompt('>>>', history=FileHistory('history.txt'), 
                        auto_suggest=AutoSuggestFromHistory())
    if user_input.strip().lower() == 'exit':
        break
    print(user_input)

prompt_toolkit 將以暗色字體顯示匹配的歷史輸入：

四. 實(shí)現(xiàn)輸入的自動(dòng)補(bǔ)全

所謂自動(dòng)補(bǔ)全，即用戶(hù)輸入了關(guān)鍵字的一部分，我們的交互式程序能夠根據(jù)已有的輸入進(jìn)行提示，用戶(hù)可以使用 Tab 鍵補(bǔ)全選擇提示的內(nèi)容。以上功能，prompt_toolkit 提供了名為 WorldCompleter 的類(lèi)來(lái)幫助我們實(shí)現(xiàn)。下面我們來(lái)模仿 MySQL 客戶(hù)端的提示功能：

from __future__ import print_function
from __future__ import unicode_literals
from prompt_toolkit import prompt
from prompt_toolkit.history import FileHistory
from prompt_toolkit.auto_suggest import AutoSuggestFromHistory
from prompt_toolkit.contrib.completers import WordCompleter
 
SQLCompleter = WordCompleter(['select', 'from', 'insert', 'update', 'delete'
                              'drop'], ignore_case=True)
 
while True:
    user_input = prompt('SQL>', history=FileHistory('history.txt'), 
                        auto_suggest=AutoSuggestFromHistory(), 
                        completer=SQLCompleter)
    if user_input.strip().lower() == 'exit':
        break
    print(user_input)

到此這篇關(guān)于Python 命令行 - prompt_toolkit 庫(kù)的文章就介紹到這了,更多相關(guān)Python prompt_toolkit 庫(kù)內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: