Question

使用type方法的argparse.add_argument参数，您可以要求参数为可读文件：

parser.add_argument('--sqlite-file', type=argparse.FileType('r'))

作为指定此类型的一个好处，argparse会检查是否可以读取文件，如果没有，则会向用户显示错误。

有没有办法获取传递的文件名而不是io.TextIOWrapper或io.BufferedReader的实例？

由于文件名出现在解析器（'sqlite_file': <_io.TextIOWrapper name='data/export.sqlite' ...或'sqlite_file': <_io.BufferedReader name='data/export.sqlite' ...>）的字符串表示中，因此应该可以。

怎么做？

Answer 1

是的，使用文件对象上的.name attribute。

演示：

>>> import argparse
>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--sqlite-file', type=argparse.FileType('r'))
_StoreAction(option_strings=['--sqlite-file'], dest='sqlite_file', nargs=None, const=None, default=None, type=FileType('r'), choices=None, help=None, metavar=None)
>>> args = parser.parse_args(['--sqlite-file', '/tmp/demo.db'])
>>> args.sqlite_file
<_io.TextIOWrapper name='/tmp/demo.db' mode='r' encoding='UTF-8'>
>>> args.sqlite_file.name
'/tmp/demo.db'

如何为此参数指定type = FileType（...）时使用argparse获取文件名

1 个答案: