python 获取 当前行号,函数名称, 文件名

博客http://blog.csdn.net/ksearch/article/details/23131269
中介绍了使用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)

输出为:
[quote][('/***/ScriptInfo.py', 59, '<module>', 'test_traceback()'), ('/***/ScriptInfo.py', 56, 'test_traceback', 'info = traceback.extract_stack(limit=2)')][/quote]

有了以上的知识, 就可以自定义带行号等信息的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))


特殊的使用方法
[i]from */ScriptInfo.py import print_with_pos as print[/i]
则,所有的print语句都带信息了
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章