programing tip

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

itbloger 2020. 11. 25. 07:45
반응형

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

나는 볼 이이 optparse의에서 수행하는 방법 ,하지만 무엇을 (안-되지 않음) argparse 어떻습니까? 아니면 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.......>)

I order to have access to each parameter value, the following code may be helpful.

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

print "first parameter:" + str(args["arg"][0])
print "second parameter:" + str(args["arg"][1])
print "third parameter:" + str(args["arg"][2])

참고URL : https://stackoverflow.com/questions/9398065/python-argh-argparse-how-can-i-pass-a-list-as-a-command-line-argument

반응형