如何有條件地讓蟒蛇的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)