将标准输出重定向到Python中的文件中？

Question

如何将stdout重定向到Python中的任意文件？

当一个长时间运行的Python脚本（例如，web应用程序）从ssh会话和backgounded中启动，并且ssh会话关闭时，应用程序将引发IOError并在试图写入stdout时失败。我需要找到一种方法，使应用程序和模块输出到一个文件，而不是标准输出，以防止由于IOError而失败。目前，我使用nohup将输出重定向到一个文件，这就完成了工作，但出于好奇，我想知道是否有一种方法可以在不使用nohup的情况下完成这项工作。

我已经尝试过sys.stdout = open('somefile', 'w')，但这似乎并不能阻止某些外部模块仍然输出到终端（或者sys.stdout = ...行根本没有触发）。我知道它应该从我测试过的更简单的脚本开始工作，但是我还没有时间在web应用程序上测试。

Answer 1

你可以试试这个

import sys

class Logger(object):
    def __init__(self, filename="Default.log"):
        self.terminal = sys.stdout
        self.log = open(filename, "a")

    def write(self, message):
        self.terminal.write(message)
        self.log.write(message)

sys.stdout = Logger("yourlogfilename.txt")
print "Hello world !" # this is should be saved in yourlogfilename.txt

Answer 2

如果要在Python脚本中进行重定向，那么将sys.stdout设置为file对象就可以：

import sys
sys.stdout = open('file', 'w')
print('test')

更常见的方法是在执行时使用shell重定向（在Windows和Linux上相同）：

$ python foo.py > file

Answer 3

Python 3.4中有^{} function：

from contextlib import redirect_stdout

with open('help.txt', 'w') as f:
    with redirect_stdout(f):
        print('it now prints to `help.text`')

它类似于：

import sys
from contextlib import contextmanager

@contextmanager
def redirect_stdout(new_target):
    old_target, sys.stdout = sys.stdout, new_target # replace sys.stdout
    try:
        yield new_target # run some code with the replaced stdout
    finally:
        sys.stdout = old_target # restore to the previous value

可以在早期的Python版本上使用。后一个版本不是reusable。如果需要的话可以做一个。

它不会在文件描述符级别重定向stdout，例如：

import os
from contextlib import redirect_stdout

stdout_fd = sys.stdout.fileno()
with open('output.txt', 'w') as f, redirect_stdout(f):
    print('redirected to a file')
    os.write(stdout_fd, b'not redirected')
    os.system('echo this also is not redirected')

b'not redirected'和'echo this also is not redirected'不会重定向到output.txt文件。

要在文件描述符级别重定向，可以使用os.dup2()：

import os
import sys
from contextlib import contextmanager

def fileno(file_or_fd):
    fd = getattr(file_or_fd, 'fileno', lambda: file_or_fd)()
    if not isinstance(fd, int):
        raise ValueError("Expected a file (`.fileno()`) or a file descriptor")
    return fd

@contextmanager
def stdout_redirected(to=os.devnull, stdout=None):
    if stdout is None:
       stdout = sys.stdout

    stdout_fd = fileno(stdout)
    # copy stdout_fd before it is overwritten
    #NOTE: `copied` is inheritable on Windows when duplicating a standard stream
    with os.fdopen(os.dup(stdout_fd), 'wb') as copied: 
        stdout.flush()  # flush library buffers that dup2 knows nothing about
        try:
            os.dup2(fileno(to), stdout_fd)  # $ exec >&to
        except ValueError:  # filename
            with open(to, 'wb') as to_file:
                os.dup2(to_file.fileno(), stdout_fd)  # $ exec > to
        try:
            yield stdout # allow code to be run with the redirected stdout
        finally:
            # restore stdout to its previous value
            #NOTE: dup2 makes stdout_fd inheritable unconditionally
            stdout.flush()
            os.dup2(copied.fileno(), stdout_fd)  # $ exec >&copied

如果使用stdout_redirected()而不是redirect_stdout()，则现在也可以使用相同的示例：

import os
import sys

stdout_fd = sys.stdout.fileno()
with open('output.txt', 'w') as f, stdout_redirected(f):
    print('redirected to a file')
    os.write(stdout_fd, b'it is redirected now\n')
    os.system('echo this is also redirected')
print('this is goes back to stdout')

只要stdout_redirected()上下文管理器处于活动状态，以前在stdout上打印的输出现在就转到output.txt。

注意：stdout.flush()不刷新 Python 3上的C stdio缓冲区，其中I/O直接在read()/write()系统调用上实现。要刷新所有打开的C stdio输出流，如果某些C扩展使用基于stdio的I/O，则可以显式调用libc.fflush(None)：

try:
    import ctypes
    from ctypes.util import find_library
except ImportError:
    libc = None
else:
    try:
        libc = ctypes.cdll.msvcrt # Windows
    except OSError:
        libc = ctypes.cdll.LoadLibrary(find_library('c'))

def flush(stream):
    try:
        libc.fflush(None)
        stream.flush()
    except (AttributeError, ValueError, IOError):
        pass # unsupported

可以使用stdout参数重定向其他流，而不仅仅是sys.stdout，例如，合并sys.stderr和sys.stdout：

def merged_stderr_stdout():  # $ exec 2>&1
    return stdout_redirected(to=sys.stdout, stdout=sys.stderr)

示例：

from __future__ import print_function
import sys

with merged_stderr_stdout():
     print('this is printed on stdout')
     print('this is also printed on stdout', file=sys.stderr)

注意：stdout_redirected()混合了缓冲I/O（通常是sys.stdout）和非缓冲I/O（直接对文件描述符进行操作）。小心，可能有bufferingissues。

要回答这个问题，您可以使用^{}对脚本进行守护，并使用logging模块（作为@erikb85 suggested）而不是print语句，只需重定向您现在使用nohup运行的长期运行的Python脚本的stdout。