Python中使用@singledispatch时Type注解类型参数报错的问题求助

问题原因 #

1. 泛型注解不被支持：Python 3.8的singledispatch无法识别Type[bool]这类泛型类型提示，它要求注册的必须是实际的类对象（如bool、int），而Type[bool]只是用于类型提示的泛型，并非一个类，因此抛出Invalid annotation错误。
2. 用法逻辑误解：你希望传入类对象（如bool）时触发对应处理函数，但singledispatch是根据参数的类型来分发调用的——所有类对象的类型都是type，所以即使你注册了bool，也只会在参数是布尔值（如True）时触发，而非传入bool类时。

正确实现方式 #

方案一：字典映射（最直接）

放弃singledispatch，用字典存储类与生成函数的映射，直接匹配传入的类对象：

from datetime import datetime
from random import choice, randint


def load_value(cls):
    generators = {
        bool: lambda: choice([False, True]),
        int: lambda: randint(0, 1000),
        datetime: datetime.now
    }
    if gen := generators.get(cls):
        return gen()
    raise NotImplementedError(f'Unsupported type: {cls}')


print(load_value(bool))
print(load_value(int))
print(load_value(datetime))

方案二：基于type的singledispatch处理

如果一定要用singledispatch，可以注册type类型的处理函数，然后在函数内部判断传入的具体类：

from datetime import datetime
from functools import singledispatch
from random import choice, randint
from typing import NoReturn


@singledispatch
def load_value(value) -> NoReturn:
    raise NotImplementedError(f'Unsupported type: {value}')


@load_value.register(type)
def _(cls: type):
    if cls is bool:
        return choice([False, True])
    elif cls is int:
        return randint(0, 1000)
    elif cls is datetime:
        return datetime.now()
    raise NotImplementedError(f'Unsupported type: {cls}')


print(load_value(bool))
print(load_value(int))
print(load_value(datetime))

原代码的核心问题

原代码中用Type[bool]作为参数注解，singledispatch会尝试将这个注解作为要注册的类型，但Type[bool]不是一个类对象，只是类型提示的泛型，因此报错。即使去掉Type[...]改成value: bool，也只会在传入布尔值时触发，而不是传入bool类时，这不符合你的需求。

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