如何读取命令行参数传递的txt

Question

我想使用argparse模块通过命令行参数传递txt文件，并使用open()函数读取文件，但是我遇到的第一件事是此错误：

AttributeError: 'dict' object has no attribute 's'

这是我的代码

import argparse
parser = argparse.ArgumentParser(description="My program!", formatter_class=argparse.RawTextHelpFormatter)
parser.add_argument("-s", type=argparse.FileType('r'), help="Filename/path to be passed")
args = vars(parser.parse_args())
subfile = (open(args.s, "r")).read()

Answer 1

您可以直接从args['s']读取，因为type=argparse.FileType('r') 。 不需要open() ：

import argparse
parser = argparse.ArgumentParser(description="My program!", formatter_class=argparse.RawTextHelpFormatter)
parser.add_argument("-s", type=argparse.FileType('r'), help="Filename/path to be passed")
args = vars(parser.parse_args())
data = args['s'].readlines()
print(data)

现在您可以从命令提示符处调用脚本，例如python .\\test.py -s'D:/test.txt'其中test.txt包含两行带有字母'abc'的行：

# prints ['abc\n', 'abc\n']

编辑-您可以简化代码以

parser = argparse.ArgumentParser(description="My program!", formatter_class=argparse.RawTextHelpFormatter)
parser.add_argument("-f", type=argparse.FileType('r', encoding='UTF-8'), help="path/filename of file to be passed")
data = parser.parse_args().f.readlines()

变化：

• 将参数名称更改为f ，我觉得它更适合文件名输入
• 添加并编码为type=argparse.FileType('r') -否则，它将采用操作系统默认值（例如，在Windows上为cp1252）。 我认为最好明确指定编码。
• 直接使用parser.parse_args().f访问io.TextIOWrapper ，而不是先使用vars()创建dict对象

Answer 2

替换最后一行：

subfile = (open(args.s, "r")).read()

与：

with open( args['s'] ) as fin :
    subfile = fin.read()

事情开始神奇地起作用，并且没有特殊原因，您没有打开的文件在附近徘徊。