python筆記_查看函數調用棧的一個小技巧

一、背景

最近在看一個開源框架的源碼，涉及到的內容非常雜亂，有的函數不知道是在什么時候被誰給調用了？調用的時候傳入了什么參數？為了解決這個問題，寫了一個小的裝飾器。

二、實現

這個裝飾器函數主要參考了logging模塊中的findCaller，源碼如下：

def findcaller(func):
    def wrapper(*args,**kwargs):
        import sys
        f=sys._getframe()
        filename=f.f_back.f_code.co_filename
        lineno=f.f_back.f_lineno
        print '######################################'
        print 'caller filename is ',filename
        print 'caller lineno is',lineno
        print 'the passed args is',args,kwargs
        print '######################################'
        func(*args,**kwargs)
    return wrapper

只要加上這個裝飾器，就能在調用函數前看到這個函數被哪個文件中的第幾行調用，並且傳入的參數是什么。例子如下：

#caller.py
from class_A import A
a1=A()
a2=A(1)
a3=A(2)
a2.func(3)

#class_A.py
class A(object):
    def __init__(self,num=0):
        print num
    @findcaller
    def func(self,num=None):
        print num

#執行caller,得到結果如下：
######################################
caller filename is  caller.py
caller lineno is 5
the passed args is (<class_A.A object at 0xb7200a4c>, 3) {}
######################################

可以看到是caller.py中的第5行a2.func(3)調用到了func函數，傳入參數是3.