python如何获取文件的行号_python 获取 当前行号，函数名称， 文件名

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, '', '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语句都带信息了

分享到：

2017-02-15 15:30

浏览 2752

评论