从别名函数中确定函数名

Question

如何确定一个函数是使用函数名还是该函数的别名调用的？

我可以通过以下方式检查函数以从函数体内获取其名称：

import inspect

def foo():
   print(inspect.stack()[0][3])

foo() # prints 'foo'

来源：Determine function name from within that function (without using traceback)

但是，如果我为函数设置别名并尝试相同的操作，我会得到原始函数名称（不是别名）

bar = foo
bar() # prints 'foo'

我希望能够做到以下几点：

def foo():
    print(... some code goes here ...)

bar = foo

foo() # prints 'foo'
bar() # prints 'bar'

Answer 1

我有一个（有点老套）的解决方案，它依赖于正则表达式从字符串中解析出函数名。可能有一个更清洁的解决方案，但至少使用 inspect only 这是我能找到的最好的。

import inspect
import re


function_from_call = re.compile("\w+(?=\(\))")


def foo():
    _, call_frame, *_ = inspect.stack()
    _, _, _, _, call, *_ = call_frame
    print(re.search(function_from_call, str(call)).group())

bar = foo
bar()  # prints bar
foo()  # prints foo

简短说明：首先，我正在抓取导致调用此函数的调用的检查帧。然后，我从这个框架中提取实际的调用字符串，并将一个正则表达式应用于这个调用字符串，它只给我们函数名称。

注意：在解释器 shell 中，inspect 的行为不同，我上面的代码会产生错误，因为我的正则表达式无法匹配实际的函数名称。 @user2357112 在对此问题的评论中指出了一个额外的警告：调用与名称直接相关并不明显，如

l = [foo]; l[0]()

当从脚本运行时，我的解决方案将正确处理简单的重命名案例（如本问题中给出的那样），但我不提倡使用它，因为上述极端案例会导致令人困惑的错误。

Answer 2

根据我对您问题范围的有限了解，这是可行的：

import inspect

def foo():
  print(inspect.stack()[1][4][0].strip())

foo()
bar = foo
bar()

结果：

foo()
bar()