3

有人知道这段代码有什么问题吗？

def paginated_instance_method(default_page_size=25):
    def wrap(func):
        @functools.wraps(func)
        def inner(self, page=1, page_size=default_page_size, *args, **kwargs):
            objects = func(self=self, *args, **kwargs)
            return _paginate(objects, page, page_size)
        return inner
    return wrap

class Event(object):
    ...
    @paginated_instance_method
    def get_attending_users(self, *args, **kwargs):
        return User.objects.filter(pk__in=self.attending_list)

我收到以下错误：

    Traceback (most recent call last):
      File "<console>", line 1, in <module>
      File "/Users/zarathustra/Virtual_Envs/hinge/hinge_services/hinge/api/decorators.py", line 108, in wrap
        def inner(self, page=1, page_size=default_page_size, *args, **kwargs):
      File "/System/Library/Frameworks/Python.framework/Versions/2.7/lib/python2.7/functools.py", line 33, in update_wrapper
        setattr(wrapper, attr, getattr(wrapped, attr))
    AttributeError: 'Event' object has no attribute '__name__'

我认为这会起作用的原因是，通过反复试验，我得到了以下装饰器，就像类方法的魅力一样：

def paginated_class_method(default_page_size=25):
    def wrap(func):
        @functools.wraps(func)
        def inner(cls, page=1, page_size=default_page_size, *args, **kwargs):
            objects = func(cls=cls, *args, **kwargs)
            return _paginate(objects, page, page_size)
        return inner
    return wrap
4

3 回答 3

3

paginated_instance_method不是装饰器，它是一个返回装饰器的函数。所以

@paginated_instance_method()
def get_attending_users(self, *args, **kwargs):

（注意括号。）

于 2013-02-26T20:32:18.747 回答
1

你的装饰器有一个额外的间接层次，它把事情扔掉了。当你这样做时：

@paginated_instance_method
def get_attending_users(self, *args, **kwargs):
    return User.objects.filter(pk__in=self.attending_list)

你正在这样做：

def get_attending_users(self, *args, **kwargs):
    return User.objects.filter(pk__in=self.attending_list)
get_attending_users = paginated_instance_method(get_attending_users)

这就是装饰器所做的。请注意，它paginated_instance_method被称为 withget_attending_users作为其参数。这意味着在您的装饰器中，参数default_page_size设置为 function get_attending_users。您的装饰器返回函数wrap，因此get_attending_users设置为该wrap函数。

然后，当您调用Event().get_attending_users()它时wrap(self)，self您的 Event 实例在哪里。 wrap期望参数是一个函数，并尝试返回一个包装该函数的新函数。但是参数不是一个函数，它是一个Event对象，所以functools.wrap在尝试包装它时会失败。

我有一种预感，你想要做的是：

@paginated_instance_method()
def get_attending_users(self, *args, **kwargs):
    return User.objects.filter(pk__in=self.attending_list)

也就是说，你想paginated_instance_method争论。但即使您想使用该参数的默认值，您仍然必须实际调用 paginated_instance_method. 否则，您只需将该方法作为参数传递，这不是paginated_instance_method预期的。

它对类方法“起作用”的原因是类方法将类作为第一个参数，并且类（与实例不同）确实具有__name__属性。但是，我怀疑如果您进一步测试它，您会发现它并没有真正做您想要做的事情，因为它仍然包装类而不是方法。

于 2013-02-26T20:36:44.607 回答
0

这真的很容易，但乍一看很棘手。看看pep 318。

@dec2
@dec1
def func(arg1, arg2, ...):
    pass

这相当于：

def func(arg1, arg2, ...):
    pass
func = dec2(dec1(func))

你有一个额外的包装器，它需要一个装饰器的参数来在包装的函数中使用它（闭包设计模式）。所以你的装饰器看起来像这样：

@dec(arg=True)
def func(arg1, arg2, ...):
    pass

相当于：

def func(arg1, arg2, ...):
    pass
func = dec(arg=True)(func)
于 2014-02-06T12:34:35.043 回答