詳解Python中的argparse模塊
argparse模塊
argparse是一個用于解析命令行參數(shù)和選項的python標準庫模塊。
它可以讓你輕松地編寫用戶友好的命令行界面,定義你的程序需要的參數(shù),自動生成幫助和用法信息,以及處理用戶給出的無效參數(shù)。
在這篇博客中,我將介紹argparse的基本用法,以及一些高級功能,如子命令、互斥組、文件類型等。
我將用豐富的例子和解釋來幫助你理解和掌握argparse的使用方法。
一、一個最簡單的示例
使用argparse,你可以指定程序所需的參數(shù)、選項以及如何解析它們。
下面是一個基本的示例,演示了如何使用argparse來解析命令行參數(shù):
import argparse # 創(chuàng)建ArgumentParser對象 parser = argparse.ArgumentParser(description='這是一個使用argparse的示例程序') # 添加位置參數(shù) parser.add_argument('name', help='姓名') parser.add_argument('age', type=int, help='年齡') # 添加可選參數(shù) parser.add_argument('--gender', choices=['male', 'female'], help='性別') parser.add_argument('--height', type=float, help='身高') # 解析命令行參數(shù) args = parser.parse_args() # 訪問解析后的參數(shù) print(f'姓名: {args.name}') print(f'年齡: {args.age}') print(f'性別: {args.gender}') print(f'身高: {args.height}')
在上面的示例中,我們首先創(chuàng)建了一個ArgumentParser對象,并提供了一個描述性的文字作為參數(shù)。
然后,我們添加了兩個位置參數(shù)(name和age),并分別指定了它們的幫助文本。
接下來,我們添加了兩個可選參數(shù)(–gender和–height),并使用choices和type參數(shù)限制了它們的取值范圍和類型。
最后,我們調(diào)用parse_args()方法來解析命令行參數(shù),并將解析后的結(jié)果存儲在args變量中。我們可以通過訪問args對象的屬性來獲取解析后的參數(shù)值,并進行相應(yīng)的處理。
例如,如果我們在命令行中運行上述代碼并提供以下輸入:
python my_program.py John 25 --gender male --height 1.75
輸出將是:
姓名: John
年齡: 25
性別: male
身高: 1.75
或者在pycharm輸入這些參數(shù):
這樣可以獲得相同的結(jié)果:
如果不輸入默認參數(shù)直接運行,會如下圖的報錯:
二 、詳細參數(shù)講解
創(chuàng)建一個解析器
要使用argparse,首先你需要創(chuàng)建一個argparse.ArgumentParser的實例,它是一個包含參數(shù)規(guī)范和一些全局選項的容器。
你可以在創(chuàng)建時傳入一些參數(shù),如程序名、描述、幫助信息等,來定制你的解析器。
例如:
import argparse # 創(chuàng)建一個解析器 parser = argparse.ArgumentParser( prog="myprog", # 程序名,默認為sys.argv[0] description="This is a program that does something", # 程序描述 epilog="This is some text at the bottom of help" # 幫助信息底部的文本 )
添加參數(shù)
創(chuàng)建了一個解析器后,你需要用add_argument()方法來添加你想要解析的參數(shù)。
argparse支持兩種類型的參數(shù):位置參數(shù)和可選參數(shù)。
位置參數(shù)是根據(jù)它們在命令行中出現(xiàn)的位置來識別的,它們通常是必須提供的參數(shù)。
例如:
# 添加一個位置參數(shù) parser.add_argument("filename", help="the name of the file to process")
可選參數(shù)是根據(jù)它們前面的短橫線或雙橫線來識別的,它們通常是可以省略或指定不同值的參數(shù)。例如:
# 添加一個可選參數(shù) parser.add_argument("-v", "--verbose", help="increase output verbosity", action="store_true")
add_argument()方法可以接受很多參數(shù)來定制你的參數(shù)行為和屬性,如類型、默認值、選項、必需性、幫助信息等。你可以查看[argparse文檔]來了解更多細節(jié)。
解析參數(shù)
添加了所有想要解析的參數(shù)后,你可以用parse_args()方法來運行解析器,并將提取出來的數(shù)據(jù)放在一個argparse.Namespace對象中。例如:
# 解析參數(shù) args = parser.parse_args() # 使用參數(shù) if args.verbose: print(f"Processing {args.filename} in verbose mode") else: print(f"Processing {args.filename}")
如果用戶給出了無效或不符合規(guī)范的參數(shù),argparse會自動報錯并退出程序。如果用戶沒有給出任何參數(shù)或給出了-h或–help選項,argparse會自動打印幫助信息并退出程序。
子命令
有時候,你可能想要讓你的程序支持不同的子命令,就像git或pip那樣。例如:
$ git clone https://github.com/python/cpython.git $ git status $ git commit -m "some message"
要實現(xiàn)這個功能,你可以使用add_subparsers()方法來創(chuàng)建一個子解析器,并為每個子命令添加一個子解析器。例如:
# 創(chuàng)建一個主解析器 parser = argparse.ArgumentParser(description="A simple calculator") # 創(chuàng)建一個子解析器 subparsers = parser.add_subparsers(help="sub-command help") # 為每個子命令添加一個子解析器 parser_add = subparsers.add_parser("add", help="add two numbers") parser_add.add_argument("x", type=int, help="the first number") parser_add.add_argument("y", type=int, help="the second number") parser_sub = subparsers.add_parser("sub", help="subtract two numbers") parser_sub.add_argument("x", type=int, help="the first number") parser_sub.add_argument("y", type=int, help="the second number") parser_mul = subparsers.add_parser("mul", help="multiply two numbers") parser_mul.add_argument("x", type=int, help="the first number") parser_mul.add_argument("y", type=int, help="the second number") parser_div = subparsers.add_parser("div", help="divide two numbers") parser_div.add_argument("x", type=int, help="the first number") parser_div.add_argument("y", type=int, help="the second number") # 解析參數(shù) args = parser.parse_args() # 使用參數(shù) if args.subcommand == "add": print(args.x + args.y) elif args.subcommand == "sub": print(args.x - args.y) elif args.subcommand == "mul": print(args.x * args.y) elif args.subcommand == "div": print(args.x / args.y)
互斥組
有時候,你可能想要讓你的程序只接受一些互斥的選項,即不能同時出現(xiàn)的選項。例如:
$ python3 prog.py -a $ python3 prog.py -b $ python3 prog.py -a -b # 錯誤
要實現(xiàn)這個功能,你可以使用add_mutually_exclusive_group()方法來創(chuàng)建一個互斥組,并為該組添加一些選項。例如:
# 創(chuàng)建一個解析器 parser = argparse.ArgumentParser(description="A simple program") # 創(chuàng)建一個互斥組 group = parser.add_mutually_exclusive_group() # 為該組添加一些選項 group.add_argument("-a", action="store_true") group.add_argument("-b", action="store_true") # 解析參數(shù) args = parser.parse_args() # 使用參數(shù) if args.a: print("Option A is selected") elif args.b: print("Option B is selected") else: print("No option is selected")
文件類型
有時候,你可能想要讓你的程序接受一個文件名作為參數(shù),并自動打開或創(chuàng)建該文件。例如:
$ python3 prog.py input.txt output.txt
要實現(xiàn)這個功能,你可以使用argparse.FileType類來指定你的參數(shù)的類型,它會返回一個類似于open()函數(shù)返回的文件對象。例如:
# 創(chuàng)建一個解析器 parser = argparse.ArgumentParser(description="A simple program that reads from a file and writes to another file") # 添加兩個文件類型的參數(shù) parser.add_argument("infile", type=argparse.FileType("r"), help="the input file") parser.add_argument("outfile", type=argparse.FileType("w"), help="the output file") # 解析參數(shù) args = parser.parse_args() # 使用參數(shù) for line in args.infile: args.outfile.write(line.upper())
總結(jié)
argparse是Python標準庫中一個強大而靈活的模塊,用于解析命令行參數(shù)和生成美觀的幫助文檔。
無論是簡單的腳本還是復雜的應(yīng)用程序,argparse都能幫助開發(fā)者輕松地處理各種命令行參數(shù),并提供友好的用戶界面。
通過使用argparse,開發(fā)者可以定義參數(shù)的類型、默認值、限制條件以及互斥關(guān)系,從而確保命令行輸入的合法性和一致性。
無需編寫繁瑣的解析代碼,argparse的簡潔接口使得命令行參數(shù)的處理變得簡單而愉快。
不僅如此,argparse還能自動生成詳細的幫助文檔,為用戶提供清晰明了的命令行使用說明。
在Python中,argparse被廣泛應(yīng)用于各種項目,成為處理命令行參數(shù)的首選工具。
無論是初學者還是有經(jīng)驗的開發(fā)者,掌握argparse的使用技巧都將極大地提升命令行工具的質(zhì)量和用戶體驗。
到此這篇關(guān)于詳解Python中的argparse模塊的文章就介紹到這了,更多相關(guān)Python的argparse模塊內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
- 關(guān)于Python參數(shù)解析器argparse的應(yīng)用場景
- Python命令行參數(shù)解析之a(chǎn)rgparse模塊詳解
- Python?內(nèi)置模塊?argparse快速入門教程
- 深入講解Python命令行解析模塊argparse
- python?argparse的使用步驟(全網(wǎng)最全)
- python中的argparse基本用法(使用步驟)
- python?命令行參數(shù)模塊argparse的實現(xiàn)
- Python中命令行參數(shù)argparse模塊的使用
- python命令行參數(shù)argparse模塊基本用法詳解
- Python中argparse基本用法小結(jié)
相關(guān)文章
Python Django form 組件動態(tài)從數(shù)據(jù)庫取choices數(shù)據(jù)實例
這篇文章主要介紹了Python Django form 組件動態(tài)從數(shù)據(jù)庫取choices數(shù)據(jù)實例,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-05-05Python實現(xiàn)自動化處理Word文檔的方法詳解
本文主要介紹了如何使用Python實現(xiàn)Word文檔的自動化處理,包括批量生成Word文檔、在Word文檔中批量進行查找和替換、將Word文檔批量轉(zhuǎn)換成PDF等,希望對你有所幫助2022-08-08