在代码中打印方法的当前调用堆栈

在Python中,如何从方法中打印当前调用堆栈(用于调试目的)。

207849 次浏览

下面是一个通过回溯模块获取堆栈并打印它的例子:

import traceback


def f():
g()


def g():
for line in traceback.format_stack():
print(line.strip())


f()


# Prints:
# File "so-stack.py", line 10, in <module>
#     f()
# File "so-stack.py", line 4, in f
#     g()
# File "so-stack.py", line 7, in g
#     for line in traceback.format_stack():

如果你真的只想打印堆栈到stderr,你可以使用:

traceback.print_stack()

或者打印到stdout(如果想保持重定向输出在一起很有用),使用:

traceback.print_stack(file=sys.stdout)

但是通过traceback.format_stack()获取它可以让你对它做任何你喜欢的事情。

import traceback
traceback.print_stack()

inspect.stack()返回当前堆栈,而不是异常回溯:

import inspect
print inspect.stack()

有关log_stack实用函数,请参阅https://gist.github.com/FredLoney/5454553

如果您使用python调试器,不仅可以交互式探测变量,还可以使用“where”命令或“w”获取调用堆栈。

在程序的顶部

import pdb

然后在代码中查看发生了什么

pdb.set_trace()

然后你就进入了提示符

下面是@RichieHindle的精彩答案的一个变体,它实现了一个可以选择性地应用于所需函数的装饰器。适用于Python 2.7.14和3.6.4。

from __future__ import print_function
import functools
import traceback
import sys


INDENT = 4*' '


def stacktrace(func):
@functools.wraps(func)
def wrapped(*args, **kwds):
# Get all but last line returned by traceback.format_stack()
# which is the line below.
callstack = '\n'.join([INDENT+line.strip() for line in traceback.format_stack()][:-1])
print('{}() called:'.format(func.__name__))
print(callstack)
return func(*args, **kwds)


return wrapped


@stacktrace
def test_func():
return 42


print(test_func())

样本输出:

test_func() called:
File "stacktrace_decorator.py", line 28, in <module>
print(test_func())
42

安装检查它

pip3 install inspect-it --user

代码

import inspect;print(*['{:40}| {}:{}\n'.format(x.function, x.filename, x.lineno) for x in inspect.stack()])

你可以对这一行做一个代码片段

它会显示一个包含文件名和行号的函数调用栈列表

列出从开始到放这一行的位置

对于那些在使用pdb时需要打印调用堆栈的人,就这样做吧

(Pdb) where