如何让mypy正确类型检查使用functools.partial的函数，同时解决运行时参数冲突问题？

我来帮你搞定这个头疼的问题！你遇到的其实是functools.partial在静态类型检查和实际运行行为之间的典型矛盾——当你用关键字参数绑定partial时，mypy没办法自动识别剩余参数的合法传参方式，导致两边各卡一步。咱们先拆解下问题根源，再给出解决方案：

为什么会出现这个矛盾？ #

1. 第一个版本运行报错的原因：
   你用partial(f, j=a)绑定了关键字参数j，但后续调用fun(4,5)时，Python会把4、5按顺序传给f的第一个和第二个位置参数（也就是i和j）。可j已经被partial绑定过了，自然就抛出了“multiple values for argument 'j'”的冲突错误。

2. 第二个版本mypy报错的原因：
   你给g的返回类型标注的是Callable[[int, int], int]，这个类型告诉mypy：“这个函数只能接受两个位置参数，不支持关键字传参”。所以当你用fun(i=4, k=5)传参时，mypy会判定这是非法的关键字参数。

解决方案：用Protocol明确函数的传参规则 #

既然mypy没法自动识别partial绑定关键字后的合法传参方式，咱们就手动给它定义清楚。可以用typing.Protocol来创建一个支持位置传参+关键字传参的函数类型，这样两边都能满意：

from functools import partial
from typing import Protocol

def f(i: int, j: float, k: int) -> int:
    return i + int(j) + k

# 定义一个协议，明确这个函数既支持位置传参i、k，也支持关键字传参
class PartialF(Protocol):
    def __call__(self, i: int, k: int) -> int: ...
    def __call__(self, *, i: int, k: int) -> int: ...

def g(a: float) -> PartialF:
    return partial(f, j=a)

# 两种调用方式都能正常运行，且通过mypy检查
fun = g(3.0)
r1: int = fun(4, 5)
r2: int = fun(i=4, k=5)
print(r1, r2)  # 输出12 12

另一种更灵活的方案（Python 3.10+） #

如果你用的是Python 3.10及以上版本，可以用ParamSpec和TypeVar来更通用地标注剩余参数，适合参数很多的场景：

from collections.abc import Callable
from functools import partial
from typing import ParamSpec, TypeVar

# 定义参数规范和返回值类型变量
P = ParamSpec("P")
R = TypeVar("R")

def f(i: int, j: float, k: int) -> int:
    return i + int(j) + k

def g(a: float) -> Callable[P, R]:
    # 用type: ignore跳过partial的类型推导，因为我们已经手动明确了返回类型
    return partial(f, j=a)  # type: ignore[return-value]

fun = g(3.0)
# 不管是位置还是关键字传参，mypy都能正确识别
r1: int = fun(4, 5)
r2: int = fun(i=4, k=5)

为什么手动嵌套函数能正常工作？ #

你提到的不用partial的版本之所以没问题，是因为嵌套函数ret(i,k)明确告诉了mypy：“我只接受i和k两个参数，位置或关键字都可以”。而partial的类型系统本身对关键字绑定的场景支持不够完善，所以需要我们手动补充类型定义。

这样一来，你既能享受partial带来的简洁性（尤其是参数很多的函数），又能让代码同时通过mypy检查和正常运行啦！

备注：内容来源于stack exchange，提问作者Pierre-olivier Gendraud