问题:指定输入参数argparse python的格式
我有一个需要一些命令行输入的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中有一个选项可以完成此任务。
回答 0
根据文档:
的
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)
回答 1
只是为了补充上面的答案,如果您想将lambda函数保持为单一格式,则可以使用该函数。例如:
parser.add_argument('--date', type=lambda d: datetime.strptime(d, '%Y%m%d'))
旧线程,但问题至少仍然与我有关!
回答 2
对于其他通过搜索引擎实现此目标的人:在Python 3.7中,您可以使用标准的.fromisoformat
class方法,而不是为ISO-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)
声明:本站所有文章,如无特殊说明或标注,均为本站原创发布。任何个人或组织,在未征得本站同意时,禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益,可联系我们进行处理。