From 5732b8f34e3edc129135731217fd1a654ef961ad Mon Sep 17 00:00:00 2001 From: Kevin Kirsche Date: Tue, 26 Jul 2022 05:17:45 -0400 Subject: [PATCH] email.Message.walk yields self (#8398) This updates the generator used by `Message` to return `Self` so that it correctly changes when a `MIMEPart` or `EmailMessage`. Fixes #8290 --- stdlib/email/message.pyi | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/stdlib/email/message.pyi b/stdlib/email/message.pyi index 6544f8fc2..4e8f600f7 100644 --- a/stdlib/email/message.pyi +++ b/stdlib/email/message.pyi @@ -1,3 +1,4 @@ +from _typeshed import Self from collections.abc import Generator, Iterator, Sequence from email import _ParamsType, _ParamType from email.charset import Charset @@ -55,7 +56,7 @@ class Message: def set_boundary(self, boundary: str) -> None: ... def get_content_charset(self, failobj: _T = ...) -> _T | str: ... def get_charsets(self, failobj: _T = ...) -> _T | list[str]: ... - def walk(self) -> Generator[Message, None, None]: ... + def walk(self: Self) -> Generator[Self, None, None]: ... def get_content_disposition(self) -> str | None: ... def as_string(self, unixfrom: bool = ..., maxheaderlen: int = ..., policy: Policy | None = ...) -> str: ... def as_bytes(self, unixfrom: bool = ..., policy: Policy | None = ...) -> bytes: ...