1

我正在尝试使以下代码适用于终端和 IDLE shell。它旨在将标准输出拆分为屏幕和日志文件。

class Tee(object):
    def __init__(self, name, mode):
        self.filename = open(name, mode)
        self.stdout = sys.stdout
        sys.stdout = self
    def __del__(self):
        sys.stdout = self.stdout
        self.filename.close()
    def write(self, data):
        self.stdout.write(data)
        self.filename.write(data)

Tee('logfile.log', 'wb')

print 'test123'

如果我从终端运行此代码，它工作正常。如果我从 IDLE 执行它，它会按原样打印，但日志文件中没有文本，也不会引发异常。为什么这可以从命令提示符而不是 IDLE 中工作？我该怎么做才能让它在这两种环境中工作？我在 Win7 和 Python 2.7 上。

编辑：我把它放在一起用于 with 语句。这是我在课堂上的第一次尝试，因此欢迎提出改进建议。

newlog = True
class Tee(object):
    def __init__(self, name, mode=''):
        global newlog
        if newlog:
            self.filename = open(name, 'w'+mode)
        else:
            self.filename = open(name, 'a'+mode)
        newlog = False
    def __enter__(self):
        self.stdout = sys.stdout
        sys.stdout = self
        return self
    def __exit__(self, name, mode, data):
        sys.stdout = self.stdout
        self.filename.close()
    def __del__(self):
        try:
            self.filename.close()
            sys.stdout = self.stdout
        except:
            pass
    def startlog(self):
       self.stdout = sys.stdout
       sys.stdout = self
    def stoplog(self):
        self.filename.close()
        sys.stdout = self.stdout 
    def write(self, data):
        self.stdout.write(data)
        self.filename.write(data)
        self.filename.flush()


with Tee('logfile.log', 'b'):
    print 'I am the walrus'

print 'coocoocachoo'

Tee = Tee('logfile.log')
Tee.startlog()
print 'I am the eggman'
Tee.stoplog()

print 'coocoocachoo'
4

1 回答 1

2

logfile.log 不是空的，它只是没有被刷新。尝试：

sys.stdout.filename.flush()

你能做的就是写一个更完整的包装器。想法-您甚至可以以更通用的方式执行此操作，例如（此处可能需要解决一些问题）：

import functools
class Tee(object):
    def __init__(self, name, mode)
    .
    .
    for name in ('write', 'flush', ...):
        setattr(self, name, functools.partial(self._wrapper, name))

    def _wrapper(func_name, *args, **kw):
        getattr(self.filename, name)(*args, **kw)
        getattr(self.stdout, name)(*args, **kw)
于 2013-03-12T17:26:11.047 回答