1. ホーム
  2. python

[解決済み] argparseで選択されたサブコマンドを取得する

2022-04-25 14:16:31

質問

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] は、グローバルな引数が考えられるので、うまくいきません。サブコマンドそのものを取得するにはどうしたらよいでしょうか?

解決方法は?

の一番下は argparse サブコマンドに関する Python ドキュメント がその方法を説明しています。

>>> 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() というメソッドが、私が見つけた例のすぐ上で参照されています。