利用Python內(nèi)置庫實現(xiàn)創(chuàng)建命令行應(yīng)用程序

更新時間：2022年06月07日 08:17:53 作者：MousevsPython

Python?有一個叫做argparse的內(nèi)置庫，可以用它來創(chuàng)建一個命令行界面。本文將詳解如何利用argparse實現(xiàn)創(chuàng)建一個命令行應(yīng)用程序，需要的可以參考一下

介紹

當創(chuàng)建一個應(yīng)用程序時，通常希望能夠告訴你的應(yīng)用程序如何做某事。有兩種流行的方法來完成這項任務(wù)，你可以讓應(yīng)用程序接受命令行參數(shù)，或者創(chuàng)建一個圖形化的用戶接口。有些應(yīng)用程序兩者都支持。

當你需要在服務(wù)器上運行你的代碼時，命令行接口很有幫助。大多數(shù)服務(wù)器沒有圖形化界面，特別當它們是Linux服務(wù)器時。在這種情況下，即使你想運行圖形用戶界面，你也可能無法運行。

Python 有一個叫做 argparse 的內(nèi)置庫，可以用它來創(chuàng)建一個命令行界面。在這篇文章中，你將了解到以下內(nèi)容。

解析實參
創(chuàng)建有用的信息
添加別名
使用相互排斥的參數(shù)
創(chuàng)建一個簡單的搜索工具

argparse模塊的內(nèi)容比本文要介紹的多得多。如果你想知道更多關(guān)于它的信息，你可以查看文檔。

現(xiàn)在是時候開始從命令行中解析參數(shù)了。

解析參數(shù)

在學習如何使用 argparse 之前，最好知道還有一種方法可以向 Python 腳本傳遞參數(shù)。你可以向 Python 腳本傳遞任何參數(shù)，并通過使用 sys 模塊訪問這些參數(shù)。

為了了解它是如何工作的，創(chuàng)建一個名為 sys_args.py 的文件，并在其中輸入以下代碼：

# sys_args.py
   import sys
   def main():
       print('You passed the following arguments:')
       print(sys.argv)
   if __name__ == '__main__':
       main()

這段代碼導入sys并打印出sys.argv中的任何內(nèi)容。argv屬性包含了所有傳遞給腳本的東西的列表，其中第一項是腳本本身。

下面是一個例子，說明當你運行這段代碼和幾個參數(shù)的時候會發(fā)生什么。

$ python3 sys_args.py --s 45
   You passed the following arguments:
   ['sys_args.py', '--s', '45']

使用sys.argv的問題是，你無法控制可以傳遞給你的應(yīng)用程序的參數(shù)。

不能忽略參數(shù)
不能創(chuàng)建默認參數(shù)
無法判斷什么是有效參數(shù)

這就是為什么使用 argparse 是使用 Python 標準庫的方法。argparse 模塊是非常強大和有用的。讓我們想一想，一個命令行應(yīng)用程序所遵循的常見過程。

pass：傳入一個文件
do：在你的程序中對該文件做一些處理
output ：輸出結(jié)果

下面是一個關(guān)于如何工作的通用例子。繼續(xù)創(chuàng)建file_parser.py并添加以下代碼。

# file_parser.py
   import argparse
   def file_parser(input_file, output_file=''):
       print(f'Processing {input_file}')
       print('Finished processing')
       if output_file:
           print(f'Creating {output_file}')
   def main():
       parser = argparse.ArgumentParser('File parser')
       parser.add_argument('--infile', help='Input file')
       parser.add_argument('--out', help='Output file')
       args = parser.parse_args()
       if args.infile:
           file_parser(args.infile, args.out)
   if __name__ == '__main__':
       main()

file_parser()函數(shù)是進行解析的邏輯所在。在這個例子中，它只接收一個文件名，并將其打印出來。output_file參數(shù)的默認值是一個空字符串。

