BytesIO.readline无换行符时未返回空字节，不符合预期问题咨询

问题原因 #

io.BytesIO的readline()方法逻辑是：从当前指针位置开始读取，直到遇到换行符（b'\n'）或缓冲区末尾为止。如果缓冲区中没有换行符，它会直接读取从当前指针到缓冲区末尾的所有内容；只有当指针已经处于缓冲区末尾时，调用readline()才会返回b''。

你的代码中写入的是b"foo"（无换行符），执行seek(0)后指针回到缓冲区开头，此时调用readline()会读取到缓冲区末尾的全部内容，所以返回b'foo'而非预期的b''。

解决办法 #

如果需要实现“未找到换行符则返回空字节”的自定义逻辑，可以封装一个函数来实现：

import io

def readline_only_with_newline(buff):
    current_pos = buff.tell()
    # 先检查当前位置到末尾是否存在换行符
    remaining_content = buff.read()
    buff.seek(current_pos)  # 重置指针到原位置
    
    if b'\n' not in remaining_content:
        return b''
    # 存在换行符时正常执行readline
    return buff.readline()

# 测试示例
b = io.BytesIO()
b.write(b"foo")
b.seek(0)
print(readline_only_with_newline(b))  # 输出: b''

b.seek(0)
b.write(b"foo\nbar")
b.seek(0)
print(readline_only_with_newline(b))  # 输出: b'foo\n'

补充说明 #

如果只是需要判断是否已经读取到缓冲区末尾，可以直接对比指针位置和缓冲区总长度：

b = io.BytesIO(b"foo")
b.seek(0)
b.read()  # 读取全部内容后指针到末尾
if b.tell() == b.getbuffer().nbytes:
    print(b.readline())  # 输出: b''

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