1

我写了一个装饰器来显示对象创建的成功消息:

from django.contrib import messages

def success_message(klass):
    def form_valid(self, form):
        response = super(klass, self).form_valid(form)
        messages.success(self.request, 'Object added successfully')
        return response

    klass.form_valid = form_valid
    return klass

并用它来装饰基于类的通用视图:

@success_message
class BandCreateView(CreateView):
    model = Band

现在我想参数化装饰器,所以这是可能的:

@success_message('Band created successfully.')
class BandCreateView(CreateView):
    model = Band

我该怎么做?我尝试添加message参数,success_message但编译器抱怨参数计数不匹配,所以我认为必须有另一种方法。

4

3 回答 3

3

看起来你必须使用闭包:

def decorator(arg):
    def wrap(klass): ...
    return wrap

因为你的电话被评估为

class BandCreateView(CreateView): ...
BandCreateView = @success_message('Band created successfully.')(BandCreateView)

注意双重通话

于 2011-09-29T00:09:57.463 回答
1

我以前将这种事情作为 mixin 进行,但装饰器更有意义(并且添加到类中的麻烦更少)。您可能希望获取表单实例的详细名称,如下所示,以消除向装饰器传递消息的需要:

from django.utils.translation import ugettext_lazy as _

def ucfirst(value):
    return value[0].upper() + value[1:]

def success_message(klass):
    __orig_form_valid = klass.form_valid
    def form_valid(self, form):
        response = __orig_form_valid(self, form)
        messages.success(self.request, _("%(object)s \"%(object_name)s\" was saved successfully.") %
                         {'object': ucfirst(form.instance._meta.verbose_name), 'object_name': unicode(form.instance)})
        return response

    klass.form_valid = form_valid
    return klass

这将产生类似于以下内容的成功消息:

客户“ACME Inc.” 已成功保存。

于 2012-02-28T12:06:58.687 回答
0

TL;博士

无参数装饰器被声明为class -> class函数。
带参数的装饰器被声明为高阶args -> (class -> class)函数。

它接受参数并返回一个函数,该函数将接受类并返回类(即“简单”装饰器)。

解释

t.dubrownik相关主题的回答确实救了我。

无论如何,带参数的装饰器的语法有点不同——<strong>带参数的装饰器应该返回一个函数,该函数将接受一个函数并返回另一个函数。所以它应该真的返回一个普通的装饰器。

虽然他说的是函数装饰器,但这个概念也适用于类装饰器。
这是我得到的:

from django.contrib import messages

def success_message(message):
    def actual_decorator(klass):
        def form_valid(self, form):
            response = super(klass, self).form_valid(form)
            messages.success(self.request, message)
            return response

        klass.form_valid = form_valid
        return klass

    return actual_decorator

注意它actual_decorator本身是如何重复无参数版本的success_message,以及success_message现在是一个高阶函数,因为它的返回值是函数本身。

于 2011-09-29T00:09:35.000 回答