diff --git a/pyrightconfig.stricter.json b/pyrightconfig.stricter.json index 9682568f3f9c..126241011a99 100644 --- a/pyrightconfig.stricter.json +++ b/pyrightconfig.stricter.json @@ -102,7 +102,6 @@ "stubs/tqdm", "stubs/vobject", "stubs/workalendar", - "stubs/wurlitzer", ], "typeCheckingMode": "strict", // TODO: Complete incomplete stubs diff --git a/stubs/wurlitzer/wurlitzer.pyi b/stubs/wurlitzer/wurlitzer.pyi index 6e68f5f45291..948f6e58985b 100644 --- a/stubs/wurlitzer/wurlitzer.pyi +++ b/stubs/wurlitzer/wurlitzer.pyi @@ -5,6 +5,7 @@ import io import logging from _typeshed import SupportsWrite from contextlib import _GeneratorContextManager +from threading import Thread from types import TracebackType from typing import Any, Final, Literal, Protocol, TextIO, TypeVar, overload from typing_extensions import Self, TypeAlias @@ -29,6 +30,10 @@ _InteractiveShell: TypeAlias = Any class Wurlitzer: flush_interval: float encoding: str | None + thread: Thread | None + handle: tuple[ + _LogPipe | SupportsWrite[str] | SupportsWrite[bytes] | None, _LogPipe | SupportsWrite[str] | SupportsWrite[bytes] | None + ] def __init__( self, @@ -37,7 +42,11 @@ class Wurlitzer: encoding: str | None = ..., bufsize: int | None = ..., ) -> None: ... - def __enter__(self): ... + def __enter__( + self, + ) -> tuple[ + _LogPipe | SupportsWrite[str] | SupportsWrite[bytes] | None, _LogPipe | SupportsWrite[str] | SupportsWrite[bytes] | None + ]: ... def __exit__( self, exc_type: type[BaseException] | None, exc_value: BaseException | None, traceback: TracebackType | None ) -> None: ...