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 回答