Python 명령줄 인자 완벽 가이드 | sys.argv와 argparse 사용법

1. Python에서 명령줄 인수의 기본적인 사용법

명령줄 인수란?

sys.argv를 사용한 명령줄 인수 가져오기

샘플 코드: 간단한 계산 프로그램

import sys

def main():
    if len(sys.argv) < 3:
        print("오류: 인수가 2개 필요합니다")
        return

    num1 = float(sys.argv[1])
    num2 = float(sys.argv[2])
    print(f"합계: {num1 + num2}")

if __name__ == "__main__":
    main()

$ python script.py 3 5
합계: 8.0

2. argparse 모듈에서의 고급 인자 처리

argparse의 기본

인자 파싱과 파서 생성

샘플 코드: 기본적인 argparse의 사용법

import argparse

def main():
    parser = argparse.ArgumentParser(description='샘플 프로그램입니다')
    parser.add_argument('arg1', help='첫 번째 인자')
    parser.add_argument('arg2', type=float, help='두 번째 숫자 인자')
    args = parser.parse_args()

    print(f"arg1 = {args.arg1}")
    print(f"arg2 = {args.arg2}")

if __name__ == "__main__":
    main()

$ python script.py test 5.5
arg1 = test
arg2 = 5.5

옵션 인자와 플래그 처리

샘플 코드: 옵션 인자와 플래그 사용

import argparse

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('--verbose', action='store_true', help='상세 출력을 표시')
    parser.add_argument('numbers', nargs='+', type=float, help='숫자 목록')
    args = parser.parse_args()

    total = sum(args.numbers)
    if args.verbose:
        print(f"계산 상세: {args.numbers} 의 합계는 {total} 입니다")
    else:
        print(f"합계: {total}")

if __name__ == "__main__":
    main()

$ python script.py 1 2 3 --verbose
계산 상세: [1.0, 2.0, 3.0] 의 합계는 6.0 입니다

3. 오류 처리와 인수 검증

인수의 개수와 타입 검증

샘플 코드: 인수의 개수와 타입 검증

import sys

def main():
    if len(sys.argv) != 3:
        print("오류: 두 개의 인수가 필요합니다")
        return

    try:
        num1 = float(sys.argv[1])
        num2 = float(sys.argv[2])
    except ValueError:
        print("오류: 인수는 숫자여야 합니다")
        return

    print(f"합계: {num1 + num2}")

if __name__ == "__main__":
    main()

4. 실제 사용 사례와 응용

커맨드라인 도구 제작 예

샘플 코드: 파일의 줄 수를 세는 도구

import argparse

def main():
    parser = argparse.ArgumentParser(description='파일의 줄 수를 세는 도구')
    parser.add_argument('filename', help='세는 파일 이름')
    args = parser.parse_args()

    try:
        with open(args.filename, 'r') as file:
            lines = file.readlines()
            print(f"파일 {args.filename}의 줄 수: {len(lines)}")
    except FileNotFoundError:
        print(f"오류: 파일 {args.filename}을 찾을 수 없습니다")

if __name__ == "__main__":
    main()

$ python count_lines.py example.txt
파일 example.txt의 줄 수: 100

5. 정리