如何有条件地让蟒蛇的argparse模块需要额外的参数

Question

基本预期用途：

my_framework create Project_title /path/to/project

OR

my_framework create Project_title（即用当前的工作目录）

OR

my_framework update （即更新my_framework而不是创建新项目）

我知道我可以做name可选向其提供一个默认的，但是，在现实中name是不提供可选用户输入create作为第一个参数。

我想出的最佳解决方案是使用name的默认值，然后如果参数name等于其默认值，则会抛出错误。但是，如果有一种方法可以为我做这个工作，我宁愿学会这样做。

写两个脚本，my_framework_create和my_framework_update并不吸引我审美。

#!/usr/bin/env python 


import argparse 
import os 
import shutil 
from subprocess import call 

template_path = "/usr/local/klibs/template" 
parser = argparse.ArgumentParser("MY_FRAMEWORK CLI", description='Creates a new MY_FRAMEWORK project or updates MY_FRAMEWORK') 
parser.add_argument('action', choices=['create', 'update'], type=str, help='<help text>') 
parser.add_argument('name', type=str, help='<help text>') 
parser.add_argument('path', default=os.getcwd(), nargs="?", type=str, help='<help text>') 
args = parser.parse_args() 

if args.action == "create": 
    # do the create stuff 

if args.action == "update": 
    # do the update stuff 

Answer 1

做到这一点，最好的办法是用subparser

从文档的一个例子：

>>> parser = argparse.ArgumentParser() 
>>> subparsers = parser.add_subparsers(title='subcommands', 
...         description='valid subcommands', 
...         help='additional help') 
>>> subparsers.add_parser('foo') 
>>> subparsers.add_parser('bar') 
>>> parser.parse_args(['-h']) 
usage: [-h] {foo,bar} ... 

optional arguments: 
    -h, --help show this help message and exit 

subcommands: 
    valid subcommands 

    {foo,bar} additional help 

你的情况，你会create和update作为单独subparsers。

例子：

def create(args): 
    # do the create stuff 
    print(args) 


def update(args): 
    # do the update stuff 
    print(args) 


parser = argparse.ArgumentParser() 
subparsers = parser.add_subparsers(title='subcommands', 
            description='valid subcommands', 
            help='additional help') 

create_parser = subparsers.add_parser('create') 
create_parser.add_argument('name', type=str) 
create_parser.set_defaults(func=create) 

update_parser = subparsers.add_parser('update') 
update_parser.set_defaults(func=update)