2012-05-24 90 views

回答

3

對於這種特殊的情況下,你可以重定向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()如果你需要更全面的信息。

+0

+1 - 漂亮的劈! – ThiefMaster

+0

太好了......我希望有一種方法可以對任意內建函數執行此操作,但是我可以使用此功能追蹤打印內容。非常感謝! – user961826

1

我能想到的唯一的一點就是用codecs.getwriter('utf8')返回的streamwriter代替sys.stdout。然後你可以在pdb中設置write方法的斷點。或用調試代碼替換它的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' 
相關問題