Python自定义容器切片返回类型的正确类型提示实现方案问询

这是Python类型提示中非常典型的重载场景，不用每次手动标注类型，用typing.overload就能完美解决你的问题。核心思路是给__getitem__方法定义多套类型签名，让类型检查器能根据输入参数的类型（索引/切片）自动推断返回值类型。

正确实现方案 #

我们通过@overload装饰器为__getitem__分别定义索引和切片场景的类型规则，实际运行逻辑保持不变：

from typing import Union, List, overload

class CustomContainer:
    def __init__(self, values: List[int]):
        self.values = values

    # 重载1：接收int类型索引，返回int
    @overload
    def __getitem__(self, item: int) -> int:
        ...

    # 重载2：接收slice类型切片，返回CustomContainer实例
    @overload
    def __getitem__(self, item: slice) -> 'CustomContainer':
        ...

    # 统一实现逻辑，类型提示兼容所有重载场景
    def __getitem__(self, item: Union[int, slice]) -> Union[int, 'CustomContainer']:
        if isinstance(item, slice):
            return CustomContainer(self.values[item])
        return self.values[item]

效果验证 #

现在使用mypy、PyCharm等类型检查工具时，会自动识别正确的类型，完全不需要手动强制标注：

a = CustomContainer([1, 2])
b = a[0]  # 类型检查器自动识别为int
c = a[:]  # 类型检查器自动识别为CustomContainer

方案优势说明 #

• 类型精确性：@overload告诉类型检查器，不同输入参数对应不同返回类型，彻底解决联合类型的模糊问题。
• 可扩展性：如果以后容器需要支持其他索引类型（比如tuple用于多维索引），只需新增一套@overload签名即可。
• 运行时无影响：实际的__getitem__实现还是一个统一方法，不会改变代码的运行逻辑。

注意事项 #

1. 所有重载的签名必须覆盖所有可能的输入场景，否则类型检查器可能抛出警告。
2. 在类内部引用自身类型时，用字符串'CustomContainer'是Python 3.7+的标准写法，也可以通过from __future__ import annotations开启延迟注解来直接写类名。

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