Python获取当前函数名|极客笔记

Python获取当前函数名

在编程过程中，有时候我们希望在代码中获取当前函数的名称，这在调试、日志记录等场景下非常有用。Python提供了几种方法来获取当前函数的名称，本文将详细介绍这些方法。

1. 使用`inspect`模块

inspect模块是Python标准库中的一个工具模块，它提供了许多有用的函数，可以用来检查模块、类、函数等对象的信息。我们可以使用inspect模块的currentframe函数来获取当前的调用栈帧，然后通过FrameInfo对象的function属性获取当前函数的名称。

下面是一个示例：

import inspect

def get_current_function_name():
    return inspect.currentframe().f_back.f_code.co_name

def func():
    print("Current function name is:", get_current_function_name())

func()

运行以上代码，得到的输出为：

Current function name is: func

在上面的代码中，我们定义了一个get_current_function_name函数，它通过inspect模块获取当前的调用栈帧，并返回当前函数的名称。然后在func函数中调用get_current_function_name函数，输出当前函数的名称。

2. 使用`sys`模块

另一种获取当前函数名称的方法是使用sys模块。我们可以通过sys模块的_getframe函数来获取当前的调用栈帧，然后通过frame对象的f_code属性和co_name属性来获取当前函数的名称。

下面是一个示例代码：

import sys

def get_current_function_name():
    return sys._getframe().f_code.co_name

def func():
    print("Current function name is:", get_current_function_name())

func()

运行以上代码，得到的输出为：

Current function name is: func

在上面的代码中，我们定义了一个get_current_function_name函数，它通过sys模块获取当前的调用栈帧，并返回当前函数的名称。然后在func函数中调用get_current_function_name函数，输出当前函数的名称。

3. 使用`decorator`装饰器

除了上面介绍的两种方法，我们还可以使用decorator装饰器来获取当前函数的名称。我们可以定义一个get_function_name装饰器，在装饰器中通过inspect或sys模块获取当前函数的名称，并将其作为参数传递给被装饰的函数。

下面是一个使用decorator装饰器获取当前函数名称的示例：

import inspect

def get_function_name(func):
    def wrapper(*args, **kwargs):
        print("Current function name is:", func.__name__)
        return func(*args, **kwargs)
    return wrapper

@get_function_name
def func():
    print("Inside func")

func()

运行以上代码，得到的输出为：

Current function name is: func
Inside func

在上面的代码中，我们定义了一个get_function_name装饰器，它接受一个函数作为参数，并返回一个包装函数wrapper。在wrapper函数中通过func.__name__获取被装饰函数的名称，并输出到控制台。然后我们使用decorator装饰器将func函数装饰以获取当前函数的名称。