Python类型注解下重载场景的替代实现方案咨询

方案1：调整Protocol签名，兼容所有参数选项 #

修改User协议的login方法，将所有可能的参数定义为可选关键字参数，让子类仅关注自身需要的参数，同时保持签名兼容性：

from typing import Protocol, runtime_checkable, Optional

@runtime_checkable
class User(Protocol):
    def login(self, *, user: Optional[str] = None, password: Optional[str] = None, token: Optional[str] = None) -> None:
        ...

class UserUP(User):
    def login(self, *, user: str, password: str, token: Optional[str] = None) -> None:
        print(user)
        print(password)

class UserToken(User):
    def login(self, *, token: str, user: Optional[str] = None, password: Optional[str] = None) -> None:
        print(token)

def login_user(user: User) -> None:
    user.login(user="user", password="password", token="Token")

login_user(UserUP())
login_user(UserToken())

子类仅对自身需要的参数做必填约束，保留父类所有参数定义，mypy会认可该签名兼容性。

方案2：子类使用**kwargs并内部校验参数 #

让子类严格遵循父类的**kwargs签名，在方法内部提取并校验所需参数，同时用TypedDict增强类型提示：

from typing import Protocol, runtime_checkable, TypedDict

@runtime_checkable
class User(Protocol):
    def login(self, **kwargs) -> None:
        ...

class UPCredentials(TypedDict):
    user: str
    password: str

class TokenCredentials(TypedDict):
    token: str

class UserUP(User):
    def login(self, **kwargs) -> None:
        credentials: UPCredentials = kwargs
        print(credentials["user"])
        print(credentials["password"])

class UserToken(User):
    def login(self, **kwargs) -> None:
        credentials: TokenCredentials = kwargs
        print(credentials["token"])

def login_user(user: User) -> None:
    user.login(user="user", password="password", token="Token")

login_user(UserUP())
login_user(UserToken())

该方式保持协议灵活性，同时通过TypedDict明确参数类型。

方案3：拆分协议+函数重载 #

为不同登录类型定义独立协议，通过函数重载让login_user根据类型传递对应参数，避免冗余参数：

from typing import Protocol, runtime_checkable, overload

@runtime_checkable
class UserUPProtocol(Protocol):
    def login(self, user: str, password: str) -> None:
        ...

@runtime_checkable
class UserTokenProtocol(Protocol):
    def login(self, token: str) -> None:
        ...

User = UserUPProtocol | UserTokenProtocol

class UserUP:
    def login(self, user: str, password: str) -> None:
        print(user)
        print(password)

class UserToken:
    def login(self, token: str) -> None:
        print(token)

@overload
def login_user(user: UserUPProtocol) -> None:
    ...

@overload
def login_user(user: UserTokenProtocol) -> None:
    ...

def login_user(user: User) -> None:
    if isinstance(user, UserUP):
        user.login(user="user", password="password")
    elif isinstance(user, UserToken):
        user.login(token="Token")
    else:
        raise ValueError("不支持的用户登录类型")

login_user(UserUP())
login_user(UserToken())

该方式让每个登录类职责更清晰，类型检查更严格，避免不必要的参数传递。

方案4：数据类封装登录参数 #

定义统一的登录参数数据类，所有子类的login方法接收该数据类实例，内部按需提取数据：

from typing import Protocol, runtime_checkable, Optional
from dataclasses import dataclass

@dataclass
class LoginParams:
    user: Optional[str] = None
    password: Optional[str] = None
    token: Optional[str] = None

@runtime_checkable
class User(Protocol):
    def login(self, params: LoginParams) -> None:
        ...

class UserUP(User):
    def login(self, params: LoginParams) -> None:
        assert params.user is not None and params.password is not None
        print(params.user)
        print(params.password)

class UserToken(User):
    def login(self, params: LoginParams) -> None:
        assert params.token is not None
        print(params.token)

def login_user(user: User) -> None:
    params = LoginParams(user="user", password="password", token="Token")
    user.login(params)

login_user(UserUP())
login_user(UserToken())

通过数据类统一参数结构，子类内部断言提取所需参数，既保证签名兼容性，又明确参数类型。

内容的提问来源于stack exchange，提问作者Tlaloc-ES