Python 生成器和yield：如何知道程序位于哪一行

Question

假设你有一个简单的 Python 生成器，如下所示：

更新：

def f(self):        
    customFunction_1(argList_1)
    yield
    customFunction_2(argList_2)
    yield
    customFunction_3(argList_3)
    yield
    ...

我在另一个脚本中调用 f()，例如：

    h=f()
    while True:
        try:
            h.next()
            sleep(2)
        except KeyboardInterrupt:
                              ##[TODO] tell the last line in f() that was executed

有没有办法可以执行上面的 [TODO] 部分？也就是说知道 f() 中在键盘中断发生之前执行的最后一行？

Answer 1

您可以使用 enumerate() 来计数：（

def f():

    ...
    yield
    ...
    yield
    ... 


for step, value in enumerate(f()):
    try:
        time.sleep(2)
    except KeyboardInterrupt:
        print(step) # step holds the number of the last executed function

因为在您的示例中 yield 不会产生值，value 当然将为 None）

或者非常明确地使用详细指示：

def f():

    ...
    yield 'first function finished'
    ...
    yield 'almost done'
    ... 


for message in f():
    try:
        time.sleep(2)
    except KeyboardInterrupt:
        print(message)

Answer 2

如果您想知道行号以进行调试，那么在 CPython 中您可以使用 h.gi_frame.f_lineno。这是接下来要执行的行，索引为 1。我不确定这是否适用于 CPython 以外的 Python 实现。

h=f()
while True:
    try:
        h.next()
        sleep(2)
    except KeyboardInterrupt:
        print h.gi_frame.f_lineno - 1 # f_lineno is the line to be executed next.

如果您不想出于调试目的了解这一点，那么 Remi 的 enumerate 解决方案 更加简洁。

Answer 3

为什么不从 f() 中产生 i 并使用它？

val = h.next()

Answer 4

def f(self):        
    sleep(10)
    yield
    sleep(10)
    yield
    sleep(10)
    yield


h=f()
while True:
    try:
        h.next()
    except KeyboardInterrupt:
        stack_trace = sys.exc_info()[2]    # first traceback points into this function where the exception was caught
        stack_trace = stack_trace.tb_next  # now we are pointing to where it happened (in this case)
        line_no = stack_trace.tb_lineno    # and here's the line number
        del stack_trace                    # get rid of circular references

我将对 sleep() 的调用移至 f 中，因为只有当异常发生在 f() 内部时，这才有效。