파이썬[Python] 스크립트 실행 시 인자값 사용하기

파이썬[Python] 스크립트 실행 시 인자값 사용하기

파이썬[Python] 스크립트 실행 시 인자값 받아오기

파이썬 코드를 받아서 실행하시다보면, 자연스럽게 argument를 다음과 같이 Argument를 사용하는 경우가 있습니다. 하지만, 무조건 모든 코드에 다음과 같이 Arugment를 사용할 수 있는 것은 아닙니다. 

python example.py a b c

그래서 이번 포스팅에서는 위와 같이 Argument를 사용할 수 잇는 방법에 대해서 이야기를 해보고자 합니다. 

 

일단, argument를 사용하기 위한 모듈은 여러 모듈이 있지만, 이번 포스팅에서는 필요한 만큼만 사용할 예정임으로 가장 간단하게 사용할 수 있는 sys 모듈을 사용하겠습니다. 그럼 간단히 살펴보겠습니다. 

import sys

arguments = sys.argv
print(arguments)

먼저, argument들을 사용하기 위해서 가장 쉽게 사용하는 방법입니다. 이와 같이 저장된 파일을 바탕으로 실행을 해보도록 하겠습니다. 

D:\BlogProject\Blog_ETC>python argument_example.py example.vs example
['argument_example.py', 'example.vs', 'example']

argument의 첫번째 값은 실행된 파일이고, 그다음 인자들은 입력된 부분들을 바탕으로 인자값을 출력합니다. 

 

물론 다른 모듈에 비해서 매우 간단합니다. 하지만 간단하기 때문에 옵션등을 적용하기 위해서는 몇가지 작업등이 요구됩니다. 그럼 옵션을 입력할 수 있게 만들어 보도록 하겠습니다. 

 

일단은 가장 쉽게 -h / -H옵션을 바탕으로 Help 명령어를 출력할 수 있게 작업을 할 예정입니다. 관련해서 다음과 같은 부분들은 코드로 살펴보겠습니다. 

import sys

def print_help():
    print("Argument Example Help")

if __name__ == '__main__':
    arguments = sys.argv
  
    if len(arguments) == 1:
        print("No Argument")
    elif arguments[1] == "-h" or arguments[1] == "-H":
        print_help()
    else :
        print("Unsupported Argument")

일단 총 3가지 관점에서 살펴봤습니다. 일단은 최초에 Argument가 없는 경우에 문제를 삼도록 하였습니다. 그리고 -h/-H가 argument로 들어오면 동작하게 하였습니다. 그외의 부분들에 대해서는 지원하지 않는 argument 처리를 하도록 하였습니다. 위의 코드를 바탕으로 실행을 해보도록 하겠습니다. 

D:\BlogProject\Blog_ETC>python argument_example.py -h
Argument Example Help

D:\BlogProject\Blog_ETC>python argument_example.py -va
Unsupported Argument

위와 같은 실행 결과가 나타나게됩니다. 물론 이 부분을 바탕으로 보다 많은 작업을 진행할 수 있을 것으로 보입니다. 

 

이번 포스팅에서는 파이썬[Python] 스크립트 실행 시 인자값 사용하기라는 주제로 간단히 파이썬 스크립트를 실행하면서 인자값을 사용하는 방법에 대해서 살펴봤습니다. 혹 궁금하신 점이나 문의 사항이 있의시면 언제든지 댓글 및 방명록에 글 남겨주시길 바랍니다.