如何实现Python动态继承类型的Try类并支持自动补全？

需求回顾 #

• 接收一个或多个Optional[T]参数，取第一个非None值进行包装
• 实例可链式调用包装类型的所有属性和方法，返回新的Try实例
• 最终通过.into属性获取原始值（T | None）
• 支持IDE自动补全与语法高亮

修正后的实现 #

from typing import Any, Generic, Optional, TypeVar, Iterable

T = TypeVar('T')
U = TypeVar('U')

class Try(Generic[T]):
    def __init__(self, *options: Optional[T]) -> None:
        # 取第一个非None的参数
        self.into: Optional[T] = None
        for opt in options:
            if opt is not None:
                self.into = opt
                break
        self._wrapped_type = type(self.into) if self.into is not None else None

    def __getattr__(self, name: str) -> 'Try[U]':
        # 当访问不存在的属性时，尝试从包装对象获取
        if self.into is None:
            return Try(None)
        attr = getattr(self.into, name, None)
        return Try(attr)
    
    def __call__(self, *args: Any, **kwargs: Any) -> 'Try[U]':
        # 处理包装对象是可调用的情况
        if self.into is None or not callable(self.into):
            return Try(None)
        try:
            result = self.into(*args, **kwargs)
            return Try(result)
        except Exception:
            # 可选：如果调用出错返回None，可根据需求调整逻辑
            return Try(None)
    
    def __dir__(self) -> Iterable[str]:
        # 合并自身属性和包装类型的属性，支持自动补全
        self_attrs = set(super().__dir__())
        wrapped_attrs = set(dir(self._wrapped_type)) if self._wrapped_type else set()
        return sorted(self_attrs.union(wrapped_attrs))
    
    # 可选：让实例表现更接近包装类型
    def __str__(self) -> str:
        return str(self.into)
    
    def __repr__(self) -> str:
        return f"Try({repr(self.into)})"

关键改进点 #

1. 多参数支持：构造函数接收*options，遍历找到第一个非None值，满足"接收一个或多个Option[T]参数"的需求。
2. 类型提示优化：使用TypeVar[U]标注__getattr__和__call__的返回类型，让IDE能推断链式调用后的Try包装类型，提升自动补全准确性。
3. 自动补全修复：__dir__方法合并了Try自身属性（如into）和包装类型的所有属性/方法，IDE会显示完整的可调用列表。
4. 容错处理：__call__中添加异常捕获（可选），避免方法调用出错导致程序崩溃，可根据需求移除或调整。
5. 友好的字符串表示：重写__str__和__repr__，让Try实例的打印结果更直观。

使用示例 #

import random

# 基础使用
t = Try('my string' if random.random() > 0.5 else None)
print(t.into)  # 输出'my string'或None

# 链式调用
t = Try('my string   ' if random.random() > 0.5 else None).rstrip().upper()
print(t.into)  # 输出'MY STRING'或None

# 多参数取第一个非None
t = Try(None, 123, 'hello')
print(t.into)  # 输出123

关于自动补全的说明 #

大部分现代IDE（如PyCharm、VS Code + Pylance）会识别__dir__返回的属性列表，结合类型提示提供自动补全。如果仍有问题，可以：

• 确保IDE开启类型检查功能
• 使用typing.cast在需要的地方显式标注类型
• 复杂场景可考虑用Protocol定义包装类型接口，但上述实现已满足常规需求。

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