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

最新推荐文章于 2025-01-05 14:58:00 发布

原创最新推荐文章于 2025-01-05 14:58:00 发布 · 3.3k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#python

Python 专栏收录该内容

9 篇文章

订阅专栏

本文介绍了如何使用Python的traceback模块来获取当前行号、函数名称和文件名，以替代非标准的sys._getframe()方法。通过traceback模块，我们可以自定义带有行号等信息的print语句，例如在调用print_with_pos作为print时，所有输出都将包含位置信息。

博客http://blog.youkuaiyun.com/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))