Question

我有一个django模型需要参考自定义用户模型进行一些处理。

我无法在类加载时使用此模型的类，因为类的加载顺序未知。

所以我需要在运行时添加一些类属性，此时我将它们添加到__init__或__new__中：

def __new__(cls, *args, **kwargs):
    # hack to avoid INSTALLED_APPS initialization conflicts.
    # get_user_model() can't be called from this module at class loading time,
    # so some class attributes must be added later.
    # Metaclasses could me more appropiate but I don't want to override
    # dango's metaclasses.
    if not hasattr(cls, '_reverse_field_name_to_user'):
        cls._find_reverse_field_name_to_user()
    return Group.__new__(cls, *args, **kwargs)

它有效，但看起来很糟糕，所以我考虑过使用像@lazyclassproperty这样的属性。

我找到了几个@classproperty和@lazyproperty装饰器，但两者都没有，我不知道如何自己写一个。

问题：我怎么能编写这样的装饰器？或建议另一种更清洁的替代方案，以解决我目前的愚蠢行为。

Answer 1

Pyramid框架有一个非常好的装饰器叫reify，但它只能在实例级别工作，你需要类级别，所以让我们稍微修改一下

class class_reify(object):
    def __init__(self, wrapped):
        self.wrapped = wrapped
        try:
            self.__doc__ = wrapped.__doc__
        except: # pragma: no cover
            pass

    # original sets the attributes on the instance
    # def __get__(self, inst, objtype=None):
    #    if inst is None:
    #        return self
    #    val = self.wrapped(inst)
    #    setattr(inst, self.wrapped.__name__, val)
    #    return val

    # ignore the instance, and just set them on the class
    # if called on a class, inst is None and objtype is the class
    # if called on an instance, inst is the instance, and objtype 
    # the class
    def __get__(self, inst, objtype=None):
        # ask the value from the wrapped object, giving it
        # our class
        val = self.wrapped(objtype)

        # and set the attribute directly to the class, thereby
        # avoiding the descriptor to be called multiple times
        setattr(objtype, self.wrapped.__name__, val)

        # and return the calculated value
        return val

class Test(object):
    @class_reify
    def foo(cls):
        print "foo called for class", cls
        return 42

print Test.foo
print Test.foo

运行程序并打印

foo called for class <class '__main__.Test'>
42
42

懒惰的类属性装饰器

1 个答案: