arg.__new__ = staticmethod(__new__)
- original_init_subclass = arg.__init_subclass__
- # We need slightly different behavior if __init_subclass__
- # is a bound method (likely if it was implemented in Python)
- if isinstance(original_init_subclass, MethodType):
- original_init_subclass = original_init_subclass.__func__
+ if "__init_subclass__" in arg.__dict__:
+ # __init_subclass__ is directly present on the decorated class.
+ # Synthesize a wrapper that calls this method directly.
+ original_init_subclass = arg.__init_subclass__
+ # We need slightly different behavior if __init_subclass__
+ # is a bound method (likely if it was implemented in Python).
+ # Otherwise, it likely means it's a builtin such as
+ # object's implementation of __init_subclass__.
+ if isinstance(original_init_subclass, MethodType):
+ original_init_subclass = original_init_subclass.__func__
@functools.wraps(original_init_subclass)
def __init_subclass__(*args, **kwargs):
_wm.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:
- @functools.wraps(original_init_subclass)
- def __init_subclass__(*args, **kwargs):
+ def __init_subclass__(cls, *args, **kwargs):
_wm.warn(msg, category=category, stacklevel=stacklevel + 1)
- return original_init_subclass(*args, **kwargs)
+ return super(arg, cls).__init_subclass__(*args, **kwargs)
- arg.__init_subclass__ = __init_subclass__
+ arg.__init_subclass__ = classmethod(__init_subclass__)
arg.__deprecated__ = __new__.__deprecated__ = msg
__init_subclass__.__deprecated__ = msg
self.assertEqual(D.inited, 3)
+ def test_existing_init_subclass_in_sibling_base(self):
+ @deprecated("A will go away soon")
+ class A:
+ pass
+ class B:
+ def __init_subclass__(cls, x):
+ super().__init_subclass__()
+ cls.inited = x
+
+ with self.assertWarnsRegex(DeprecationWarning, "A will go away soon"):
+ class C(A, B, x=42):
+ pass
+ self.assertEqual(C.inited, 42)
+
+ with self.assertWarnsRegex(DeprecationWarning, "A will go away soon"):
+ class D(B, A, x=42):
+ pass
+ self.assertEqual(D.inited, 42)
+
def test_init_subclass_has_correct_cls(self):
init_subclass_saw = None