在Python的argparse模塊中,可以通過設置type
參數為一個callable對象來限制參數的值范圍。當參數值超出范圍時,argparse會自動顯示錯誤消息并退出程序。
要改變幫助消息的外觀,可以使用metavar
參數來設置在幫助消息中顯示的參數名稱。下面是一個示例代碼:
import argparse
# 創建一個ArgumentParser對象
parser = argparse.ArgumentParser()
# 添加一個參數,并設置值范圍為1到10
parser.add_argument('--number', type=int, choices=range(1, 11), metavar='N',
help='an integer between 1 and 10')
# 解析命令行參數
args = parser.parse_args()
在這個示例中,metavar='N'
將在幫助消息中顯示參數名稱為N
,而不是--number
。用戶輸入的值如果不在范圍內,argparse會自動顯示以下錯誤消息:
usage: script.py [-h] [--number N]
script.py: error: argument --number: invalid choice: VALUE (choose from 1 to 10)
你可以根據需要修改metavar
的值來適應你的參數名稱和幫助消息的外觀。