From beba8b19014a644dcab75609a3696ef18d648c02 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ho=C3=ABl=20Bagard?=
 <34478245+hoel-bagard@users.noreply.github.com>
Date: Sat, 9 Mar 2024 23:52:25 +0900
Subject: [PATCH] Add `wurlitzer` stubs (#11459)

Co-authored-by: Alex Waygood <Alex.Waygood@Gmail.com>
Co-authored-by: Akuli <akuviljanen17@gmail.com>
---
 pyrightconfig.stricter.json   |  1 +
 stubs/wurlitzer/METADATA.toml |  2 +
 stubs/wurlitzer/wurlitzer.pyi | 98 +++++++++++++++++++++++++++++++++++
 3 files changed, 101 insertions(+)
 create mode 100644 stubs/wurlitzer/METADATA.toml
 create mode 100644 stubs/wurlitzer/wurlitzer.pyi

diff --git a/pyrightconfig.stricter.json b/pyrightconfig.stricter.json
index 958d489e9c89..526fdf21ab08 100644
--- a/pyrightconfig.stricter.json
+++ b/pyrightconfig.stricter.json
@@ -79,6 +79,7 @@
         "stubs/ttkthemes",
         "stubs/vobject",
         "stubs/workalendar",
+        "stubs/wurlitzer",
     ],
     "typeCheckingMode": "strict",
     // TODO: Complete incomplete stubs
diff --git a/stubs/wurlitzer/METADATA.toml b/stubs/wurlitzer/METADATA.toml
new file mode 100644
index 000000000000..7e633c7caf21
--- /dev/null
+++ b/stubs/wurlitzer/METADATA.toml
@@ -0,0 +1,2 @@
+version = "3.0.*"
+upstream_repository = "https://github.com/minrk/wurlitzer"
diff --git a/stubs/wurlitzer/wurlitzer.pyi b/stubs/wurlitzer/wurlitzer.pyi
new file mode 100644
index 000000000000..f2b1c7fa5a01
--- /dev/null
+++ b/stubs/wurlitzer/wurlitzer.pyi
@@ -0,0 +1,98 @@
+__all__ = ["STDOUT", "PIPE", "Wurlitzer", "pipes", "sys_pipes", "sys_pipes_forever", "stop_sys_pipes"]
+
+import contextlib
+import io
+from _typeshed import SupportsWrite
+from contextlib import _GeneratorContextManager
+from types import TracebackType
+from typing import Any, Final, Literal, Protocol, TextIO, TypeVar, overload
+from typing_extensions import TypeAlias
+
+STDOUT: Final = 2
+PIPE: Final = 3
+_STDOUT: TypeAlias = Literal[2]
+_PIPE: TypeAlias = Literal[3]
+_T_contra = TypeVar("_T_contra", contravariant=True)
+_StreamOutT = TypeVar("_StreamOutT", bound=_Stream[str] | _Stream[bytes])
+_StreamErrT = TypeVar("_StreamErrT", bound=_Stream[str] | _Stream[bytes])
+
+class _Stream(SupportsWrite[_T_contra], Protocol):
+    def seek(self, __offset: int, __whence: int = ...) -> int: ...
+
+# Alias for IPython.core.interactiveshell.InteractiveShell.
+# N.B. Even if we added ipython to the stub-uploader allowlist,
+# we wouldn't be able to declare a dependency on ipython here,
+# since `wurlitzer` does not declare a dependency on `ipython` at runtime
+_InteractiveShell: TypeAlias = Any
+
+class Wurlitzer:
+    flush_interval: float
+    encoding: str | None
+
+    def __init__(
+        self,
+        stdout: SupportsWrite[str] | SupportsWrite[bytes] | None = None,
+        stderr: _STDOUT | SupportsWrite[str] | SupportsWrite[bytes] | None = None,
+        encoding: str | None = ...,
+        bufsize: int | None = ...,
+    ) -> None: ...
+    def __enter__(self): ...
+    def __exit__(
+        self, exc_type: type[BaseException] | None, exc_value: BaseException | None, traceback: TracebackType | None
+    ) -> None: ...
+
+def dup2(a: int, b: int, timeout: float = 3) -> int: ...
+def sys_pipes(encoding: str = ..., bufsize: int | None = None) -> contextlib._GeneratorContextManager[tuple[TextIO, TextIO]]: ...
+
+# stubtest does not support overloaded context managers, hence the _GeneratorContextManager[Foo] return types.
+@overload
+def pipes(
+    stdout: _PIPE, stderr: _STDOUT, encoding: None, bufsize: int | None = None
+) -> _GeneratorContextManager[tuple[io.BytesIO, None]]: ...
+@overload
+def pipes(
+    stdout: _PIPE, stderr: _PIPE, encoding: None, bufsize: int | None = None
+) -> _GeneratorContextManager[tuple[io.BytesIO, io.BytesIO]]: ...
+@overload
+def pipes(
+    stdout: _PIPE, stderr: _StreamErrT, encoding: None, bufsize: int | None = None
+) -> _GeneratorContextManager[tuple[io.BytesIO, _StreamErrT]]: ...
+@overload
+def pipes(
+    stdout: _PIPE, stderr: _STDOUT, encoding: str = ..., bufsize: int | None = None
+) -> _GeneratorContextManager[tuple[io.StringIO, None]]: ...
+@overload
+def pipes(
+    stdout: _PIPE, stderr: _PIPE, encoding: str = ..., bufsize: int | None = None
+) -> _GeneratorContextManager[tuple[io.StringIO, io.StringIO]]: ...
+@overload
+def pipes(
+    stdout: _PIPE, stderr: _StreamErrT, encoding: str = ..., bufsize: int | None = None
+) -> _GeneratorContextManager[tuple[io.StringIO, _StreamErrT]]: ...
+@overload
+def pipes(
+    stdout: _StreamOutT, stderr: _STDOUT, encoding: str | None = ..., bufsize: int | None = None
+) -> _GeneratorContextManager[tuple[_StreamOutT, None]]: ...
+@overload
+def pipes(
+    stdout: _StreamOutT, stderr: _PIPE, encoding: None, bufsize: int | None = None
+) -> _GeneratorContextManager[tuple[_StreamOutT, io.BytesIO]]: ...
+@overload
+def pipes(
+    stdout: _StreamOutT, stderr: _PIPE, encoding: str = ..., bufsize: int | None = None
+) -> _GeneratorContextManager[tuple[_StreamOutT, io.StringIO]]: ...
+@overload
+def pipes(
+    stdout: _StreamOutT, stderr: _StreamErrT, encoding: str | None = ..., bufsize: int | None = None
+) -> _GeneratorContextManager[tuple[_StreamOutT, _StreamErrT]]: ...
+@overload
+def pipes(
+    stdout: _PIPE | _StreamOutT = 3,
+    stderr: _STDOUT | _PIPE | _StreamErrT = 3,
+    encoding: str | None = ...,
+    bufsize: int | None = None,
+) -> _GeneratorContextManager[tuple[io.BytesIO | io.StringIO | _StreamOutT, io.BytesIO | io.StringIO | _StreamErrT | None]]: ...
+def sys_pipes_forever(encoding: str = ..., bufsize: int | None = None) -> None: ...
+def stop_sys_pipes() -> None: ...
+def load_ipython_extension(ip: _InteractiveShell) -> None: ...
+def unload_ipython_extension(ip: _InteractiveShell) -> None: ...