In Python, usando argparse, consenti solo interi positivi

Question

Il titolo riassume in sostanza ciò che mi piacerebbe accadere.

Ecco cosa ho, e mentre il programma non esplode su un numero intero non intenzionale, voglio che l'utente sia informato che un intero non intenzionale è fondamentalmente privo di senso.

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() 

E l'output:

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

uscita con un valore negativo:

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

Ora, ovviamente potrei solo aggiungere un caso per determinare if args.games è negativo, ma ero curioso di sapere se c'era un modo per intrappolarlo al livello argparse, in modo da sfruttare la stampa automatica.

Idealmente, sarebbe stampare qualcosa di simile a questo:

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' 

Come così:

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' 

Per ora sto facendo questo, e credo di essere felice:

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

Answer 1

Questo dovrebbe essere possibile utilizzando type. Avrai ancora bisogno di definire un metodo effettivo che decide questo per voi:

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) 

Questo è fondamentalmente solo un esempio tratto dal funzione perfect_square nel docs su argparse.

Answer 2

Il modo più rapido e sporco, se si dispone di un massimo prevedibile così come min per il vostro arg, è utilizzare choices con una gamma

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

Answer 3

type sarebbe l'opzione consigliata per gestire condizioni/controlli, come nella risposta di Yuushi.

Nel tuo caso specifico, è possibile anche utilizzare il parametro choices se il limite superiore è anche conosciuto:

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

Answer 4

un'alternativa più semplice, soprattutto se sottoclasse argparse.ArgumentParser, è quello di avviare la convalida da dentro il metodo parse_args .

All'interno di un tale sottoclasse:

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 

Questa tecnica non può essere freddo come un callable personalizzato, ma non il lavoro.

---

Circa 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.

---

credito: answer by jonatan