[python] 실행 명령어에 파라메터 추가하기

python 파라메터 추가

test.py

if __name__ == '__main__' :
    print('hello~')

이런 코드가 있을 때 실행 명령은

$ python test.py

이렇게 실행하죠.

아래와 같이 실행명령어 뒤에 파라메터를 추가하려면

$ python test.py --name bryan -a 100

 

argparse 를 import 합니다.

import argparse

그리고 사용 소스는

import argparse

if __name__ == '__main__' :

    parser = argparse.ArgumentParser()
    parser.add_argument("--name", "-n", help="user name", default='jjang-gu')
    parser.add_argument("--age", "-a", type=int, help="user age", default=6)

    args = parser.parse_args()

    print('hello~ {} : {}'.format(args.name, args.age))

이렇게 하면 아래의 명령어로 실행했을 때 결과값입니다.

python test.py -n bryan --age 100
# 결과 : hello~ bryan : 100

python test.py
# 결과 : hello~ jjang-gu : 6

 

타입/기본값을 지정할 수 있습니다.

그리고 help="" 를 작성한것은 python test.py -h 라는 명령어를 쳤을 때 보여집니다.

$ python test.py -h

usage: test.py [-h] [--name NAME] [--age AGE]

optional arguments:
  -h, --help            show this help message and exit
  --name NAME, -n NAME  user name
  --age AGE, -a AGE     user age