Python argparse子命令

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'])