如何为boto3.client生成的动态客户端添加正确的类型注解？

boto3的客户端是动态生成的（比如botocore.client.S3并非实际存在的类），仅用BaseClient做注解会丢失服务专属的方法提示与类型校验，以下是两种更精准的方案：

方案一：使用boto3-stubs提供的静态类型定义 #

boto3-stubs是专门为boto3提供类型支持的第三方库，包含所有AWS服务的专属客户端类型定义，能让IDE提供自动补全、参数校验等功能。

1. 安装对应服务的类型包：

pip install boto3-stubs[s3]

2. 导入并使用专属类型注解：

import boto3
from mypy_boto3_s3 import S3Client

client: S3Client = boto3.client('s3')
# 此时调用client.list_buckets()等方法时，IDE会给出正确的参数和返回值提示

方案二：自定义Protocol模拟专属类型 #

如果不想依赖第三方库，可以通过Python的typing.Protocol定义一个包含目标服务常用方法的协议类，以此作为注解类型，实现针对性的类型检查。

示例代码：

import boto3
from typing import Protocol, List, Dict

class S3ClientProtocol(Protocol):
    def list_buckets(self) -> Dict[str, List[Dict]]:
        ...
    def get_object(self, Bucket: str, Key: str) -> Dict[str, any]:
        ...
    # 按需添加你需要用到的其他S3客户端方法

client: S3ClientProtocol = boto3.client('s3')

这种方式无需额外安装依赖，适合仅用到S3客户端部分方法的场景，能在不引入第三方库的前提下实现强类型约束。

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