在一个类中创建一个装饰器，可以访问（当前）类本身

Question

目前，我正在以这种方式进行操作：

class Spam(object):

    decorated = None

    @classmethod
    def decorate(cls, funct):
        if cls.decorated is None:
            cls.decorated = []
        cls.decorated.append(funct)
        return funct


class Eggs(Spam):
    pass


@Eggs.decorate
def foo():
    print "spam and eggs"


print Eggs.decorated # [<function foo at 0x...>]
print Spam.decorated # None

我需要能够在所示的子类中执行此操作。 问题是，我似乎无法弄清楚如何使decorated字段在实例之间不共享。 现在，我有了一个棘手的解决方案，方法是先将其设置为“ None ，然后在装饰函数时进行检查，但这仅是一种方法。 换句话说，如果我将Eggs子类化，然后使用Eggs.decorate函数装饰某些东西，则它将影响所有子类。

我想我的问题是：是否有可能在基类和子类之间没有共享的可变类字段？

Answer 1

我通过使用元类弄清楚了。 感谢所有发布的人。 如果有人遇到类似问题，这是我的解决方案：

class SpamMeta(type):

    def __new__(cls, name, bases, dct):
        SpamType = type.__new__(cls, name, bases, dct)
        SpamType.decorated = []
        return SpamType


class Spam(object):

    __metaclass__ = SpamMeta

    @classmethod
    def decorate(cls, funct):
        cls.decorated.append(funct)
        return funct


class Eggs(Spam):
    pass


@Eggs.decorate
def foo():
    print "spam and eggs"


print Eggs.decorated # [<function foo at 0x...>]
print Spam.decorated # []

Answer 2

我很确定你不会。 我考虑过使用property（）进行此操作，但是不幸的是，该类本身的类（需要添加属性的类）是ClassType本身。

您可以这样编写装饰器，但是它会稍微改变界面：

class Spam(object):
    decorated = {}

    @classmethod
    def get_decorated_methods(cls):
        return cls.decorated.setdefault(cls, [])

    @classmethod
    def decorate(cls, funct):
        cls.get_decorated_methods().append(funct)
        return funct


class Eggs(Spam):
    pass


@Spam.decorate
def foo_and_spam():
    print "spam"

@Eggs.decorate
def foo_and_eggs():
    print "eggs"

print Eggs.get_decorated_methods() # [<function foo_and_eggs at 0x...>]
print Spam.get_decorated_methods() # [<function foo_and_spam at 0x...>]

Answer 3

并不是说我有任何针对元类的东西，但是如果没有它们，您也可以解决它：

from collections import defaultdict

class Spam(object):
    _decorated = defaultdict(list)

    @classmethod
    def decorate(cls, func):
        cls._decorated[cls].append(func)
        return func

    @classmethod
    def decorated(cls):
        return cls._decorated[cls]


class Eggs(Spam):
    pass

@Eggs.decorate
def foo():
    print "spam and eggs"

print Eggs.decorated() # [<function foo at 0x...>]
print Spam.decorated() # []

不能在类对象上具有属性（除非再次还原为元类），因此必须通过类方法再次获得修饰方法的列表。 与元类解决方案相比，还涉及一个额外的间接层。