我通过添加生成器类型的注册扩展了https://docs.python.org/3/library/functools.html#functools.singledispatch的示例
from functools import singledispatch
from typing import Generator
@singledispatch
def fun(arg, verbose):
if verbose:
print("Let me just say,", end=" ")
print(arg)
@fun.register
def _(arg: list, verbose):
if verbose:
print("Enumerate this:")
for i, elem in enumerate(arg):
print(i, elem)
# NEW CODE BELOW
@fun.register
def _(arg: Generator, verbose):
if verbose:
print("Enumerate this:")
for i, elem in enumerate(arg):
print(i, elem)
fun([3,4,5], verbose=True)
fun((i for i in range(6, 10)), verbose=True)
虽然它适用于列表,但它似乎不适用于带有错误的生成器
raise TypeError(
TypeError: Invalid annotation for 'arg'. typing.Generator is not a class.
预计singledispatch
不适用于生成器吗?