如何使用Typer库解析日期时间字符串且避免datetime转字符串
解决Typer函数中datetime类型无需手动转字符串的方案
核心原理
你当前代码将target_date的类型声明为Optional[str],且默认值直接转成了ISO字符串,才需要手动做类型转换。只要把参数类型改为datetime,利用Typer的参数解析能力自动处理字符串到datetime的转换即可。
方案1:自定义解析回调(兼容所有Typer版本)
首先补全必要的依赖导入,新增日期解析函数,修改参数声明:
from datetime import datetime from typing import Optional # 你的原代码漏掉了这个导入 import typer TODAYS_DATE = datetime.now() # 自定义日期解析逻辑 def parse_date(input_str: Optional[str]) -> datetime: if input_str is None: # 无传参时直接返回datetime类型的当前日期 return TODAYS_DATE # 自动解析输入的ISO格式字符串为datetime return datetime.fromisoformat(input_str) def main( s3_bucket: str, # 类型直接声明为datetime,用typer.Option绑定解析函数 target_date: datetime = typer.Option(None, parser=parse_date), region: str = "us-east-1"): # 此处直接使用datetime类型的target_date,无需额外转换 print(f"target_date类型:{type(target_date)},值:{target_date}") pass if __name__ == "__main__": typer.run(main)
如果你需要支持
年-月-日这类非ISO格式的输入,只需要修改parse_date的解析逻辑,比如替换为datetime.strptime(input_str, "%Y-%m-%d")即可。
方案2:基于Pydantic自动解析(Typer >= 0.12.0 适用)
新版本Typer默认集成Pydantic类型校验,无需手写解析函数,直接声明类型即可:
from datetime import datetime from typing import Optional import typer TODAYS_DATE = datetime.now() def main( s3_bucket: str, # 直接用datetime作为类型,默认值传datetime对象即可 target_date: datetime = TODAYS_DATE, region: str = "us-east-1"): print(f"target_date类型:{type(target_date)},值:{target_date}") pass if __name__ == "__main__": typer.run(main)
Typer会自动将命令行输入的日期字符串解析为datetime类型,全程不需要手动做类型转换。
内容的提问来源于stack exchange,提问作者AlexLordThorsen
相关产品推荐
相关产品推荐

