如何在 Django 中创建装饰器来装饰视图

Question

我有意见：

@decorator
def func(request):
  hello = "hello"
  return render_to_responce("test.html", locals() )

和模板test.html：

{{ hello }}
{{ username }}

我想为func(request) 编写装饰器，它将一个变量USERNAME 添加到函数中，并在模板中返回两个参数。我试图使它如下：

def decorator(func):
    def wrapper( request, *args, **kwargs):
        username = request.user.username
        q = func(request, *args, **kwargs)
        #what I need add here I do not know ...
        return q   
    return wrapper

Answer 1

如果您的视图如下所示：

def func(request, username):
    hello = "hello"
    return render_to_responce("test.html", locals() )

你可以这样写一个装饰器：

from functools import wraps
def pass_username(view):
    @wraps(view)
    def wrapper(request, *args, **kwargs):
        return view(request, request.user.username, *args, **kwargs)
    return wrapper

然后像这样使用它：

@pass_username
def func(request, username):
    hello = "hello"
    return render_to_response("test.html", locals())

（只要确保您将在urls.py 中将其视为def func(request):，没有username - 此参数将由装饰器提供）

但这确实是一个非常简单的情况，并不需要单独的装饰器（无论如何，它只是视图定义中的额外一行）。