python argh / argparse: Как передать список в качестве аргумента командной строки?

Я пытаюсь передать список аргументов скрипту Python, используя библиотеку argh. Что-то, что может принимать такие входные данные:

./my_script.py my-func --argA blah --argB 1 2 3 4
./my_script.py my-func --argA blah --argB 1
./my_script.py my-func --argA blah --argB 

Мой внутренний код выглядит так:

import argh

@argh.arg('--argA', default="bleh", help='My first arg')
@argh.arg('--argB', default=[], help='A list-type arg--except it\'s not!')
def my_func(args):
    "A function that does something"

    print args.argA
    print args.argB

    for b in args.argB:
        print int(b)*int(b)  #Print the square of each number in the list
    print sum([int(b) for b in args.argB])  #Print the sum of the list

p = argh.ArghParser()
p.add_commands([my_func])
p.dispatch()

И вот как он себя ведет:

$ python temp.py my-func --argA blooh --argB 1
blooh
['1']
1
1

$ python temp.py my-func --argA blooh --argB 10
blooh
['1', '0']
1
0
1

$ python temp.py my-func --argA blooh --argB 1 2 3
usage: temp.py [-h] {my-func} ...
temp.py: error: unrecognized arguments: 2 3

Проблема кажется довольно простой: argh принимает только первый аргумент и обрабатывает его как строку . Как мне заставить его «ожидать» список целых чисел?

Я вижу , как это делается в optparse , но как насчет (не устаревшего) argparse? Или использовать гораздо более красивый синтаксис argh? Они кажутся намного более питоническими.

64
задан Community 23 May 2017 в 12:02
поделиться