Python生成器中StopIteration无法被main()捕获的修复咨询

这个问题的根源是PEP 479的规定：从Python 3.7开始，生成器内部手动抛出StopIteration异常会被自动转换为RuntimeError——因为StopIteration是Python用来标记迭代器自然结束的信号，不应该被手动触发。

要解决这个问题，我们需要调整逻辑，让生成器在达到最大ID时自然终止，从而让迭代器正常抛出StopIteration，而不是在check_id_valid里手动抛出它。以下是修复后的完整代码：

class illigalException(Exception):
    def __init__(self, id_number):
        super().__init__(f"Invalid ID: {id_number} has less than 9 digits")

def check_id_valid(id_number):
    id_str = str(id_number)
    if len(id_str) < 9:
        raise illigalException(id_number)
    # 不再抛出StopIteration，改为返回布尔值标记ID是否合规
    return int(id_number) <= 999999999

def id_generator(Id_number):
    while True:
        Id_number += 1
        try:
            # 检查ID长度和最大值限制
            if not check_id_valid(Id_number):
                # 超过最大值时终止生成器，自然触发StopIteration
                return
            # 这里可以添加其他ID有效性校验逻辑（比如校验码规则）
            yield Id_number
        except illigalException:
            # 跳过长度不足的无效ID，继续生成下一个
            continue

def main():
    id_gen = id_generator(1234567800)
    try:
        for item in range(10):
            print(next(id_gen))
    except StopIteration:
        print("Reached max number, all subsequent IDs are over 9 digits")
    except illigalException as e:
        print(e)

if __name__ == "__main__":
    main()

关键修改说明： #

• 调整check_id_valid的逻辑：
  移除了手动抛出StopIteration的代码，改为返回布尔值表示ID是否在合法范围内；长度不足的情况依然保留自定义异常抛出。
• 生成器的终止方式：
  当检测到ID超过最大值时，用return终止生成器——生成器执行到return时，会自动在迭代过程中抛出StopIteration，这个异常可以被main中的try-except正常捕获。
• 异常处理优化：
  在生成器内部捕获长度不足的异常，跳过无效ID继续迭代，避免中断整个生成流程。

如果坚持要在check_id_valid中使用内置异常标识超限，可以改用ValueError，再在生成器中捕获后终止迭代，同样能达到预期效果：

def check_id_valid(id_number):
    id_str = str(id_number)
    if len(id_str) < 9:
        raise illigalException(id_number)
    if int(id_number) > 999999999:
        raise ValueError(f"ID {id_number} exceeds 9 digits")

def id_generator(Id_number):
    while True:
        Id_number += 1
        try:
            check_id_valid(Id_number)
            yield Id_number
        except illigalException:
            continue
        except ValueError:
            # 捕获超限异常后终止生成器
            return

这样修改后，无论是直接调用check_id_valid、通过IDIterator迭代器，还是通过生成器id_generator，异常都能被正确处理，不会触发RuntimeError。

内容的提问来源于stack exchange，提问作者user13279074