From 46c759d76d02cb873bae55fac37dfe71750e3bef Mon Sep 17 00:00:00 2001 From: Yury Selivanov Date: Wed, 27 May 2015 21:56:53 -0400 Subject: Issue 24298: Fix signature() to properly unwrap wrappers around bound methods --- Lib/inspect.py | 8 ++++++++ Lib/test/test_inspect.py | 13 +++++++++++++ Misc/NEWS | 3 +++ 3 files changed, 24 insertions(+) diff --git a/Lib/inspect.py b/Lib/inspect.py index a91f749..49e66ce 100644 --- a/Lib/inspect.py +++ b/Lib/inspect.py @@ -1911,6 +1911,14 @@ def _signature_internal(obj, follow_wrapper_chains=True, skip_bound_arg=True): # Was this function wrapped by a decorator? if follow_wrapper_chains: obj = unwrap(obj, stop=(lambda f: hasattr(f, "__signature__"))) + if isinstance(obj, types.MethodType): + # If the unwrapped object is a *method*, we might want to + # skip its first parameter (self). + # See test_signature_wrapped_bound_method for details. + return _signature_internal( + obj, + follow_wrapper_chains=follow_wrapper_chains, + skip_bound_arg=skip_bound_arg) try: sig = obj.__signature__ diff --git a/Lib/test/test_inspect.py b/Lib/test/test_inspect.py index 8d92f82..58a6cd2 100644 --- a/Lib/test/test_inspect.py +++ b/Lib/test/test_inspect.py @@ -1939,6 +1939,19 @@ class TestSignatureObject(unittest.TestCase): with self.assertRaisesRegex(ValueError, 'invalid method signature'): self.signature(Test()) + def test_signature_wrapped_bound_method(self): + # Issue 24298 + class Test: + def m1(self, arg1, arg2=1) -> int: + pass + @functools.wraps(Test().m1) + def m1d(*args, **kwargs): + pass + self.assertEqual(self.signature(m1d), + ((('arg1', ..., ..., "positional_or_keyword"), + ('arg2', 1, ..., "positional_or_keyword")), + int)) + def test_signature_on_classmethod(self): class Test: @classmethod diff --git a/Misc/NEWS b/Misc/NEWS index a58ddf2..7658009 100644 --- a/Misc/NEWS +++ b/Misc/NEWS @@ -274,6 +274,9 @@ Library - Issue #23898: Fix inspect.classify_class_attrs() to support attributes with overloaded __eq__ and __bool__. Patch by Mike Bayer. +- Issue #24298: Fix inspect.signature() to correctly unwrap wrappers + around bound methods. + IDLE ---- -- cgit v0.12