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語句都帶信息了
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章