Python 如何判断变量可调用函数的参数是否都属于某个子类

发布于03月02日

这可能是一个艰难的.

假设我有一种类型

JSON = Union[Mapping[str, "JSON"], Sequence["JSON"], str, int, float, bool, None]

我有一个功能

def memoize[**P, T: JSON](fn: Callable[P,T]) -> Callable[P,T]:
    # ...make the function memoizable
    return wrapped_fn

如何约束FN的参数都是JSON的子类型？或者，如果这不能静态完成，我如何在创建包装器之前在Mememize中判断它？

我试着给ParamSpec变量 **P设置边界，但似乎还没有实现.我也试过issubclass，但这对typehints不好用.

推荐答案

如果fn有一个任意的签名，目前还没有办法做到这一点.我认为接下来最好的事情是在调用点生成错误(请参见 Pyright playground):

import collections.abc as cx
import typing as t

type JSON = cx.Mapping[str, JSON] | cx.Sequence[JSON] | str | int | float | bool | None

class _JSONOnlyCallable(t.Protocol):
    def __call__(self, /, *args: JSON, **kwargs: JSON) -> JSON: ...

def memoize[F: cx.Callable[..., t.Any]](fn: F, /) -> F | _JSONOnlyCallable:
    return fn

@memoize
def f(a: int, b: str, c: set[int]) -> str: ...

>>> f(1, "", {1, 2})
             ^^^^^^
pyright: Argument of type "set[int]" cannot be assigned to parameter "args" of type "JSON" in function "__call__"