在argparse中,我们可以使用argparse.FileType
来处理文件输入。这允许我们使用解析后的文件作为文件对象进行操作。下面是一个示例代码:
import argparse
def main(file):
# 使用解析后的文件对象进行操作
for line in file:
print(line.strip())
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('file', type=argparse.FileType('r'), help='输入文件')
args = parser.parse_args()
main(args.file)
在上面的代码中,我们定义了一个file
参数,类型为argparse.FileType('r')
。这将告诉argparse将该参数解析为文件对象,并以只读模式打开文件。然后,在main
函数中,我们可以直接使用file
对象进行操作。
你可以通过以下方式运行上述代码:
$ python script.py input.txt
这将打开input.txt
文件,并将其作为file
参数传递给main
函数。你可以根据自己的需求修改代码,例如修改文件模式、添加其他参数等。