中介绍了使用sys._getframe()方法获取行号等信息的方法,
但是知乎上有人说这些方法不在python标准内,
这里采用traceback模块实现
我们平时所见的stack trace就是通过traceback模块实现的,比如:
def test_traceback(): import traceback traceback.print_stack(limit=2) # 打印函数调用堆栈 traceback.print_exc() # 无异常时为None info = traceback.extract_stack(limit=2) #返回调用堆栈(一个list),最多两层, 其中最后一个元素为当前行的信息,包括(文件名,行号,函数名,这一行的代码) print(info)
输出为:
引用
[('/***/ScriptInfo.py', 59, '<module>', 'test_traceback()'), ('/***/ScriptInfo.py', 56, 'test_traceback', 'info = traceback.extract_stack(limit=2)')]
有了以上的知识, 就可以自定义带行号等信息的print 语句了:
from __future__ import absolute_import from __future__ import division from __future__ import print_function import traceback def _stack_tuple_to_function_line_filename(stackTuple): ''' stackTuple: (文件名,行号,函数名,这一行的代码) ''' filename = stackTuple[0] linenumber = stackTuple[1] funcname = stackTuple[2] # codeline=stackTuple[3] return 'func: %s, \t\tline: %d, \tfile: %s' % (funcname, linenumber, filename); def get_filename_function_line(limit=1): # limit = 1 表示抽取该函数调用者的位置,注意输入到extract_stack中的limit=limit+1 stackTuple = traceback.extract_stack(limit=limit + 1)[0] return _stack_tuple_to_function_line_filename(stackTuple) def print_with_pos(*args, **kwargs): #输出语句,同print,只是之后会再输出位置信息 print(*args, **kwargs) print('\t%s' % get_filename_function_line(limit=2))
特殊的使用方法
from */ScriptInfo.py import print_with_pos as print
则,所有的print语句都带信息了