python获取代码所在行号,输出到终端或日志文件中
文章目录
在编写程序过程中,通常需要获取代码所在的行号,以便进行问题的分析及定位,常用的获取行号的方式有:
-
使用 sys 模块
-
使用inspect模块
-
使用linecache模块
-
使用traceback模块
-
使用enumerate函数
但它们各有特点:
-
使用linecache模块可以很方便地获取指定行的代码内容,但不适用于获取当前行号。
-
使用inspect模块可以获取当前行号以及其他代码信息,但需要在函数或方法中使用。
-
使用traceback模块可以获取当前行号以及其他堆栈信息,但更适合处理异常和追踪代码执行路径。
1、使用 sys 模块
import sys
def PRINT(string):
print('func:{}(),line:{},'.format(sys._getframe().f_code.co_name, sys._getframe().f_lineno), end="")
# /print('调用该函数的上级为{}'.format(sys._getframe(1).f_code.co_name))
print(string)
PRINT("help :%d" % 123654)
2、使用inspect模块
import inspect
def print_line_number():
frame = inspect.currentframe()
lineno = frame.f_back.f_lineno
print(f'Line {lineno}: Hello, World! ')
print(f'Line {lineno}: print("Hello, World!")')
print_line_number()
再比如:
import inspect
frame = inspect.currentframe()
lineno = frame.f_back.f_lineno
print(f'Line {lineno}: print("is ok!")')
print(f'Line {inspect.currentframe().f_back.f_lineno}: print("is ok!")')
print(f'Line {inspect.currentframe().f_lineno}: print("is ok!")')
上述代码中,我们使用inspect.currentframe()函数来获取当前代码的堆栈帧对象。然后,通过frame.f_lineno属性来获取行号。最后,我们使用print()函数来打印行号。
需要注意的是,inspect模块是一个非常强大的工具,除了行号,它还可以获取函数、类、方法等的相关信息,可以用于编写更复杂的调试工具。
3、使用linecache模块
Python的linecache模块提供了一些函数用于缓存和提供源代码的行信息。我们可以使用linecache模块来打印特定行号的代码。
import linecache
def print_line_number():
line = linecache.getline(__file__, 3)
print(f'Line 3: {line.strip()}')
print_line_number()
4、使用traceback模块
Python的内置模块traceback提供了一种更灵活的方法来获取代码的行号。我们可以利用traceback模块来获取当前代码所在的栈帧信息。
以下是一个使用traceback模块的示例代码:
import traceback
def print_line_number():
stack = traceback.extract_stack()
filename, line_number, _, _ = stack[-2]
print(f'Line {line_number}')
print_line_number()
上述代码中,我们使用traceback.extract_stack()函数来获取当前代码的栈帧信息。然后,通过索引-2来获取上一个栈帧的文件名和行号。最后,我们使用print()函数来打印行号。
需要注意的是,traceback模块还提供了其他有用的函数和方法,用于处理异常和追踪堆栈信息。
5、使用enumerate函数
lines = ['print("Hello, World!")', 'x = 1', 'print(x)']
for i, line in enumerate(lines, start=1):
print(f'Line {i}: {line}')
在上面的代码中,enumerate(lines, start=1)表示从1开始对lines中的元素进行迭代,i表示行号,line表示代码行。我们通过f-string将行号和代码行打印出来。
在Python中,可以使用enumerate函数来打印行号。下面是几种不同的实现方法:
方法1:
lines = ["line1", "line2", "line3"]
for i, line in enumerate(lines, start=1):
print(f"Line {i}: {line}")
方法2:
lines = ["line1", "line2", "line3"]
for i in range(len(lines)):
print(f"Line {i+1}: {lines[i]}")
方法3:
with open('file.txt') as f:
for i, line in enumerate(f, start=1):
print(f"Line {i}: {line.strip()}")