在标准输出文件和控制台中打印结果

Question

我的要求是在控制台中打印文件的输出以及日志文件。以下代码为我做了这件事，除了一个小问题。我在文件末尾调用了一个 perl 脚本，其输出显示在控制台中，但没有打印到文件中。

import subprocess
import sys
class Tee(object):
    def __init__(self, *files):
        self.files = files
    def write(self, obj):
        for f in self.files:
            f.write(obj)

f = open('MyFile.txt', 'w')
original = sys.stdout
sys.stdout = Tee(sys.stdout, f)
print "Logging Started"
# My code
print "A"

subprocess.call(['./MyScript])
sys.stdout = original
print "Logging Stopped"  # Only on stdout
f.close()

谁能告诉我如何实现？或者有没有可能达到同样的效果？

Answer 1

下面的代码对我有用。感谢大家的帮助。

#!/usr/bin/python
import os
import subprocess
import sys
class Tee(object):
    def __init__(self, *files):
        self.files = files
    def write(self, obj):
        for f in self.files:
            f.write(obj)

f = open('MyFile.txt', 'w')
original = sys.stdout
sys.stdout = Tee(sys.stdout, f)
print "Logging Started"
# My code
print "A"

def check_output(*popenargs, **kwargs):
    process = subprocess.Popen(stdout=subprocess.PIPE, *popenargs, **kwargs)
    output, unused_err = process.communicate()
    retcode = process.poll()
    if retcode:
        cmd = kwargs.get("args")
        if cmd is None:
            cmd = popenargs[0]
        error = subprocess.CalledProcessError(retcode, cmd)
        error.output = output
        raise error
    return output

location = "%s/folder"%(os.environ["Home"])
myoutput = check_output(['./MyFile'])
print myoutput
sys.stdout = original
print "Logging Stopped"  # Only on stdout
f.close()

Answer 2

使用subprocess.check_output:

print subprocess.check_output(['./MyScript])

在 Python 2.6 中，使用反向端口 subprocess32，或复制 2.7 source for check_output。

Answer 3

如果你看check_output是在Python2.7中实现的，你应该就能知道如何使用subprocess.Popen

def check_output(*popenargs, **kwargs):
    if 'stdout' in kwargs:
        raise ValueError('stdout argument not allowed, it will be overridden.')
    process = Popen(stdout=PIPE, *popenargs, **kwargs)
    output, unused_err = process.communicate()
    retcode = process.poll()
    if retcode:
        cmd = kwargs.get("args")
        if cmd is None:
            cmd = popenargs[0]
        raise CalledProcessError(retcode, cmd, output=output)
    return output

Answer 4

subprocess.check_ouput

在python 2.7中引入，如果你可以使用以前的版本，你可以使用Popen并将stdout设置为你的输出流（但在你的情况下，你已经覆盖了sys.stdout，我认为它不需要)，只需更改为：

p = subprocess.Popen(['./MyScript'])