Python argparse子命令

最新推荐文章于 2024-10-21 21:30:48 发布

原创最新推荐文章于 2024-10-21 21:30:48 发布 · 1k 阅读

CC 4.0 BY-SA版权

Python的argparse模块提供了一种简单而有效的方法来处理命令行参数，包括生成帮助和错误信息。它允许程序员定义所需的参数，并自动从sys.argv中解析这些参数。此外，argparse支持子命令功能，使得复杂程序的不同部分可以拥有各自的参数。例如，通过创建子命令'add'和'sub'，用户可以根据需要传递不同的选项，如'-x'和'-y'，来进行加法或减法操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

argparse

argparse可以轻松编写用户友好的命令行界面。

程序定义了它需要哪些参数，argparse将找出如何从sys.argv中解析这些参数。

argparse还会自动生成帮助和使用信息，并且在用户给程序提供无效参数时发出错误信息。

import argparse

# creating a parser
parser = argparse.ArgumentParser(description='Process some integers.') 
# Adding arguments
parser.add_argument('--sum', dest='accumulate', action='store_const',
...                     const=sum, default=max,
...                     help='sum the integers (default: find the max)')
# parsing arguments
parser.parse_args(['--sum', '7', '-1', '42']) # In script
parse_args() # automatically determine the command line arguments from sys.argv

Sub-commands

当程序比较复杂且不同功能都需要不同参数时，可以使用子命令。

import argparse

parser = argparse.ArgumentParser(prog='PROG')
# argparse使用add_subparsers方法创建子命令
subparsers = parser.add_subparsers(help='sub-command help')
#添加子命令 add
parser_a = subparsers.add_parser('add', help='add help')
parser_a.add_argument('-x', type=int, help='x value')
parser_a.add_argument('-y', type=int, help='y value')
#添加子命令 sub
parser_s = subparsers.add_parser('sub', help='sub help')
parser_s.add_argument('-x', type=int, help='x value')
parser_s.add_argument('-y', type=int, help='y value')



# 使用不同的子命令
args = parser.parse_args(['add'])