1

我正在尝试编写一个小型图形解析器,该解析器从以下几行读取stdin和写入处理后的输出stdout

# parser.py
G = defaultdict(list)
for line in sys.stdin:
    node, neighbor = line.split()
    G[node].append(neighbor)
print(G)

我想调用脚本python -i parser.py < data.txt并与我创建的对象进行交互,但是即使我使用该-i选项调用 Python,解释器也会在代码运行后退出。注意同样的事情发生在ipython; 它甚至向我证实了我“真的想退出”。

一种解决方法是编写代码以使用作为参数传入的特定文件,但我想知道是否有办法让 Python 不退出上面示例中的解释器。

4

1 回答 1

3

REPL(交互式控制台)在耗尽标准输入时退出。通常,标准输入是控制台,因此只有在您键入 ^D 或quit()手动调用时才会退出。但是如果你从一个文件重定向标准输入,当你到达文件末尾时标准输入将被耗尽。

您可以使用argparse在命令行上接受文件,默认为sys.stdin

parser = argparse.ArgumentParser()
parser.add_argument('input', type=argparse.FileType(), nargs='?', default=sys.stdin)
args = parser.parse_args()
G = defaultdict(list)
for line in args.input:
    ...
于 2015-02-11T20:30:33.217 回答