From 16471aef1c3f858626698cd6ca3c469b27df969e Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Tue, 30 Sep 2025 13:47:47 +0000
Subject: [PATCH 1/2] Initial plan


From 0636deab909b2cba4505888f5a9903266f7cdf98 Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Tue, 30 Sep 2025 13:59:55 +0000
Subject: [PATCH 2/2] Add _WriteStream protocol for MicroPython Buffer write
 methods

Co-authored-by: Josverl <981654+Josverl@users.noreply.github.com>
---
 reference/_mpy_shed/__init__.pyi  |  6 ++-
 reference/_mpy_shed/buffer_mp.pyi | 64 ++++++++++++++++++++++++++++++-
 2 files changed, 68 insertions(+), 2 deletions(-)

diff --git a/reference/_mpy_shed/__init__.pyi b/reference/_mpy_shed/__init__.pyi
index 0098f6ac0..d49be7309 100644
--- a/reference/_mpy_shed/__init__.pyi
+++ b/reference/_mpy_shed/__init__.pyi
@@ -27,7 +27,11 @@ from .blockdevice import (
     _OldAbstractBlockDev,
     _OldAbstractReadOnlyBlockDev,
 )
-from .buffer_mp import AnyReadableBuf as AnyReadableBuf, AnyWritableBuf as AnyWritableBuf
+from .buffer_mp import (
+    AnyReadableBuf as AnyReadableBuf,
+    AnyWritableBuf as AnyWritableBuf,
+    _WriteStream as _WriteStream,
+)
 
 from .io_mp import (
     BytesIO as BytesIO,
diff --git a/reference/_mpy_shed/buffer_mp.pyi b/reference/_mpy_shed/buffer_mp.pyi
index a1e3846b2..2dd5dc28c 100644
--- a/reference/_mpy_shed/buffer_mp.pyi
+++ b/reference/_mpy_shed/buffer_mp.pyi
@@ -1,8 +1,70 @@
 from _typeshed import Incomplete, structseq, AnyStr_co
-from typing_extensions import TypeAlias, TypeVar
+from typing_extensions import TypeAlias, TypeVar, Protocol
+from typing import overload
 from array import array
 
 # ------------------------------------------------------------------------------------
 # TODO: need some to allow string to be passed in : uart_1.write("hello")
 AnyReadableBuf: TypeAlias = bytearray | array | memoryview | bytes | Incomplete
 AnyWritableBuf: TypeAlias = bytearray | array | memoryview | Incomplete
+
+
+# ------------------------------------------------------------------------------------
+# MicroPython Buffer Protocol
+# See: https://github.com/micropython/micropython/pull/17938#discussion_r2329804709
+# 
+# This protocol defines the extended write method that supports:
+# - write(buf) - basic write
+# - write(buf, max_len) - write with maximum length
+# - write(buf, off, max_len) - write with offset and maximum length
+
+
+class _WriteStream(Protocol):
+    """
+    Protocol for MicroPython streams that support the extended write signature
+    with optional offset and max_len parameters.
+    
+    This is commonly used in MicroPython's IO streams, sockets, and serial interfaces.
+    """
+
+    @overload
+    def write(self, buf: AnyReadableBuf, /) -> int | None:
+        """
+        Write the buffer of bytes to the stream.
+        
+        Args:
+            buf: Buffer to write
+            
+        Returns:
+            Number of bytes written, or None if non-blocking and would block
+        """
+        ...
+
+    @overload
+    def write(self, buf: AnyReadableBuf, max_len: int, /) -> int | None:
+        """
+        Write up to max_len bytes from the buffer to the stream.
+        
+        Args:
+            buf: Buffer to write
+            max_len: Maximum number of bytes to write
+            
+        Returns:
+            Number of bytes written, or None if non-blocking and would block
+        """
+        ...
+
+    @overload
+    def write(self, buf: AnyReadableBuf, off: int, max_len: int, /) -> int | None:
+        """
+        Write up to max_len bytes from the buffer starting at offset off to the stream.
+        
+        Args:
+            buf: Buffer to write
+            off: Offset into the buffer to start writing from
+            max_len: Maximum number of bytes to write
+            
+        Returns:
+            Number of bytes written, or None if non-blocking and would block
+        """
+        ...