파이썬 입력 인수의 형식 지정 argparse python

일부 명령 줄 입력이 필요한 파이썬 스크립트가 있으며 구문 분석을 위해 argparse를 사용하고 있습니다. 설명서가 약간 혼란스럽고 입력 매개 변수의 형식을 확인하는 방법을 찾을 수 없습니다. 형식을 확인한다는 의미는이 예제 스크립트를 통해 설명됩니다.

parser.add_argument('-s', "--startdate", help="The Start Date - format YYYY-MM-DD ", required=True)
parser.add_argument('-e', "--enddate", help="The End Date format YYYY-MM-DD (Inclusive)", required=True)
parser.add_argument('-a', "--accountid", type=int, help='Account ID for the account for which data is required (Default: 570)')
parser.add_argument('-o', "--outputpath", help='Directory where output needs to be stored (Default: ' + os.path.dirname(os.path.abspath(__file__)))

사용자의 입력이 YYYY-MM-DD 형식인지 -s 및 -e 옵션을 확인해야합니다. 내가 이것을 수행하는 것을 모르는 argparse 옵션이 있습니까?

 

해결 방법

 

add_argument () 의 type 키워드 인수를 사용하면 필요한 유형 검사 및 유형 변환을 수행 할 수 있습니다 ... type = 은 모든 호출 가능 단일 문자열 인수를 취하고 변환 된 값을 반환합니다.

다음과 같이 할 수 있습니다.

def valid_date(s):
    try:
        return datetime.strptime(s, "%Y-%m-%d")
    except ValueError:
        msg = "Not a valid date: '{0}'.".format(s)
        raise argparse.ArgumentTypeError(msg)

그런 다음 type 으로 사용합니다.

parser.add_argument("-s", 
                    "--startdate", 
                    help="The Start Date - format YYYY-MM-DD", 
                    required=True, 
                    type=valid_date)

 

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