如何为需要文件名而不是变量的python子进程调用提供输入?

时间:2012-01-27 05:18:39

标签: python shell variables subprocess pipe

我正在尝试在python中调用shell脚本(segment.sh)。 在控制台上生成正确结果的语法是:

> ./segment.sh ctb file.txt utf-8 0

可以看出,这个shell脚本需要一个文本文件作为输入。我想要做的是打开文件并在python中读取它的内容(后来最好从HTML POST表单),并以某种方式将包含内容的变量传递给python子进程 调用

以下功能无效。但是如果我只提供一个文件名,例如:

Popen(["/bin/bash", "./segment.sh", "ctb", "file.txt", "utf-8", "0"])`

然后,它会工作,但我想从变量传递输入。你能给我一些指示吗?

def pySegment(text):
    op = subprocess.Popen(["/bin/bash", "./segment.sh", "ctb", "utf-8", "0"],
                      stdout = subprocess.PIPE,
                      stdin  = subprocess.PIPE,
                      stderr = subprocess.STDOUT,                          
                      )
    results = op.communicate(input=text)[0]
    return results

if __name__ == "__main__":
    filename = "./file.txt"
    text = open(filename).read()
    result = pySegment(text)
    print result

1 个答案:

答案 0 :(得分:5)

我建议使用named pipe

import os, tempfile, shutil, subprocess

temp_dir = tempfile.mkdtemp()
filename = os.path.join(temp_dir, 'file.txt')
text = '<text>'
os.mkfifo(filename)

try:
    subprocess.Popen(('segment.sh', 'ctf', filename, 'utf-8', '0'))
    with open(filename, 'w') as f:
        f.write(text)
finally:
    shutil.rmtree(temp_dir)

命名管道将提供相同的文件接口,而无需根据需要真正创建该文件。