Use Literal type in io and _compression (#6917)

python · Jan 14, 2022 · 74d043b · 74d043b
1 parent f9c650d
commit 74d043b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/stdlib/_compression.pyi b/stdlib/_compression.pyi
@@ -1,8 +1,8 @@
 from _typeshed import WriteableBuffer
-from io import BufferedIOBase, RawIOBase
+from io import DEFAULT_BUFFER_SIZE, BufferedIOBase, RawIOBase
 from typing import Any, Callable, Protocol
 
-BUFFER_SIZE: Any
+BUFFER_SIZE = DEFAULT_BUFFER_SIZE
 
 class _Reader(Protocol):
     def read(self, __n: int) -> bytes: ...

diff --git a/stdlib/io.pyi b/stdlib/io.pyi
@@ -5,12 +5,13 @@ from _typeshed import ReadableBuffer, Self, StrOrBytesPath, WriteableBuffer
 from os import _Opener
 from types import TracebackType
 from typing import IO, Any, BinaryIO, Callable, Iterable, Iterator, TextIO
+from typing_extensions import Literal
 
-DEFAULT_BUFFER_SIZE: int
+DEFAULT_BUFFER_SIZE: Literal[8192]
 
-SEEK_SET: int
-SEEK_CUR: int
-SEEK_END: int
+SEEK_SET: Literal[0]
+SEEK_CUR: Literal[1]
+SEEK_END: Literal[2]
 
 open = builtins.open