如何调整装饰器解决mypy的arg-type类型不兼容错误？

问题根源 #

你当前的装饰器使用了宽泛的Any类型注解，mypy无法识别装饰器内部的非None检查逻辑，因此仍然认为传入Optional[Dict]不符合函数要求的Dict类型，从而抛出类型不兼容错误。

解决方案 #

通过给装饰器添加精确的类型注解（利用ParamSpec和TypeVar），并配合类型断言cast，让mypy认可装饰器能将Optional[Dict]转换为合法的Dict类型。

步骤1：导入必要的类型工具 #

from typing import Callable, Optional, Dict, ParamSpec, TypeVar, cast

步骤2：定义参数签名和返回值的类型变量 #

# 捕获原函数的参数签名
P = ParamSpec("P")
# 捕获原函数的返回值类型
R = TypeVar("R")

步骤3：重写带精确类型注解的装饰器 #

def ensure_config_is_dict(func: Callable[P, R]) -> Callable[P, R]:
    def wrapper(*args: P.args, **kwargs: P.kwargs) -> R:
        # 严格检查第一个参数是dict类型（和手动检查逻辑一致）
        if not isinstance(args[0], dict):
            raise Exception("config needs to be a dict")
        # 用cast告诉mypy：经过检查，该参数已确定为Dict类型
        validated_args = (cast(Dict, args[0]),) + args[1:]
        return func(*validated_args, **kwargs)
    return wrapper

步骤4：使用装饰器修饰目标函数 #

@ensure_config_is_dict
def method(config: Dict):
    """docstring"""
    print(type(config))
    print(config)

步骤5：调用函数（mypy不再报错） #

class Stuff:
    config: Optional[Dict] = None

    def __init__(self):
        pass

stuff = Stuff()
method(stuff.config)

原理说明 #

• ParamSpec[P]和TypeVar[R]让装饰器完整保留原函数的参数和返回值类型信息，避免类型丢失。
• cast(Dict, args[0])向mypy明确声明：运行时检查已确保该参数是Dict类型，消除类型不兼容的告警。
• 装饰器内部的检查逻辑和手动检查完全一致，保证运行时的安全性，同时满足静态类型检查的要求。

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