主页 / 问题 / 709220
Accepted
nick_gabpe
Asked:2020-08-23 01:57:48 +0000 UTC2020-08-23 01:57:48 +0000 UTC

如何同时写入标准输出和文件？

  • 772

有一个python脚本：

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

它将所有标准输出写入文件。问题：如何同时写入文件并输出到控制台？

python

4 个回答

  1. Best Answer

    作为一个从字面上执行问题中提出的任务的下意识的解决方案：您可以创建一个接受两个文件对象的类，并且在写入此类的对象时，写入两个文件对象。

    import sys    

class DoubleWrite:
    def __init__(self, file1, file2):
        self.file1 = file1
        self.file2 = file2

    def write(self, s):
        self.file1.write(s)
        self.file2.write(s)

    def flush(self):
        self.file1.flush()
        self.file2.flush()

logfile = open('my_log.log', 'w')
sys.stdout = DoubleWrite(sys.stdout, logfile)

print "test"

    通常，建议使用现有的日志记录解决方案，尤其是logging.

    更新。

    为了在 Python 3 中正确重定向和后续恢复sys.stdout，最好使用上下文管理器contextlib.redirect_stdout。不幸的是，Python 2 标准库中没有这样的上下文管理器。

    使用示例（对于 Python 3）：

    import sys
from contextlib import redirect_stdout

class DoubleWrite:
    ...

with open('my_log.log', 'w') as logfile:
    with redirect_stdout(DoubleWrite(sys.stdout, logfile)):
        print('test')

    退出区块后，该with redirect_stdout(...)值将被sys.stdout恢复。

    • 11

  2. 使用模块logging：

    import sys

from logging import FileHandler
import logging


def get_logger(name=__file__, file='log.txt', encoding='utf-8'):
    log = logging.getLogger(name)
    log.setLevel(logging.DEBUG)

    # Будут строки вида: "[2017-08-23 09:54:55,356] main.py:34 DEBUG    foo"
    # formatter = logging.Formatter('[%(asctime)s] %(filename)s:%(lineno)d %(levelname)-8s %(message)s')
    formatter = logging.Formatter('%(message)s')

    # В файл
    fh = FileHandler(file, encoding=encoding)
    fh.setFormatter(formatter)
    log.addHandler(fh)

    # В stdout
    sh = logging.StreamHandler(stream=sys.stdout)
    sh.setFormatter(formatter)
    log.addHandler(sh)

    return log


log = get_logger()
log.debug('foo')
log.debug('bar')

    您可以通过指定 Formatter 来自定义输出格式'[%(asctime)s] %(filename)s:%(lineno)d %(levelname)-8s %(message)s'并获取以下字符串：

    [2017-08-23 09:54:55,356] main.py:34 DEBUG    foo
[2017-08-23 09:54:55,356] main.py:34 DEBUG    bar
    • 5

  3. 代理对列表全部内容的调用

    import sys


class CallList(list):
    def __getattr__(self, attr):
        def func(*args, **kwargs):
            for obj in self:
                getattr(obj, attr)(*args,**kwargs)
        return func


cl = CallList([open('file.test', 'w'), sys.stdout])
cl.write("0xDEFACE")
cl.flush()
    • 3 
  4. def echo(file):
    with open(file, 'w') as log:
        while True:
            text = yield
            log.write(text + '\n')
            print(text)

e = echo('log.txt')
next(e)
e.send('123')
e.send('456')
    • 2

相关问题