Python: как получить имя метода вызывающего абонента в вызываемом методе?
Предположим, у меня есть 2 метода:
def method1(self):
...
a = A.method2()
def method2(self):
...
Если я не хочу делать какие-либо изменения для метода1, как получить имя вызывающего (в этом примере имя is method1) в методе2?
inspect.getframeinfo и другие связанные функции inspect могут помочь:
>>> import inspect
>>> def f1(): f2()
...
>>> def f2():
... curframe = inspect.currentframe()
... calframe = inspect.getouterframes(curframe, 2)
... print('caller name:', calframe[1][3])
...
>>> f1()
caller name: f1
этот самоанализ призван помочь отладке и разработке; не рекомендуется полагаться на это в производственных целях.
Более короткая версия:
import inspect
def f1(): f2()
def f2():
print 'caller name:', inspect.stack()[1][3]
f1()
(благодаря @Alex и Stefaan Lippen)
Кажется, что все работает отлично:
import sys
print sys._getframe().f_back.f_code.co_name
Я придумал немного более длинную версию, которая пытается создать полное имя метода, включая модуль и класс.
https://gist.github.com/2151727 (версия 9cccbf)
# Public Domain, i.e. feel free to copy/paste
# Considered a hack in Python 2
import inspect
def caller_name(skip=2):
"""Get a name of a caller in the format module.class.method
'skip' specifies how many levels of stack to skip while getting caller
name. skip=1 means "who calls me", skip=2 "who calls my caller" etc.
An empty string is returned if skipped levels exceed stack height
"""
stack = inspect.stack()
start = 0 + skip
if len(stack) < start + 1:
return ''
parentframe = stack[start][0]
name = []
module = inspect.getmodule(parentframe)
# 'modname' can be None when frame is executed directly in console
# TODO(techtonik): consider using __main__
if module:
name.append(module.__name__)
# detect classname
if 'self' in parentframe.f_locals:
# I don't know any way to detect call from the object method
# XXX: there seems to be no way to detect static method call - it will
# be just a function call
name.append(parentframe.f_locals['self'].__class__.__name__)
codename = parentframe.f_code.co_name
if codename != '<module>': # top level usually
name.append( codename ) # function or a method
## Avoid circular refs and frame leaks
# https://docs.python.org/2.7/library/inspect.html#the-interpreter-stack
del parentframe, stack
return ".".join(name)
Немного об объединении вещей выше. Но здесь моя трещина.
def print_caller_name(stack_size=3):
def wrapper(fn):
def inner(*args, **kwargs):
import inspect
stack = inspect.stack()
modules = [(index, inspect.getmodule(stack[index][0]))
for index in reversed(range(1, stack_size))]
module_name_lengths = [len(module.__name__)
for _, module in modules]
s = '{index:>5} : {module:^%i} : {name}' % (max(module_name_lengths) + 4)
callers = ['',
s.format(index='level', module='module', name='name'),
'-' * 50]
for index, module in modules:
callers.append(s.format(index=index,
module=module.__name__,
name=stack[index][3]))
callers.append(s.format(index=0,
module=fn.__module__,
name=fn.__name__))
callers.append('')
print('\n'.join(callers))
fn(*args, **kwargs)
return inner
return wrapper
Использование:
@print_caller_name(4)
def foo():
return 'foobar'
def bar():
return foo()
def baz():
return bar()
def fizz():
return baz()
fizz()
вывод
level : module : name
--------------------------------------------------
3 : None : fizz
2 : None : baz
1 : None : bar
0 : __main__ : foo
Я бы использовал inspect.currentframe().f_back.f_code.co_name. Его использование не было рассмотрено ни в одном из предыдущих ответов, которые в основном относятся к одному из трех типов:
inspect.stack, но известно, что он слишком медленный.sys._getframe, которая является внутренней частной функцией, учитывая ее начальное подчеркивание, и поэтому ее использование неявно не рекомендуется.inspect.getouterframes(inspect.currentframe(), 2)[1][3], но совершенно неясно, к чему [1][3] обращается.import inspect
import types
from typing import cast
def caller_name() -> str:
"""Return the calling function name."""
# Ref: https://stackoverflow.com/a/57712700/
return cast(types.FrameType, inspect.currentframe()).f_back.f_code.co_name
if __name__ == '__main__':
def _test_caller_name() -> None:
assert caller_name() == '_test_caller_name'
_test_caller_name()
Подтверждение: предварительный комментарий 1313e для ответа answer.
Я нашел способ, если вы проходите через классы и хотите, чтобы класс принадлежал методу AND. Это требует немного работы по извлечению, но это делает ее точкой. Это работает в Python 2.7.13.
import inspect, os
class ClassOne:
def method1(self):
classtwoObj.method2()
class ClassTwo:
def method2(self):
curframe = inspect.currentframe()
calframe = inspect.getouterframes(curframe, 4)
print '\nI was called from', calframe[1][3], \
'in', calframe[1][4][0][6: -2]
# create objects to access class methods
classoneObj = ClassOne()
classtwoObj = ClassTwo()
# start the program
os.system('cls')
classoneObj.method1()
#!/usr/bin/env python
import inspect
called=lambda: inspect.stack()[1][3]
def caller1():
print "inside: ",called()
def caller2():
print "inside: ",called()
if __name__=='__main__':
caller1()
caller2()
[email protected]:~/Documents$ python test_func.py
inside: caller1
inside: caller2
[email protected]:~/Documents$
Вы можете распечатать список функций, как ответили здесь fooobar.com/info/4696/...
import inspect;print(*['\n\x1b[0;36;1m| \x1b[0;32;1m{:25}\x1b[0;36;1m| \x1b[0;35;1m{}'.format(str(x.function), x.filename+'\x1b[0;31;1m:'+str(x.lineno)+'\x1b[0m') for x in inspect.stack()])