diff options
author | James Hilton-Balfe <gobot1234yt@gmail.com> | 2022-03-08 03:50:46 (GMT) |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-03-08 03:50:46 (GMT) |
commit | f391f9bf28f0bba7939d9f9e5a7a6396d2b0df62 (patch) | |
tree | 4151190384181dca5446c61e6dd7c3230739a6e0 /Lib | |
parent | 50731297a9b6d57eec3b3f89522785b23f7b3e71 (diff) | |
download | cpython-f391f9bf28f0bba7939d9f9e5a7a6396d2b0df62.zip cpython-f391f9bf28f0bba7939d9f9e5a7a6396d2b0df62.tar.gz cpython-f391f9bf28f0bba7939d9f9e5a7a6396d2b0df62.tar.bz2 |
bpo-46170: Improve the error message when subclassing NewType (GH-30268)
Co-authored-by: Alex Waygood <alex.waygood@gmail.com>
Co-authored-by: Nikita Sobolev <mail@sobolevn.me>
Co-authored-by: Ken Jin <28750310+Fidget-Spinner@users.noreply.github.com>
Diffstat (limited to 'Lib')
-rw-r--r-- | Lib/test/test_typing.py | 11 | ||||
-rw-r--r-- | Lib/typing.py | 15 |
2 files changed, 26 insertions, 0 deletions
diff --git a/Lib/test/test_typing.py b/Lib/test/test_typing.py index 17d78cf..c76aa0a 100644 --- a/Lib/test/test_typing.py +++ b/Lib/test/test_typing.py @@ -4423,6 +4423,17 @@ class NewTypeTests: ) exec(code, {}) + def test_error_message_when_subclassing(self): + with self.assertRaisesRegex( + TypeError, + re.escape( + "Cannot subclass an instance of NewType. Perhaps you were looking for: " + "`ProUserId = NewType('ProUserId', UserId)`" + ) + ): + class ProUserId(UserId): + ... + class NewTypePythonTests(NewTypeTests, BaseTestCase): module = py_typing diff --git a/Lib/typing.py b/Lib/typing.py index 360129e..721afb7 100644 --- a/Lib/typing.py +++ b/Lib/typing.py @@ -2639,6 +2639,21 @@ class NewType: if def_mod != 'typing': self.__module__ = def_mod + def __mro_entries__(self, bases): + # We defined __mro_entries__ to get a better error message + # if a user attempts to subclass a NewType instance. bpo-46170 + superclass_name = self.__name__ + + class Dummy: + def __init_subclass__(cls): + subclass_name = cls.__name__ + raise TypeError( + f"Cannot subclass an instance of NewType. Perhaps you were looking for: " + f"`{subclass_name} = NewType({subclass_name!r}, {superclass_name})`" + ) + + return (Dummy,) + def __repr__(self): return f'{self.__module__}.{self.__qualname__}' |