程序的重點在 main()中。在這里你創(chuàng)建了一個 argparse.ArgumentParser()的實例，并給你的解析器起了一個名字。然后你添加兩個參數(shù)， --infile和 --out。為了使用這個解析器，你需要調(diào)用 parse_args()，它將返回傳遞給你的程序的任何有效參數(shù)。最后，你要檢查用戶是否使用了 --infile 標志。如果他們使用了，那么你就運行 file_parser()。

下面是你如何在你的終端中運行代碼。

$ python file_parser.py --infile something.txt
   Processing something.txt
   Finished processing

在這里，你用 --infile標志和一個文件名來運行你的腳本。這將運行 main()，然后調(diào)用 file_parser()。

下一步是使用你在代碼中聲明的兩個命令行參數(shù)嘗試運行你的應(yīng)用程序。

$ python file_parser.py --infile something.txt --out output.txt
   Processing something.txt
   Finished processing
   Creating output.txt

這一次，你得到了一個額外的輸出行，提到了輸出文件名。這代表你的代碼邏輯中的一個分支。當你指定一個輸出文件時，你可以讓你的代碼通過使用一個新的代碼塊或一個函數(shù)來生成該文件。如果你不指定一個輸出文件，那么那個代碼塊就不會運行。

當你使用argparse創(chuàng)建你的命令行工具時，你可以很容易地添加信息，當你的用戶不確定如何正確地與你的程序互動時，可以幫助他們。

現(xiàn)在是時候找出如何從你的應(yīng)用程序中獲得幫助了

創(chuàng)建幫助信息

argparse庫將使用你在創(chuàng)建每個參數(shù)時提供的信息，自動為你的應(yīng)用程序創(chuàng)建一個有用的信息。這里是代碼：

# file_parser.py
   import argparse
   def file_parser(input_file, output_file=''):
       print(f'Processing {input_file}')
       print('Finished processing')
       if output_file:
           print(f'Creating {output_file}')
   def main():
       parser = argparse.ArgumentParser('File parser')
       parser.add_argument('--infile', help='Input file')
       parser.add_argument('--out', help='Output file')
       args = parser.parse_args()
       if args.infile:
           file_parser(args.infile, args.out)
   if __name__ == '__main__':
       main()

現(xiàn)在試著用 -h標志運行這段代碼，你應(yīng)該看到以下內(nèi)容。

$ file_parser.py -h
   usage: File parser [-h] [--infile INFILE] [--out OUT]
   optional arguments:
     -h, --help       show this help message and exit
     --infile INFILE  Input file
     --out OUT        Output file

add_argument()的幫助參數(shù)被用來創(chuàng)建上面的幫助信息。argparse會自動添加 -h和 -help選項。你可以通過給它一個描述和后記來使你的幫助信息更豐富。

讓我們用它們來改進你的幫助信息。首先，把上面的代碼復(fù)制到一個新的文件中，命名為 file_parser_with_description.py，然后把它修改成下面的樣子。

# file_parser_with_description.py
   import argparse
   def file_parser(input_file, output_file=''):
       print(f'Processing {input_file}')
       print('Finished processing')
       if output_file:
           print(f'Creating {output_file}')
   def main():
       parser = argparse.ArgumentParser(
               'File parser',
               description='PyParse - The File Processor',
               epilog='Thank you for choosing PyParse!',
               )
       parser.add_argument('--infile', help='Input file for conversion')
       parser.add_argument('--out', help='Converted output file')
       args = parser.parse_args()
       if args.infile:
           file_parser(args.infile, args.out)
   if __name__ == '__main__':
       main()

在這里，把description和epilog參數(shù)傳遞給ArgumentParser。還更新了 add_argument()的幫助參數(shù)，使其更具描述性。

在做了這些修改之后，當你用 -h或 --help運行這個腳本時，你會看到以下輸出。

