입력 인수의 형식 지정 argparse python


111

일부 명령 줄 입력이 필요한 파이썬 스크립트가 있으며 구문 분석을 위해 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__)))

나는 옵션 확인해야 -s하고 -e사용자가 입력이 형식으로되어 있는지 YYYY-MM-DD. 내가 이것을 수행하는 것을 모르는 argparse 옵션이 있습니까?

답변:


235

문서 :

type키워드 인수는 add_argument()필요한 모든 유형 검사 및 유형 변환을 수행 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)

편집 한 다른 질문에 대해 문서 (있는 경우)를 알려줄 수 있습니까?
Sohaib 2014-08-24

valid_date()주어진 인수가 문자열이라고 가정 합니까 ?
Stevoisiak

1
@StevenVascellaro 예,하지만 이것이 API가 정의하는 것입니다. 문서의 인용문 당 "단일 문자열 인수를받습니다" .
jonrsharpe

71

위의 답변에 추가하기 위해 한 줄로 유지하려면 람다 함수를 사용할 수 있습니다. 예를 들면 :

parser.add_argument('--date', type=lambda d: datetime.strptime(d, '%Y%m%d'))

오래된 스레드이지만 질문은 여전히 ​​저와 관련이 있습니다!


6
당신이 뭔가 잘못을 전달하면 오류 메시지가 바로 될 것입니다하지만이 멋진 트릭invalid <lambda> value
브래드 솔로몬

38

검색 엔진을 통해이 문제를 해결하는 다른 사용자를 위해 : Python 3.7에서 .fromisoformatISO-8601 호환 날짜에 대한 휠을 다시 발명하는 대신 표준 클래스 메서드를 사용할 수 있습니다. 예 :

parser.add_argument('-s', "--startdate",
    help="The Start Date - format YYYY-MM-DD",
    required=True,
    type=datetime.date.fromisoformat)
parser.add_argument('-e', "--enddate",
    help="The End Date format YYYY-MM-DD (Inclusive)",
    required=True,
    type=datetime.date.fromisoformat)

5
또한 dateutil ISO 파서는 여전히 ISO 8601-2004를 준수하는 여러 추가 fmt를 사용합니다. 당신은 사용할 수 있습니다 import dateutil.parser-> type=dateutil.parser.isoparsetype=dateutil.parser.parse
브래드 솔로몬

@BradSolomon 실제로 좋은 제안 (따라서 내 +1)이지만 dateutil타사 패키지입니다. 추가 설치의 (작은) 번거 로움에 대한 이점을 판단해야 할 것입니다. YMMV.
Laryx Decidua
당사 사이트를 사용함과 동시에 당사의 쿠키 정책개인정보 보호정책을 읽고 이해하였음을 인정하는 것으로 간주합니다.
Licensed under cc by-sa 3.0 with attribution required.