Tengo una secuencia de comandos de python que requiere algunas entradas de línea de comandos y estoy usando argparse para analizarlas. Encontré la documentación un poco confusa y no pude encontrar una manera de verificar un formato en los parámetros de entrada. Lo que quiero decir al verificar el formato se explica con este script de ejemplo:
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__)))
Necesito verificar las opciones -s
y -e
que la entrada del usuario tenga el formato YYYY-MM-DD
. ¿Hay una opción en argparse que no conozco que logra esto?
Según la documentación :
El argumento de palabra clave de
type
add_argument()
permite que se realicen las comprobaciones de tipo y las conversiones de tipo necesarias …type=
puede tomar cualquier llamada que tome un solo argumento de cadena y devuelva el valor convertido
Podrías hacer algo como:
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)
Entonces usa eso como type
:
parser.add_argument("-s", "--startdate", help="The Start Date - format YYYY-MM-DD", required=True, type=valid_date)
Solo para agregar a la respuesta anterior, puede usar una función lambda si desea mantenerla en una sola línea. Por ejemplo:
parser.add_argument('--date', type=lambda d: datetime.strptime(d, '%Y%m%d'))
Hilo viejo, pero la pregunta seguía siendo relevante para mí al menos
Para otros .fromisoformat
que .fromisoformat
esto a través de los motores de búsqueda: en Python 3.7, puede usar el método de clase estándar .fromisoformat
lugar de reinventar la rueda para fechas compatibles con ISO-8601, por ejemplo:
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)