主页 / 问题 / 1383305
Accepted
ganz
Asked:2022-07-18 16:56:29 +0000 UTC2022-07-18 16:56:29 +0000 UTC

获取函数体中调用函数的行号

  • 772

例如

def f(x):
   line_number=how_get_5_if_call_from_f(2)

f(1)
f(2)
f(3)

即从第 4、5、6 行开始多次调用函数 f（计算函数本身的行号和空行）。必须在函数体中，即在第二行中，为变量 line_number 分配一个值，该值等于调用该函数的行号。即对于 f(1) line_number 应该等于 4 （计算函数本身的代码和一个空字符串）

python

2 个回答

  1. Best Answer

    使用 function traceback.extract_stack()，您可以获得有关调用堆栈的信息，包括从哪一行调用了当前函数：

    import traceback


def f(x):
    print(traceback.extract_stack())  # Полная информация о стеке вызовов
    print(traceback.extract_stack()[-2].lineno)  # Только номер строки, откуда была вызвана функция


f(1)
f(2)
f(3)

    结论：

    [<FrameSummary file /home/user/Projects/test.py, line 9 in <module>>, <FrameSummary file /home/user/Projects/test.py, line 5 in f>]
9
[<FrameSummary file /home/user/Projects/test.py, line 10 in <module>>, <FrameSummary file /home/user/Projects/test.py, line 5 in f>]
10
[<FrameSummary file /home/user/Projects/test.py, line 11 in <module>>, <FrameSummary file /home/user/Projects/test.py, line 5 in f>]
11

    调用堆栈中的最后一个元素对应一个函数调用traceback.extract_stack()，倒数第二个 ( [-2]) 对应一个函数调用f(x)。我们从头开始数，因为 可以有多个调用，堆栈上的第一个元素将对应于最外层的调用（例如，如果首先调用函数，然后main从它调用函数f，那么堆栈的第一个元素将是函数调用main)

    • 6

  2. 您可以计算它被调用了多少次：

    def f(x, count_call):
    if count_call == 2:
       print ('if rows2 --> call in 2 rows')
    print ('call in {0} rows'.format(x))
    return count_call + 1

count_f_call = 0
count_f_call = f(1, count_f_call)
count_f_call = f(2, count_f_call)
count_f_call = f(3, count_f_call)

    或者处理一个异常，告诉你错误在哪一行：

    import traceback

def f(x):
    try:
        x / 0
    except Exception:
        print(traceback.format_exc())

f(1)
f(2)
f(3)
    • 0

相关问题