Come creare un argomento di tipo "lista di coppie" con argparse?

Question

Ho bisogno di lasciare che l'utente finale del mio pitone tipi di script qualcosa di simile:

script.py -sizes <2,2> <3,3> <6,6> 

dove ogni elemento dell'opzione -sizes è una coppia di due numeri interi positivi. Come posso ottenere questo con argparse?

Answer 1

definire un tipo personalizzato:

def pair(arg): 
    # For simplity, assume arg is a pair of integers 
    # separated by a comma. If you want to do more 
    # validation, raise argparse.ArgumentError if you 
    # encounter a problem. 
    return [int(x) for x in arg.split(',')] 

quindi utilizzare questo come il tipo per un argomento regolare:

p.add_argument('--sizes', type=pair, nargs='+') 

Poi

>>> p.parse_args('--sizes 1,3 4,6'.split()) 
Namespace(sizes=[[1, 3], [4, 6]]) 

Answer 2

Argparse non tenta di coprire tutti i possibili formati di dati di input. È sempre possibile ottenere sizes come stringa e analizzare con poche righe di codice:

import argparse 

parser = argparse.ArgumentParser() 
parser.add_argument('--sizes', nargs='+') 

args = parser.parse_args() 

try: 
    sizes = [tuple(map(int, s.split(',', maxsplit=1))) for s in args.sizes] 
except Exception: 
    print('sizes cannot be parsed') 

print(sizes) 

"casi speciali non sono abbastanza speciale per rompere le regole."