Preserve class signature after wrapping with @warnings.deprecated

Author: XuehaiPan

Lib/warnings.py

@@ -590,17 +590,49 @@ def __call__(self, arg, /):
         if category is None:
             arg.__deprecated__ = msg
             return arg
-        elif isinstance(arg, type):
+
+        def update_signature(original_func):
+            # Ensure that the signature of the decorated callable matches the original one
+
+            def wrapper(func):
+                import inspect
+
+                try:
+                    original_signature = inspect.signature(original_func)
+                except ValueError:
+                    pass
+                else:
+                    signature = inspect.signature(func)
+                    if signature != original_signature:
+                        try:
+                            func.__text_signature__ = str(original_signature)
+                        except (AttributeError, TypeError):
+                            pass
+
+                return func
+
+            return wrapper
+
+        if isinstance(arg, type):
             import functools
             from types import MethodType
 
             original_new = arg.__new__
+            is_object_new = original_new is object.__new__
+            if is_object_new:
+                def wraps(wrapped):
+                    def identity(func):
+                        return func
+                    return identity
+            else:
+                wraps = functools.wraps
 
-            @functools.wraps(original_new)
+            @update_signature(arg)
+            @wraps(original_new)
             def __new__(cls, *args, **kwargs):
                 if cls is arg:
                     warn(msg, category=category, stacklevel=stacklevel + 1)
-                if original_new is not object.__new__:
+                if not is_object_new:
                     return original_new(cls, *args, **kwargs)
                 # Mirrors a similar check in object.__new__.
                 elif cls.__init__ is object.__init__ and (args or kwargs):
@@ -616,15 +648,19 @@ def __new__(cls, *args, **kwargs):
             if isinstance(original_init_subclass, MethodType):
                 original_init_subclass = original_init_subclass.__func__
 
+                @update_signature(original_init_subclass)
                 @functools.wraps(original_init_subclass)
                 def __init_subclass__(*args, **kwargs):
                     warn(msg, category=category, stacklevel=stacklevel + 1)
                     return original_init_subclass(*args, **kwargs)
 
                 arg.__init_subclass__ = classmethod(__init_subclass__)
+
             # Or otherwise, which likely means it's a builtin such as
             # object's implementation of __init_subclass__.
             else:
+
+                @update_signature(original_init_subclass)
                 @functools.wraps(original_init_subclass)
                 def __init_subclass__(*args, **kwargs):
                     warn(msg, category=category, stacklevel=stacklevel + 1)
@@ -639,6 +675,7 @@ def __init_subclass__(*args, **kwargs):
             import functools
             import inspect
 
+            @update_signature(arg)
             @functools.wraps(arg)
             def wrapper(*args, **kwargs):
                 warn(msg, category=category, stacklevel=stacklevel + 1)