Preserve __module__

Author: srinivasreddy

Lib/warnings.py

@@ -597,7 +597,6 @@ def __call__(self, arg, /):
             import functools
             from types import MethodType
 
-            wrapper_module = arg.__module__
             original_new = arg.__new__
 
             @functools.wraps(original_new)
@@ -611,9 +610,8 @@ def __new__(cls, *args, **kwargs):
                     raise TypeError(f"{cls.__name__}() takes no arguments")
                 else:
                     return original_new(cls)
-
+            __new__.__module__ = arg.__module__
             arg.__new__ = staticmethod(__new__)
-            arg.__module__  = wrapper_module
 
             original_init_subclass = arg.__init_subclass__
             # We need slightly different behavior if __init_subclass__
@@ -637,6 +635,7 @@ def __init_subclass__(*args, **kwargs):
 
                 arg.__init_subclass__ = __init_subclass__
 
+            __init_subclass__.__module__ = arg.__module__
             arg.__deprecated__ = __new__.__deprecated__ = msg
             __init_subclass__.__deprecated__ = msg
             return arg