From 5f2abae61ec69264b835dcabe2cdabe57b9a990e Mon Sep 17 00:00:00 2001 From: Adrian Garcia Badaracco <1755071+adriangb@users.noreply.github.com> Date: Mon, 11 Apr 2022 09:51:25 -0500 Subject: bpo-44807: Allow Protocol classes to define __init__ (GH-31628) Co-authored-by: Jelle Zijlstra --- Lib/test/test_typing.py | 26 ++++++++++++++++++++++ Lib/typing.py | 3 ++- .../2022-03-02-04-25-58.bpo-44807.gHNC9J.rst | 1 + 3 files changed, 29 insertions(+), 1 deletion(-) create mode 100644 Misc/NEWS.d/next/Library/2022-03-02-04-25-58.bpo-44807.gHNC9J.rst diff --git a/Lib/test/test_typing.py b/Lib/test/test_typing.py index e09f8aa..b884f7b 100644 --- a/Lib/test/test_typing.py +++ b/Lib/test/test_typing.py @@ -1598,6 +1598,32 @@ class ProtocolTests(BaseTestCase): with self.assertRaises(TypeError): CG[int](42) + def test_protocol_defining_init_does_not_get_overridden(self): + # check that P.__init__ doesn't get clobbered + # see https://bugs.python.org/issue44807 + + class P(Protocol): + x: int + def __init__(self, x: int) -> None: + self.x = x + class C: pass + + c = C() + P.__init__(c, 1) + self.assertEqual(c.x, 1) + + def test_concrete_class_inheriting_init_from_protocol(self): + class P(Protocol): + x: int + def __init__(self, x: int) -> None: + self.x = x + + class C(P): pass + + c = C(1) + self.assertIsInstance(c, C) + self.assertEqual(c.x, 1) + def test_cannot_instantiate_abstract(self): @runtime_checkable class P(Protocol): diff --git a/Lib/typing.py b/Lib/typing.py index 26c6b8c..ec8cbbd 100644 --- a/Lib/typing.py +++ b/Lib/typing.py @@ -1997,7 +1997,8 @@ class Protocol(Generic, metaclass=_ProtocolMeta): issubclass(base, Generic) and base._is_protocol): raise TypeError('Protocols can only inherit from other' ' protocols, got %r' % base) - cls.__init__ = _no_init_or_replace_init + if cls.__init__ is Protocol.__init__: + cls.__init__ = _no_init_or_replace_init class _AnnotatedAlias(_GenericAlias, _root=True): diff --git a/Misc/NEWS.d/next/Library/2022-03-02-04-25-58.bpo-44807.gHNC9J.rst b/Misc/NEWS.d/next/Library/2022-03-02-04-25-58.bpo-44807.gHNC9J.rst new file mode 100644 index 0000000..4757d34 --- /dev/null +++ b/Misc/NEWS.d/next/Library/2022-03-02-04-25-58.bpo-44807.gHNC9J.rst @@ -0,0 +1 @@ +:class:`typing.Protocol` no longer silently replaces :meth:`__init__` methods defined on subclasses. Patch by Adrian Garcia Badaracco. -- cgit v0.12