Pyhon生成器函数中的异常处理

问题描述

我必须创建Python函数生成器，该函数应返回正数的所有除数。 如果没有除数，则函数应返回None。 这样完成：

def divisor(num):
    try:
        final = (x for x in range(1,num + 1) if num / x == int(num / x))
    except StopIteration:
        return None
    else:
        return final

获得这样的结果：

three = divisor(3)
next(three) => 1
next(three) => 3

但是当我再次调用该函数 时，出现 StopIteration 错误：

next(three) =>
Traceback (most recent call last):
 ...
StopIteration

为什么不处理StopIteration？我究竟做错了什么？如何解决？

解决方法

之所以发生，是因为该函数不是生成器。它是一个常规函数，它在第一次调用时从final变量返回生成器。在这种情况下，最好的解决方案是将函数转换为实际的生成器并完全忽略错误处理：

def divisor(num):
    for x in range(1,num + 1):
        if num / x == int(num / x):
            yield x
    while True:
        yield None