Распространение декораторов класса на унаследованные классы

Вот как вы можете добиться этого, используя метакласс вместо украшения класса:

import inspect
import functools

def test_decorator(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        print(func.__name__, args, kwargs)
        res = func(*args, **kwargs)
        return res
    return wrapper

def make_test_deco_type(decorator):
    class TestDecoType(type):
        def __new__(cls, clsname, bases, dct):
            for name, value in dct.items():
                if name.startswith('test') and inspect.isroutine(value):
                    dct[name] = decorator(value)
            return super().__new__(cls, clsname, bases, dct)
    return TestDecoType

class Potato(object, metaclass=make_test_deco_type(test_decorator)):
    def test_method(self):
        print('in method')

class Spud(Potato):
    def test_derived(self):
        print('in derived')

В Python 2.x вы должны использовать __metaclass__ = make_test_deco_type(test_decorator) в качестве первой строки тела класса вместо части metaclass=... оператора класса. Вам также нужно будет заменить super() на super(TestDecoType, cls).

Вы не можете избежать украшения производных классов; вы можете найти подклассы класса после того, как подклассы были украшены, но не можете их автоматически украшать. Используйте метакласс вместо того, чтобы вам нужно было такое поведение.

Вы можете сделать одно из двух:

1. Обнаружение уже оформленных методов; если есть атрибут __wrapped__, у вас есть оболочка:

   def for_all_test_methods(decorator):
       def decorate(cls):
           for name, value in inspect.getmembers(cls, inspect.isroutine):
               if name.startswith('test') and not hasattr(value, '__wrapped__'):
                   setattr(cls, name, test_decorator(getattr(cls, name)))
           return cls
       return decorate
   

2. Ограничьте декоратор класса только прямыми методами:

   def for_all_test_methods(decorator):
       def decorate(cls):
           for name, value in cls.__dict__.iteritems():
               if name.startswith('test') and inspect.isroutine(value)):
                   setattr(cls, name, test_decorator(getattr(cls, name)))
           return cls
       return decorate