檢查一個函數在Python中使用打印函數

Question

我想在python中編寫一個裝飾器：如果一個被調用的函數包含print's，裝飾器會在調用該函數之前打印她的名字。我熟悉裝飾器的語法，但是我在檢查一個函數是否有print的問題。

def preceeding_name(func):   
    @wraps(func) 
    def wrapper(*args, **kwargs): 
     if 'print' in func: 
      print(func.__name__) 
     result = func(*args, **kwargs) 
     return result 
    return wrapper 

沒有必要檢查函數print是否真的被調用。

Answer 1

這可以通過保持緩衝完成從打印中檢查「打印」並查看打印是否已完成。

class Out(object): 
    def write(self,s): 
     self.s += s 
    def __init__(self) 
     self.s = '' 

我們檢查

def wrapper(*args, **kwargs): 
     our_out = Out() 
     sys.stdout = our_out 
     result = func(*args, **kwargs) 

     if len(our_out.s)>0: 
      sys.stdout = sys.__stdout__ 
      print func.__name__ 
      for s in our_out.s.split('\n'): 
       print s  

     return result 

Answer 2

我這種情況下，你可以重新打印

def preceeding_name(func): 
    @wraps(func) 
    def wrapper(*args, **kwargs): 

     old_print = print 

     def print(*arg, **kwarg): 
      old_print(func.__name__) 
      old_print(*arg, **kwarg) 

     result = func(*args, **kwargs) 
     return result 
    return wrapper 

編輯： 我測試這項工作

old_print = print 

def preceeding_name(func): 

    def print(*arg, **kwarg): 
     old_print(func.__name__, end='') 
     old_print(*arg, **kwarg) 

    def wrapper(*args, **kwargs): 
     print('') 
     result = func(*args, **kwargs) 
     return result 
    return wrapper 

@preceeding_name 
def a(): 
    print('hi') 


a() 

@preceeding_name 
def b(): 
    print('hi') 


b() 

EDIT2：

old_print = print 


def preceeding_name(func): 
    global print 
    def print(*arg, **kwarg): 
     old_print(func.__name__) 
     old_print(*arg, **kwarg) 

    def wrapper(*args, **kwargs): 
     result = func(*args, **kwargs) 
     return result 

    return wrapper 


@preceeding_name 
def a(): 
    print('hi') 


a() 


@preceeding_name 
def b(): 
    # print('hi') 
    pass 

b()