在Python中創(chuàng)建命令行界面的優(yōu)秀方式
我們先給大家介紹什么是命令行界面(CLI):
命令行界面或命令語言解釋器,也稱為命令行用戶界面、控制臺用戶界面和字符用戶界面,是一種與計算機程序交互的方式,用戶以連續(xù)的文本行形式向程序發(fā)出命令。 |
通過創(chuàng)建命令行界面(CLI),可以使程序功能強大并具有交互性。CLI允許您接受命令行參數(shù)(操作系統(tǒng)命令行中程序名稱后面的信息),以便向程序添加其他特性,使代碼易于使用和靈活。根據(jù)程序的不同,這些參數(shù)可用于添加其他特性,如查看幫助文檔、指定輸出文件或啟用測試特性,這些特性在正常使用時可能會出現(xiàn)問題。
當我們剛開始用Python編程時,我們大多數(shù)只收集用戶輸入,交互方式是這樣的:
- def main():
- first = input(“Enter your first name:”)
- last = input(“Enter your last name:”)
- print(first + ' ' + last)
雖然這段代碼對于簡單的腳本來說很好,但是這段代碼不夠靈活。當用戶運行這個程序時,它們被限制為一組定義的規(guī)則。例如,如果我想將輸出記錄到文本文件中,該怎么辦?作為一個用戶,您可以創(chuàng)建一個命令行界面來提供這些問題的解決方案。
最佳方式" src="http://p3.pstatp.com/large/pgc-image/5b4f9a79a6624f63b433053d0c17c16c" _fcksavedurl="http://p3.pstatp.com/large/pgc-image/5b4f9a79a6624f63b433053d0c17c16c" _fcksavedurl="http://p3.pstatp.com/large/pgc-image/5b4f9a79a6624f63b433053d0c17c16c" width="640" height="387">
重要的注意事項
在創(chuàng)建CLI時,重要的是要考慮以下幾點:
- 必需參數(shù):為了程序的運行,哪些參數(shù)是絕對必需的?
- 文檔:寫出每個選項和參數(shù)的函數(shù)是很重要的,這樣新用戶就可以知道你的程序是如何工作的。
- 處理錯誤情況:讓用戶確切地知道哪里出了問題
- 運行時狀態(tài):如果任務(wù)沒有立即完成,您應(yīng)該打印出當前的進度
使用argparse讀取參數(shù)
Argparse是一個用于解析命令行參數(shù)的Python標準庫模塊。作為程序員,您可以定義要接受的參數(shù),而argparse將知道如何從sys中解析這些參數(shù)。當用戶給程序提供無效參數(shù)時,Argparse還會自動生成幫助和使用消息,并輸出錯誤。它使用起來非常簡單,并且可以很容易地編寫直觀的CLI。
首先,創(chuàng)建一個名為test_cl .py的新文件,導入模塊并初始化一個新的解析器:
- import argparse
- parser = argparse.ArgumentParser()
- parser.parse_args()
現(xiàn)在使用--help選項運行代碼:
- python3 test_cli.py --help
你應(yīng)該會收到一個很好的默認幫助信息,像這樣:
- usage: test_cli.py [-h]
- optional arguments:
- -h, --help show this help message and exit
祝賀您創(chuàng)建了第一個命令行界面!
現(xiàn)在讓我們添加一個歡迎消息,簡要地讓您的用戶知道這個程序是做什么的:
- welcome = "Practicing creating interactive command-line interfaces"
- parser = argparse.ArgumentParser(description=welcome)
- parser.parse_args()
現(xiàn)在用-h標志運行程序。你應(yīng)該可以看到你的歡迎信息。
添加參數(shù)
假設(shè)我們正在編寫一個程序來爬一個網(wǎng)頁。我們可能需要的一些參數(shù)是網(wǎng)頁的域-domain或-d,日志輸出到一個輸出文件-ofile或-o的選項,可能還需要輸出到控制臺的特定行數(shù)-lines或-l的選項。對于這個例子,我們將域參數(shù)設(shè)置為必需的,而ofile和lines參數(shù)將是可選的。
通過使用.add_argument,我們可以很容易地向argparse CLI添加額外的參數(shù),該參數(shù)允許我們定義使用細節(jié)。我們可以添加必要的參數(shù)——域,如:
- parser.add_argument('--domain', '-d', required=True, help='domain name of the website you want to scrape. i.e. “https://ahadsheriff.com"')
現(xiàn)在運行帶有-h參數(shù)的程序,查看您編寫的文檔!
因為——domain是一個必需的參數(shù),嘗試運行不帶任何標志的程序,您將收到以下消息:
- usage: test_cli.py [-h] --domain DOMAIN
- test_cli.py: error: the following arguments are required: --domain/-d
成功了!
現(xiàn)在使用argparse添加額外的參數(shù)。如果沒有指定需要哪些參數(shù),argparse將假定它們是可選的。你也可以設(shè)置參數(shù)的類型,對于——lines,我們?nèi)∫粋€整數(shù)。您還可以為.add_argument設(shè)置其他有用的選項——比如action=
- parser.add_argument('--ofile', '-o', help='define output file to save results of stdout. i.e. "output.txt"')
- parser.add_argument('--lines', '-l', help='number of lines of output to print to the console"', type=int)
現(xiàn)在測試您的代碼,以確保一切正常運行。一種簡單的方法是將參數(shù)的值存儲為變量,然后打印這些值。
- args = parser.parse_args()
- domain = args.domain
- ofile = args.ofile
- lines = args.lines
- print("domain:", domain)
- print("output file:", ofile)
- print("lines:", lines)
原文:
https://medium.com/@ahadsheriff/the-best-way-to-make-command-line-interfaces-in-python-e00e8b9d10c9