Skip to content

Commit 052d2b9

Browse files
authored
Fix base class for email.generator.BytesGenerator (#10299)
And use `Self` as the return type for `email.generator.Generator.clone`. Signed-off-by: oleg.hoefling <[email protected]>
1 parent c88ce93 commit 052d2b9

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

stdlib/email/generator.pyi

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,12 @@
11
from _typeshed import SupportsWrite
22
from email.message import Message
33
from email.policy import Policy
4+
from typing_extensions import Self
45

56
__all__ = ["Generator", "DecodedGenerator", "BytesGenerator"]
67

78
class Generator:
8-
def clone(self, fp: SupportsWrite[str]) -> Generator: ...
9+
def clone(self, fp: SupportsWrite[str]) -> Self: ...
910
def write(self, s: str) -> None: ...
1011
def __init__(
1112
self,
@@ -17,9 +18,7 @@ class Generator:
1718
) -> None: ...
1819
def flatten(self, msg: Message, unixfrom: bool = False, linesep: str | None = None) -> None: ...
1920

20-
class BytesGenerator:
21-
def clone(self, fp: SupportsWrite[bytes]) -> BytesGenerator: ...
22-
def write(self, s: str) -> None: ...
21+
class BytesGenerator(Generator):
2322
def __init__(
2423
self,
2524
outfp: SupportsWrite[bytes],
@@ -28,7 +27,6 @@ class BytesGenerator:
2827
*,
2928
policy: Policy | None = None,
3029
) -> None: ...
31-
def flatten(self, msg: Message, unixfrom: bool = False, linesep: str | None = None) -> None: ...
3230

3331
class DecodedGenerator(Generator):
3432
def __init__(

0 commit comments

Comments
 (0)