Pythonで、argparseを使用して、正の整数のみを許可する

Question

タイトルは、私が何をしたいのかを要約しています。

これは私が持っているものであり、プログラムは正の整数ではなくなりますが、正の整数でないことは基本的にナンセンスであることをユーザーに伝えて欲しいのです。

import argparse 
parser = argparse.ArgumentParser() 
parser.add_argument("-g", "--games", type=int, default=162, 
        help="The number of games to simulate") 
args = parser.parse_args() 

そして出力：負と

python simulate_many.py -g 20 
Setting up... 
Playing games... 
.................... 

出力：今

python simulate_many.py -g -2 
Setting up... 
Playing games... 

、明らかにif args.gamesが負であるかを決定する場合、私はちょうど追加することができますが、私は、もし興味がありました自動使用印刷を利用するために、それを​​レベルにトラップする方法がありました。

：今、私はこれをやっている、と私は幸せだと思うのため

python simulate_many.py -g -2 
usage: simulate_many.py [-h] [-g GAMES] [-d] [-l LEAGUE] 
simulate_many.py: error: argument -g/--games: invalid positive int value: '-2' 

：そうのよう

python simulate_many.py -g a 
usage: simulate_many.py [-h] [-g GAMES] [-d] [-l LEAGUE] 
simulate_many.py: error: argument -g/--games: invalid int value: 'a' 

：

理想的には、これに似たものを印刷します

if args.games <= 0: 
    parser.print_help() 
    print "-g/--games: must be positive." 
    sys.exit(1) 

Answer 1

これはtypeを利用して可能になるはずです。あなたはまだあなたのためにこれを決定する実際のメソッドを定義する必要があります：

def check_positive(value): 
    ivalue = int(value) 
    if ivalue <= 0: 
     raise argparse.ArgumentTypeError("%s is an invalid positive int value" % value) 
    return ivalue 

parser = argparse.ArgumentParser(...) 
parser.add_argument('foo', type=check_positive) 

これは基本的に​​上docsでperfect_square機能からわずかに適応例です。あなたの引数のための分だけでなく、予測可能な最大を持っている場合

Answer 2

迅速かつ汚い方法は、よう、条件/小切手を処理するための推奨オプションとなり範囲

parser.add_argument('foo', type=int, choices=xrange(0, 1000)) 

Answer 3

typeでchoicesを使用していますユウシの答えで

parser.add_argument('foo', type=int, choices=xrange(5, 10)) 

Answer 4

シンプルな代替、argparse.ArgumentParserをサブクラス化、parse_argsメソッド内から検証を開始することである場合は特に：あなたの上限も知られている場合は、あなたの特定のケースで

、あなたもchoicesパラメータを使用することができます。

なサブクラス内部：

def parse_args(self, args=None, namespace=None): 
    """Parse and validate args.""" 
    namespace = super().parse_args(args, namespace) # super() works in Python 3 
    if namespace.games <= 0: 
     raise self.error('The number of games must be a positive integer.') 
    return namespace 

この手法は、呼び出し可能なカスタムのようにクールではないかもしれないが、それは仕事をしていません。

---

についてArgumentParser.error(message)：

This method prints a usage message including the message to the standard error and terminates the program with a status code of 2.

---

クレジット：answer by jonatan