$ python file_parser_with_description.py -h
   usage: File parser [-h] [--infile INFILE] [--out OUT]
   PyParse - The File Processor
   optional arguments:
     -h, --help       show this help message and exit
     --infile INFILE  Input file for conversion
     --out OUT        Converted output file
   Thank you for choosing PyParse!

現(xiàn)在可以在你的幫助輸出中看到新的description 和epilog。這給了你的命令行程序一些額外的修飾。

你也可以通過ArgumentParser的 add_help參數(shù)在你的應(yīng)用程序中完全禁用幫助。如果你認為你的幫助文本過于冗長，你可以像這樣禁用它。

# file_parser_no_help.py
   import argparse
   def file_parser(input_file, output_file=''):
       print(f'Processing {input_file}')
       print('Finished processing')
       if output_file:
           print(f'Creating {output_file}')
   def main():
       parser = argparse.ArgumentParser(
               'File parser',
               description='PyParse - The File Processor',
               epilog='Thank you for choosing PyParse!',
               add_help=False,
               )
       parser.add_argument('--infile', help='Input file for conversion')
       parser.add_argument('--out', help='Converted output file')
       args = parser.parse_args()
       if args.infile:
           file_parser(args.infile, args.out)
   if __name__ == '__main__':
       main()

通過將 add_help設(shè)置為 False，你將禁用 -h和 --help標志。

你可以看到下面的演示。

$ python file_parser_no_help.py --help
   usage: File parser [--infile INFILE] [--out OUT]
   File parser: error: unrecognized arguments: --help

在下一節(jié)中，你將學習如何為你的參數(shù)添加別名!

添加別名

別名是一個花哨的詞，指的是使用一個替代的標志來做同樣的事情。例如，你知道你可以使用 -h和 --help來訪問程序的幫助信息。-h是 --help的別名，反之亦然。

看看 main()里面的 parser.add_argument()方法有什么變化。

# file_parser_aliases.py
   import argparse
   def file_parser(input_file, output_file=''):
       print(f'Processing {input_file}')
       print('Finished processing')
       if output_file:
           print(f'Creating {output_file}')
   def main():
       parser = argparse.ArgumentParser(
               'File parser',
               description='PyParse - The File Processor',
               epilog='Thank you for choosing PyParse!',
               add_help=False,
               )
       parser.add_argument('-i', '--infile', help='Input file for conversion')
       parser.add_argument('-o', '--out', help='Converted output file')
       args = parser.parse_args()
       if args.infile:
           file_parser(args.infile, args.out)
   if __name__ == '__main__':
       main()

這里你改變了第一個 add_argument()，除了接受 -infile之外，還接受了 -i，你還在第二個 add_argument()中加入了 -o。這樣就可以使用兩個新的快捷標志來運行你的代碼。

下面是一個例子。

$ python3 file_parser_aliases.py -i something.txt -o output.txt
   Processing something.txt
   Finished processing
   Creating output.txt

如果你去看argparse文檔，你會發(fā)現(xiàn)也可以給子解析器添加別名。子解析器是一種在你的應(yīng)用程序中創(chuàng)建子命令的方法，這樣它就可以做其他事情。一個很好的例子是Docker，一個虛擬化或容器應(yīng)用程序。它有一系列的命令，你可以在docker下運行，以及docker compose等等。這些命令中的每一個都有獨立的子命令，你都可以使用。

下面是一個典型的docker命令，用于運行一個容器。

docker exec -it container_name bash

這將用docker啟動一個容器。而如果你要使用docker compose，你將使用一組不同的命令。exec和compose是subparsers的例子。

使用相互排斥的參數(shù)

有時你需要讓你的應(yīng)用程序接受一些參數(shù)，但不接受其他參數(shù)。例如，你可能想限制你的應(yīng)用程序，使它只能創(chuàng)建或刪除文件，而不是同時創(chuàng)建和刪除。

argparse模塊提供了 add_mutually_exclusive_group()方法，它就是這樣做的。

