指定 Python 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__)))

我需要检查选项 -s和 -e，用户输入的格式是 YYYY-MM-DD。在 argparse中是否有一个我不知道的选项可以实现这一点？

64301 次浏览
小开
最佳答案

每间 文件:

add_argument()的 type关键字参数允许执行任何必要的类型检查和类型转换... ... type的参数可以是任何接受单个字符串的可调用参数。

你可以这样做:

def valid_date(s):
try:
return datetime.strptime(s, "%Y-%m-%d")
except ValueError:
msg = "not a valid date: {0!r}".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
)
小开

如果您希望将 lambda 函数保留为一行程序，那么只需在上面的答案中添加一个 lambda 函数即可。例如:

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

虽然是陈词滥调，但这个问题至少对我来说还是有意义的！

小开

对于那些通过搜索引擎找到这个方法的人来说，在 Python 3.7中，你可以使用标准的 .fromisoformat类方法来代替 ISO-8601兼容日期的重造轮子，例如:

import datetime


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)

提交答案