Python3中类型检查获取子类属性时如何避免警告？

你的代码运行逻辑没问题，但PyCharm的静态类型分析无法识别type(obj) is SubA这类判断后的类型收缩，因此抛出警告。可以用以下几种方式解决：

1. 局部变量类型标注 #

在类型判断后，将obj赋值给标注了子类类型的局部变量，让PyCharm识别具体类型：

def func(obj: Base):
    if type(obj) is SubA:
        sub_a: SubA = obj
        sub_a.a  # 此处无警告
    elif type(obj) is SubB:
        sub_b: SubB = obj
        sub_b.b1  # 此处无警告
        sub_b.b2  # 此处无警告
    ...

2. 使用typing.cast强制类型转换 #

从typing模块导入cast，显式告诉静态分析工具变量的实际类型：

from typing import cast

def func(obj: Base):
    if type(obj) is SubA:
        obj = cast(SubA, obj)
        obj.a  # 此处无警告
    elif type(obj) is SubB:
        obj = cast(SubB, obj)
        obj.b1  # 此处无警告
        obj.b2  # 此处无警告
    ...

3. 重构为多态实现（推荐） #

这是更符合面向对象设计的方案，通过在基类定义抽象方法，子类各自实现，避免类型判断：

from abc import ABC, abstractmethod

class Base(ABC):
    @abstractmethod
    def do_something(self):
        pass

class SubA(Base):
    def __init__(self):
        self.a = ...
    
    def do_something(self):
        # 处理self.a的逻辑
        pass

class SubB(Base):
    def __init__(self):
        self.b1 = ...
        self.b2 = ...
    
    def do_something(self):
        # 处理self.b1和self.b2的逻辑
        pass

def func(obj: Base):
    obj.do_something()  # 无需类型判断，无警告

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