如何为click回调包装函数添加正确的类型提示以解决mypy的泛型Callable参数缺失警告？

嘿，这个问题我熟！你遇到的mypy警告是因为Callable是泛型类型，必须指定它的参数和返回值类型，不能直接裸用。你之前用全Any的写法虽然能过检查，但确实太宽松了，完全失去了类型提示的意义，咱们来搞个更严谨的方案。

首先，我们可以用TypeVar来抽象原函数的输入和返回类型，让包装函数的类型和原函数的类型关联起来，这样既满足mypy的要求，又能保留完整的类型检查。

下面是优化后的代码，兼容Python 3.9和3.10：

from typing import Callable, TypeVar, Any
import click

# 定义两个类型变量，分别代表原函数的输入参数类型和返回值类型
T = TypeVar('T')
U = TypeVar('U')

def format_callback(f: Callable[[T], U]) -> Callable[[click.Context, dict[str, Any], T], U]:
    """Function to wrap a function to use as a click callback."""
    return lambda _, __, x: f(x)

为什么这么写？ #

• T和U是类型变量，用来占位原函数f的输入参数类型和返回值类型，这样mypy就能跟踪具体的类型信息。比如如果f是接受int返回str的函数，包装后的回调函数就会被自动推断为接受click.Context、dict[str, Any]和int，返回str的函数。
• 相比全Any的写法，这个方案能让mypy继续帮你检查参数和返回值的类型是否正确，避免潜在的类型错误。
• 这个写法在Python 3.9和3.10下都能正常工作，不需要额外依赖typing_extensions（因为TypeVar在3.9的标准库typing里已经支持）。

举个实际使用的例子，假设你有这样一个原函数：

def process_input(value: str) -> bool:
    return value.strip() != ""

用format_callback包装后，mypy会自动识别包装后的回调函数类型是Callable[[click.Context, dict[str, Any], str], bool]，调用时如果传入错误类型的参数，mypy会立刻给你警告。

备注：内容来源于stack exchange，提问作者BioGeek