生成器调试

调试

def creat_num(all_num):
    print("----------1-----------")
    a,b = 0,1
    current_num = 0
    while current_num < all_num:
        print("----------2-----------")
        yield a  # 如果一个函数中有yield语句，那么这个就不再是函数，而是一个生成器的模板
        print("----------3-----------")
        a, b = b, a+b
        current_num += 1
        print("----------4-----------")
    return "ok......"

# 如果在调用creat_num的时候，发现这个函数中有yield，那么此时，不是调用函数，而是创建一个生成器对象
obj = creat_num(4)

ret = next(obj)
print(ret)

ret = next(obj)
print(ret)

ret = next(obj)
print(ret)

print("------循环------")
# 如果想接收return返回的值"ok......",则需要用try
while True:
    try:
        ret = next(obj)
        print(ret)
    except Exception as ret:
        # 需要用ret.value来接收return的值，在循环之后才能打印出来
        print(ret.value)
        break

结果：

----------1-----------
----------2-----------
0
----------3-----------
----------4-----------
----------2-----------
1
----------3-----------
----------4-----------
----------2-----------
1
------循环------
----------3-----------
----------4-----------
----------2-----------
2
----------3-----------
----------4-----------
ok......