รับคำสั่งย่อยที่เลือกด้วย argparse

เมื่อฉันใช้คำสั่งย่อยด้วย python argparse ฉันสามารถรับอาร์กิวเมนต์ที่เลือกได้

parser = argparse.ArgumentParser()
parser.add_argument('-g', '--global')
subparsers = parser.add_subparsers()   
foo_parser = subparsers.add_parser('foo')
foo_parser.add_argument('-c', '--count')
bar_parser = subparsers.add_parser('bar')
args = parser.parse_args(['-g, 'xyz', 'foo', '--count', '42'])
# args => Namespace(global='xyz', count='42')

ดังนั้นไม่ได้มีargs 'foo'การเขียนเพียงอย่างsys.argv[1]เดียวไม่สามารถใช้งานได้เนื่องจากโลกที่เป็นไปได้ ฉันจะรับคำสั่งย่อยได้อย่างไร

ด้านล่างสุดของเอกสาร Python บนคำสั่งย่อย argparseอธิบายวิธีการดังนี้:

>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('-g', '--global')
>>> subparsers = parser.add_subparsers(dest="subparser_name") # this line changed
>>> foo_parser = subparsers.add_parser('foo')
>>> foo_parser.add_argument('-c', '--count')
>>> bar_parser = subparsers.add_parser('bar')
>>> args = parser.parse_args(['-g', 'xyz', 'foo', '--count', '42'])
>>> args
Namespace(count='42', global='xyz', subparser_name='foo')

คุณยังสามารถใช้set_defaults()วิธีการอ้างอิงเหนือตัวอย่างที่ฉันพบ

ArgumentParser.add_subparsersมีdestการโต้แย้งอย่างเป็นทางการอธิบายว่า:

dest- ชื่อของแอ็ตทริบิวต์ที่จะเก็บชื่อคำสั่งย่อย; โดยค่าเริ่มต้นNoneและไม่มีการจัดเก็บค่าใด ๆ

ในตัวอย่างด้านล่างของรูปแบบฟังก์ชั่นการใช้งานที่ง่าย subparsers ที่ subparser parser.parse_args().subparserที่เลือกอยู่ใน

import argparse


def task_a(alpha):
    print('task a', alpha)


def task_b(beta, gamma):
    print('task b', beta, gamma)


if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    subparsers = parser.add_subparsers(dest='subparser')

    parser_a = subparsers.add_parser('task_a')
    parser_a.add_argument(
        '-a', '--alpha', dest='alpha', help='Alpha description')

    parser_b = subparsers.add_parser('task_b')
    parser_b.add_argument(
        '-b', '--beta', dest='beta', help='Beta description')
    parser_b.add_argument(
        '-g', '--gamma', dest='gamma', default=42, help='Gamma description')

    kwargs = vars(parser.parse_args())
    globals()[kwargs.pop('subparser')](**kwargs)