5

传递函数时，我通常使用typing.Callable.

文档collections.abc.Callable声明它有四种 dunder 方法：

类 collections.abc.Callable

分别提供方法的类的ABCs contains ()、hash ()、len ()和call ()。

有一次，我想检查函数是否有__wrapped__属性。这在运行时通过检查可以正常工作hasattr(func, "__wrapped__")。

当使用 进行静态类型检查时mypy，它会报告：error: "Callable[..., Any]" has no attribute "__wrapped__" [attr-defined]. 这对我来说很有意义，因为Callable不应该有一个__wrapped__属性。

如何正确键入Callable带有__wrapped__属性的提示 a？我可以做一些其他类型的提示或解决方法吗？

代码示例

我正在使用mypy==0.782和Python==3.8.2：

from functools import wraps
from typing import Callable


def print_int_arg(arg: int) -> None:
    """Print the integer argument."""
    print(arg)


@wraps(print_int_arg)
def wrap_print_int_arg(arg: int) -> None:
    print_int_arg(arg)
    # do other stuff


def print_is_wrapped(func: Callable) -> None:
    """Print if a function is wrapped."""
    if hasattr(func, "__wrapped__"):
        # error: "Callable[..., Any]" has no attribute "__wrapped__"  [attr-defined]
        print(f"func named {func.__name__} wraps {func.__wrapped__.__name__}.")


print_is_wrapped(wrap_print_int_arg)
4

2 回答 2

1

显然，简单的答案是添加# type: ignore评论。但是，这实际上并不能解决问题，IMO。

我决定为具有__wrapped__属性的可调用对象创建一个类型存根。基于这个答案，这是我目前的解决方案：

from typing import Callable, cast


class WrapsCallable:
    """Stub for a Callable with a __wrapped__ attribute."""

    __wrapped__: Callable

    __name__: str

    def __call__(self, *args, **kwargs):
        ...


def print_is_wrapped(func: Callable) -> None:
    """Print if a function is wrapped."""
    if hasattr(func, "__wrapped__"):
        func = cast(WrapsCallable, func)
        print(f"func named {func.__name__} wraps {func.__wrapped__.__name__}.")

现在mypy报道。Success: no issues found in 1 source file

我觉得好像这是很多样板代码，并且希望得到更精简的答案。

于 2020-07-08T17:45:13.837 回答
1

Mypy 抱怨使用__wrapped__in print 语句。下面的技巧让 mypy 开心

def print_is_wrapped(func: Callable) -> None:
    """Print if a function is wrapped."""
    if hasattr(func, "__wrapped__"):
        wrapped_name = getattr(func, "__wrapped__").__name__
        print(f"func named {func.__name__} wraps {wrapped_name}.")
于 2020-07-13T23:10:11.547 回答