將你的兩個參數(shù)添加到一個組對象中，使其相互排斥，如下面的例子。

# file_parser_exclusive.py
   import argparse
   def file_parser(input_file, output_file=''):
       print(f'Processing {input_file}')
       print('Finished processing')
       if output_file:
           print(f'Creating {output_file}')
   def main():
       parser = argparse.ArgumentParser(
               'File parser',
               description='PyParse - The File Processor',
               epilog='Thank you for choosing PyParse!',
               add_help=False,
               )
       group = parser.add_mutually_exclusive_group()
       group.add_argument('-i', '--infile', help='Input file for conversion')
       group.add_argument('-o', '--out', help='Converted output file')
       args = parser.parse_args()
       if args.infile:
           file_parser(args.infile, args.out)
   if __name__ == '__main__':
       main()

首先，你創(chuàng)建了一個相互排斥的組。然后，你把 -i和 -o參數(shù)添加到組中，而不是添加到解析器對象中?，F(xiàn)在這兩個參數(shù)是互斥的。

下面是當你試圖用這兩個參數(shù)運行你的代碼時發(fā)生的情況。

$ python3 file_parser_exclusive.py -i something.txt -o output.txt
   usage: File parser [-i INFILE | -o OUT]
   File parser: error: argument -o/--out: not allowed with argument -i/--infile

用這兩個參數(shù)運行你的代碼，會使你的解析器向用戶顯示一條錯誤信息，解釋他們做錯了什么。

在涵蓋了所有這些與使用argparse有關(guān)的信息之后，你已經(jīng)準備好應(yīng)用你的新技能來創(chuàng)建一個簡單的搜索工具了

創(chuàng)建一個簡單的搜索工具

在開始創(chuàng)建一個應(yīng)用程序之前，弄清楚你要完成的任務(wù)總是好的。你在本節(jié)中想要建立的應(yīng)用程序應(yīng)該能夠搜索特定文件類型的文件。為了使它更有趣，你可以添加一個額外的參數(shù)，讓你也能選擇性地搜索特定的文件大小。

你可以使用 Python 的 glob 模塊來搜索文件類型。你可以在這里閱讀關(guān)于這個模塊的所有信息。

還有一個 fnmatch 模塊，glob 自己也使用它。你現(xiàn)在應(yīng)該使用 glob，因為它更容易使用，但是如果你有興趣寫一些更專業(yè)的東西，那么 fnmatch 可能是你正在尋找的。

然而，由于你希望能夠通過文件大小來選擇性地過濾返回的文件，你可以使用 pathlib，它包括一個類似 glob 的接口。glob 模塊本身并不提供文件大小的信息。

你可以先創(chuàng)建一個名為 pysearch.py 的文件并輸入以下代碼。

# pysearch.py
   import argparse
   import pathlib
   def search_folder(path, extension, file_size=None):
       """
       Search folder for files
       """
       folder = pathlib.Path(path)
       files = list(folder.rglob(f'*.{extension}'))
       if not files:
           print(f'No files found with {extension=}')
           return
       if file_size is not None:
           files = [
                   f
                   for f in files
                   if f.stat().st_size >= file_size
                   ]
       print(f'{len(files)} *.{extension} files found:')
       for file_path in files:
           print(file_path)

在上面的代碼片段中，首先導入了argparse和pathlib。接下來，創(chuàng)建了search_folder()函數(shù)，它接收了三個參數(shù)。

path - 要搜索的文件夾
extension - 要尋找的文件擴展名
file_size - 要過濾的文件大小，以字節(jié)為單位。

把路徑變成pathlib.Path對象，然后使用其rglob()方法在文件夾中搜索用戶傳入的擴展名。如果沒有找到文件，就向用戶打印一個有意義的信息，然后退出。

如果找到了任何文件，就檢查是否已經(jīng)設(shè)置了filesize。如果它被設(shè)置了，就用一個list comprehension來過濾出小于指定的filesize的文件。

