python - 是否有破解 pdb 内置 print 的技巧？

Question

基本上，标题。

我试图追踪在大型代码库中发生虚假打印的位置，并且我想中断，或者每当打印“发生”时以某种方式获取堆栈跟踪。有任何想法吗？

Answer 1

对于这种特殊情况，您可以重定向stdout到打印输出及其调用者的帮助程序类。你也可以打破它的一种方法。

完整示例：

import sys
import inspect

class PrintSnooper:
    def __init__(self, stdout):
        self.stdout = stdout
    def caller(self):
        return inspect.stack()[2][3]
    def write(self, s):
        self.stdout.write("printed by %s: " % self.caller())
        self.stdout.write(s)
        self.stdout.write("\n")

def test():
    print 'hello from test'

def main():
    # redirect stdout to a helper class.
    sys.stdout = PrintSnooper(sys.stdout)
    print 'hello from main'
    test()

if __name__ == '__main__':
    main()

输出：

printed by main: hello from main
printed by main: 

printed by test: hello from test
printed by test: 

inspect.stack()如果您需要更详尽的信息，也可以只打印。

Answer 2

我能想到的唯一方法是替换sys.stdout，例如用codecs.getwriter('utf8'). 然后你可以write在 pdb 的方法上设置断点。或者write用调试代码替换它的方法。

import codecs
import sys

writer = codecs.getwriter('utf-8')(sys.stdout) # sys.stdout.detach() in python3
old_write = writer.write

def write(data):
    print >>sys.stderr, 'debug:', repr(data)
    # or check data + pdb.set_trace()
    old_write(data)

writer.write = write
sys.stdout = writer

print 'spam', 'eggs'