파이썬 python argh / argparse : 목록을 명령 줄 인수로 전달하려면 어떻게해야합니까?

argh 라이브러리를 사용하여 Python 스크립트에 인수 목록을 전달하려고합니다. 다음과 같은 입력을받을 수있는 것 :

./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는 첫 번째 인수 만 받아들이고이를 문자열로 취급합니다. 대신 정수 목록을 "예상"하도록하려면 어떻게해야합니까?

 

해결 방법

 

argparse 에서는 type = int 만 사용하면됩니다.

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('-a', '--arg', nargs='+', type=int)
print parser.parse_args()

출력 예 :

$ python test.py -a 1 2 3
Namespace(arg=[1, 2, 3])

편집 : 나는 argh 에 익숙하지 않지만 argparse 를 둘러싼 래퍼 인 것 같고 이것은 나를 위해 일했습니다.

import argh

@argh.arg('-a', '--arg', nargs='+', type=int)
def main(args):
    print args

parser = argh.ArghParser()
parser.add_commands([main])
parser.dispatch()

출력 예 :

$ python test.py main -a 1 2 3
Namespace(arg=[1, 2, 3], function=<function main at 0x.......>)

 

참조 페이지 https://stackoverflow.com/questions/9398065