Python获取当前函数名|极客教程

Python获取当前函数名

在Python中，有时候我们需要获取当前正在执行的函数的名称。这在调试和日志记录中特别有用。本文将详细介绍如何使用不同的方法获取当前函数名。

方法一：使用inspect模块

Python的inspect模块为我们提供了很多有用的函数，包括获取函数名的功能。我们可以使用inspect模块中的currentframe()和getframeinfo()函数来获取当前函数名。

import inspect

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

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

test_function()

运行以上代码，输出为：

Current function name is: test_function

在这个示例中，我们定义了一个名为get_current_function_name()的函数，该函数通过inspect模块获取了当前函数的名称。然后我们创建了一个名为test_function()的函数，并在其中调用了get_current_function_name()函数来获取当前的函数名。

方法二：使用装饰器

另一种常见的方法是使用装饰器来获取当前函数名。装饰器可以在函数执行前或执行后动态地修改函数或获取函数的相关信息。我们可以定义一个装饰器来获取函数名并将其传递给函数。

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

@get_function_name_decorator
def test_function():
    print("This is a test function.")

test_function()

运行以上代码，输出为：

Current function name is: test_function
This is a test function.

在这个示例中，我们定义了一个名为get_function_name_decorator的装饰器，它接受一个函数作为参数并返回另一个函数。这个返回的函数在执行时会打印当前函数的名称，并继续执行原始函数。

然后我们使用@语法将test_function函数与装饰器关联起来，这样在调用test_function函数时，装饰器会获取函数名并打印出来。

方法三：使用sys模块

除了inspect模块和装饰器，我们还可以使用sys模块来获取当前函数的名称。sys模块提供了一些系统相关的功能，其中有一个叫做_getframe()的函数可以帮助我们获取当前函数信息。

import sys

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

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

test_function()

运行以上代码，输出为：

Current function name is: test_function

在这个示例中，我们定义了一个名为get_current_function_name()的函数，该函数通过sys._getframe()来获取当前函数的名称。然后我们创建了一个名为test_function()的函数，并在其中调用了get_current_function_name()函数来获取当前的函数名。