python 小点心---获取当前函数名

2023-01-31 02:01:53 python 小点心

python 具有强大的自省能力,在函数运行时,可以在函数内部获取到当前所在的函数名称,请看示例代码

#coding=utf-8
import sys
import inspect

def my_name():
    print '1' ,sys._getframe().f_code.co_name
    print '2' ,inspect.stack()[0][3]

def get_current_function_name():
    print '5', sys._getframe().f_code.co_name
    return inspect.stack()[1][3]
class MyClass:
    def function_one(self):
        print '3',inspect.stack()[0][3]
        print '4', sys._getframe().f_code.co_name
        print "6 %s.%s invoked"%(self.__class__.__name__, get_current_function_name())

if __name__ == '__main__':
    my_name()
    myclass = MyClass()
    myclass.function_one()

示例演示了两种获取当前所在函数名称的方法,一种是用sys内置模块,一种是使用inspect模块。运行结果如下:

1 my_name
2 my_name
3 function_one
4 function_one
5 get_current_function_name
6 MyClass.function_one invoked
sys.getframe().f_code.co_name 方法永远获取当前所在的函数名称,inspect.stack()方法相对更灵活一些,在get_current_function_name函数中,sys获取的函数名称是get_cu

rrent_function_name,而inspect方法返回的结果是function_one。inspect.stack记录了当前的栈内信息,想进一步了解可以打印inspect.stack()信息。

我在function_one函数中调用了get_current_function_name,因此inspect.stack()返回的list中的第一个元组是有关get_current_function_name信息的,

第二个元组才是function_one的相关信息。

相关文章