From d0df3cd6c245c46bce5a0721b6c629af0019f078 Mon Sep 17 00:00:00 2001
From: Jelle Zijlstra <jelle.zijlstra@gmail.com>
Date: Wed, 2 Nov 2022 21:59:01 -0700
Subject: [PATCH] quopri: improve types

---
 stdlib/quopri.pyi | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/stdlib/quopri.pyi b/stdlib/quopri.pyi
index b8dc0787fd1a..549413226bdb 100644
--- a/stdlib/quopri.pyi
+++ b/stdlib/quopri.pyi
@@ -1,8 +1,11 @@
-from typing import BinaryIO
+from _typeshed import ReadableBuffer, SupportsNoArgReadline, SupportsRead, SupportsWrite
+from typing import Protocol
 
 __all__ = ["encode", "decode", "encodestring", "decodestring"]
 
-def encode(input: BinaryIO, output: BinaryIO, quotetabs: int, header: int = ...) -> None: ...
-def encodestring(s: bytes, quotetabs: int = ..., header: int = ...) -> bytes: ...
-def decode(input: BinaryIO, output: BinaryIO, header: int = ...) -> None: ...
-def decodestring(s: bytes, header: int = ...) -> bytes: ...
+class _Input(SupportsRead[bytes], SupportsNoArgReadline[bytes], Protocol): ...
+
+def encode(input: _Input, output: SupportsWrite[bytes], quotetabs: int, header: int = ...) -> None: ...
+def encodestring(s: ReadableBuffer, quotetabs: int = ..., header: int = ...) -> bytes: ...
+def decode(input: _Input, output: SupportsWrite[bytes], header: int = ...) -> None: ...
+def decodestring(s: str | ReadableBuffer, header: int = ...) -> bytes: ...