python - IPython：如何将某些内容通过管道传输到 Python 脚本中

Question

我知道我可以通过 IPython 在 IPython 中运行脚本run test.py并从那里进行调试。

但是如何将输出通过管道传输到 test.py 中？例如，通常我可以在命令行中运行grep "ABC" input.txt | ./test.py，但如何在 IPython 中执行相同的操作？

谢谢！

Answer 1

在 Python 脚本中，您应该从 sys.stdin 中读取：

import sys

INPUT = sys.stdin

def do_something_with_data(line):
    # Do your magic here
    ...
    return result

def main():
    for line in INPUT:
        print 'Result:', do_something_with_data(line)

if __name__ == '__main__':
    main()

在迭代解释器中，您可以使用子流程模块 mock sys.stdin。

In[0]: from test.py import *
In[1]: INPUT = subprocess.Popen(['grep', 'ABC', 'input.txt'], \
                               stdout=subprocess.PIPE).stdout
In[2]: main()

您还可以将输出通过管道传输到文件，然后从文件中读取。出于实际目的，stdin 只是另一个文件。

In[0]: ! grep "ABC" input.txt > output.txt
In[1]: INPUT = open('output.txt')
In[2]: main()