파이썬에서 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가 부의 결정하는 경우 난 그냥를 추가 할 수 있습니다,하지만 난 경우 궁금해서 자동 사용 인쇄 기능을 활용하기 위해 레벨을 argparse 레벨로 트랩하는 방법이있었습니다. 지금은이 일을 해요 들어

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) 

이것은 기본적으로 argparse에 docs에서 perfect_square 기능에서 단 개조 예입니다. 당신이 당신의 인수에 대한 분뿐만 아니라 예측 가능한 최대있는 경우

Answer 2

빠른 더러운 방법은, 같은 조건/수표를 처리하기 위해 권장되는 옵션이 될 것입니다 범위

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

Answer 3

type와 choices을 사용하는 것입니다 유시의 대답. 당신의 상한이 알려진 경우

특정 경우

, 당신은 또한 choices 매개 변수를 사용할 수 있습니다 :

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

Answer 4

간단한 대안을, argparse.ArgumentParser를 서브 클래 싱하기는 parse_args 방법 내부에서 유효성 검사를 시작하는 것입니다 특히 . 이 기술은 사용자 정의 호출만큼 시원하지 않을 수 있습니다

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