如何兼容BinaryIO与RawIOBase进行函数参数类型注解？

你遇到的问题本质是BinaryIO和RawIOBase属于IOBase下的不同分支，无继承关系，因此类型检查器会判定二者不兼容。解决办法是用Protocol定义一个只关注read()方法的结构类型，只要对象实现了符合要求的read()方法，就能被类型注解接受。

具体实现步骤 #

1. 定义自定义Protocol #

从typing（Python 3.8+）或typing_extensions（兼容Python 3.7及更早版本）导入Protocol，声明包含read()方法的协议：

from typing import Protocol, Optional

class ReadableBinaryStream(Protocol):
    def read(self, size: Optional[int] = ...) -> bytes:
        ...

这里的size参数签名和标准IO对象的read()保持一致：可选整数参数，返回bytes类型。

2. 使用自定义Protocol作为函数参数注解 #

在函数中用这个协议替代BinaryIO或RawIOBase：

import sys
import serial

def process_binary_stream(stream: ReadableBinaryStream) -> None:
    data = stream.read(1024)
    # 数据处理逻辑

# 测试两种场景
# 1. 传入serial.Serial对象（RawIOBase子类）
ser = serial.Serial('/dev/ttyUSB0', 9600)
process_binary_stream(ser)  # 类型检查器无报错

# 2. 传入sys.stdin.buffer（BinaryIO对象）
process_binary_stream(sys.stdin.buffer)  # 类型检查器无报错

关键说明 #

• Protocol是结构类型系统的实现，它不要求对象继承自特定类，只要对象拥有协议声明的方法/属性，类型检查器就会判定二者兼容。
• 若需兼容Python 3.7及更早版本，需先安装typing_extensions包：pip install typing_extensions，再从typing_extensions导入Protocol。
• 该方案符合PEP 544（结构化子类型）规范，是Python类型注解中处理鸭子类型场景的标准做法。

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