接下來，打印出找到的文件的數(shù)量，最后在這些文件上循環(huán)，打印出它們的名字。

為了使這一切正常工作，需要創(chuàng)建一個命令行界面。你可以通過添加一個包含argparse代碼的main()函數(shù)來做到這一點，像這樣。

def main():
       parser = argparse.ArgumentParser(
               'PySearch',
               description='PySearch - The Python Powered File Searcher',
               )
       parser.add_argument('-p', '--path',
                           help='The path to search for files',
                           required=True,
                           dest='path')
       parser.add_argument('-e', '--ext',
                           help='The extension to search for',
                           required=True,
                           dest='extension')
       parser.add_argument('-s', '--size',
                           help='The file size to filter on in bytes',
                           type=int,
                           dest='size',
                           default=None)
       args = parser.parse_args()
       search_folder(args.path, args.extension, args.size)
   if __name__ == '__main__':
       main()

這個ArgumentParser()有三個參數(shù)，與你傳遞給search_folder()的參數(shù)相對應(yīng)。讓--path和--ext參數(shù)成為必需的，而讓--size參數(shù)成為可選的。注意，--size參數(shù)被設(shè)置為type=int，這意味著你不能把它傳成字符串。

add_argument()函數(shù)有一個新的參數(shù)。它是dest參數(shù)，可以用它來告訴你的參數(shù)分析器在哪里保存?zhèn)鬟f給它們的參數(shù)。

下面是一個腳本運行的例子。

$ python3 pysearch.py -p /Users/michael/Dropbox/python101code/chapter32_argparse -e py -s 650
   6 *.py files found:
   /Users/michael/Dropbox/python101code/chapter32_argparse/file_parser_aliases2.py
   /Users/michael/Dropbox/python101code/chapter32_argparse/pysearch.py
   /Users/michael/Dropbox/python101code/chapter32_argparse/file_parser_aliases.py
   /Users/michael/Dropbox/python101code/chapter32_argparse/file_parser_with_description.py
   /Users/michael/Dropbox/python101code/chapter32_argparse/file_parser_exclusive.py
   /Users/michael/Dropbox/python101code/chapter32_argparse/file_parser_no_help.py

現(xiàn)在試試用-s和一個字符串來運行它。

$ python3 pysearch.py -p /Users/michael/Dropbox/python101code/chapter32_argparse -e py -s python
   usage: PySearch [-h] -p PATH -e EXTENSION [-s SIZE]
   PySearch: error: argument -s/--size: invalid int value: 'python'

這次我們收到了一個錯誤，因為-s和-size只接受整數(shù)。在你自己的機器上運行一下這段代碼，看看當你使用-s和整數(shù)時，它是否按你想要的方式工作。

這里有一些想法，你可以用來改進你的代碼版本。

更好地處理擴展文件?，F(xiàn)在，它將接受 *.py，這不會像你期望的那樣工作。

更新代碼，以便你可以一次搜索多個擴展名

更新代碼，以便對文件大小的范圍進行過濾（例如，1MB-5MB）。

還有很多其他的功能和改進，你可以添加到這個代碼中，比如添加錯誤處理或單元測試。

總結(jié)

argparse模塊功能齊全，可以用來創(chuàng)建龐大、靈活的命令行應(yīng)用程序。在本章中，你了解了以下內(nèi)容。

解析參數(shù)
創(chuàng)建有用的信息
添加別名
使用相互排斥的參數(shù)
創(chuàng)建一個簡單的搜索工具

你可以用argparse模塊做更多的事情，不完全包括本章所講的。請務(wù)必查看文檔以了解全部細節(jié)。

到此這篇關(guān)于利用Python內(nèi)置庫實現(xiàn)創(chuàng)建命令行應(yīng)用程序的文章就介紹到這了,更多相關(guān)Python命令行應(yīng)用程序內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: