如何让Pyright/Mypy识别create_a()返回的A实例的b属性非空？

针对你遇到的问题，有几种简洁方式能让类型检查器知道create_a()返回的A实例的b属性一定不为None，不用写大量冗余的if检查：

1. 使用断言（Assert） #

在调用doSomeStuff()前加一行断言，既能在运行时确保b非空，又能让类型检查器认可这个属性的非空状态：

a = c.create_a()
assert a.b is not None
a.b.doSomeStuff()

如果代码运行时a.b意外为None，断言会直接抛出错误，帮你提前发现问题。

2. 用类型转换（cast） #

通过typing.cast直接告诉类型检查器a.b的实际类型是B，跳过空值检查：

from typing import cast

# ...
a = c.create_a()
cast(B, a.b).doSomeStuff()

这种方式没有运行时检查，仅给类型检查器提供提示，适合你确定b一定非空的场景。

3. 定义子类/Protocol约束返回类型 #

如果create_a()返回的A实例总是带有非空的b，可以定义继承自A的子类，明确b的类型为B，再让create_a()返回这个子类：

class AWithB(A):
    b: B  # 不再是Optional[B]

class C:
    # ...
    def create_a(self) -> AWithB:
        a = AWithB()
        a.b = B()
        return a

后续使用create_a()返回的实例时，类型检查器会自动识别b是非空的，无需额外处理。

4. 调整A类的初始化逻辑（如果适用） #

如果A类实例在正常使用时b都不应该为None，直接把b的初始化移到A的__init__方法里，同时修改类型注解：

class A:
    def __init__(self):
        print("Hello")
        self.b = B()  # 直接初始化b
    b: B  # 类型不再是Optional[B]

这种方式从根源上解决了空值问题，不需要额外的类型提示，仅适合A实例必须有b的场景。

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