pythonのargparseで引数の数の可変

質問

現在、pythonのプログラムで、コマンドにオプションを付けて、それによって戻ってくる結果を表示するようにしたいと思い作成しています。

それで、次のようなプログラムを作ることができました。
$ python pythonProg.py -s 5 5 5
5
5
5
5+5+5 == 15

(githubでの現在のソースコード内容)
https://github.com/KenjiMaehara/raspberryPiTCPClient/blob/testBranch20170201_02/testSpi161101/pythonProg.py

現在、-sのオプションと３つの引数x,y,zを渡すと、x,y,zの総和を表示するようにしているのですが、

引数の数を２つにした場合、次のようなエラーが出てきます。

$ python pythonProg.py -s 5 5
usage: pythonProg.py [-h] [-v | -q | -s] x y z
pythonProg.py: error: too few arguments

”argumentsが少ない”というエラーなのでしょうが、引数を３つ以外でも総和を計算できるようにすることはできますでしょうか？

現在のソースコード内容な下記のような感じです。

(pythonProg.pyのファイル内容)
import argparse

parser = argparse.ArgumentParser()
group = parser.add_mutually_exclusive_group()
group.add_argument("-v", "--verbose", action="store_true")
group.add_argument("-q", "--quiet", action="store_true")
group.add_argument("-s", "--sum", action="store_true")
parser.add_argument("x", type=int, help="the base")
parser.add_argument("y", type=int, help="the exponent")
parser.add_argument("z", type=int, help="the exponent")
args = parser.parse_args()
answer = args.x+args.y
answer02 = args.x+args.y+args.z
(中略)

どうぞ、宜しくお願い致します。

回答No.1

wormhole · ベストアンサー

add_argumentの引数nargsは使えないですか？
リストで取れた分の要素数チェックは別途必要になるかもしれませんが。

お礼

回答頂きありがとうございます。

https://github.com/KenjiMaehara/raspberryPiTCPClient/blob/testBranch20170201_02/testSpi161101/pythonProg.py

$ python pythonProg.py --sum 1 2 3 4
10

$ python pythonProg.py --sum 2 2 2
6

$ python pythonProg.py 1 2 3 4
4

このように、--sumのオプションを付けた場合は、与えられた要素の総和を求め、

オプション無しの場合は、与えられた要素の最大値のみを表示するようなものが作成できました。

助かります。