Skip to content

Commit 03a7ac5

Browse files
authored
Use @overload for sunau.open (#7033)
1 parent 7bc9c16 commit 03a7ac5

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

stdlib/sunau.pyi

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import sys
22
from _typeshed import Self
3-
from typing import IO, Any, NamedTuple, NoReturn, Union
3+
from typing import IO, Any, NamedTuple, NoReturn, Union, overload
4+
from typing_extensions import Literal
45

56
_File = Union[str, IO[bytes]]
67

@@ -72,7 +73,11 @@ class Au_write:
7273
def writeframes(self, data: bytes) -> None: ...
7374
def close(self) -> None: ...
7475

75-
# Returns a Au_read if mode is rb and Au_write if mode is wb
76+
@overload
77+
def open(f: _File, mode: Literal["r", "rb"]) -> Au_read: ...
78+
@overload
79+
def open(f: _File, mode: Literal["w", "wb"]) -> Au_write: ...
80+
@overload
7681
def open(f: _File, mode: str | None = ...) -> Any: ...
7782

7883
if sys.version_info < (3, 9):

0 commit comments

Comments
 (0)