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)

输出为:
引用
[('/***/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语句都带信息了

猜你喜欢

转载自cherishlc.iteye.com/blog/2357245
今日推荐