如何为我的 Django/Python 视图编写装饰器？

Question

这是我的观点。基本上，它根据是否登录返回不同的响应。

@check_login()
def home(request):
    if is_logged_in(request): 
        return x
    else:
        return y

这是我的装饰器代码。我只是想检查请求是否有标头，如果有，则将其登录。

#decorator to log the user in if there are headers
def check_login():
    def check_dec(func):
        if request.META['username'] == "blah":
            login(request, user)

    return check_dec

问题是..我不知道在这种情况下如何编写正确的装饰器！论据是什么？有哪些功能？如何？

Answer 1

仅使用 @check_login 而不是 check_login() - 否则你的装饰器必须在你执行 home = check_login()(home)

这是一个装饰器示例：

def check_login(method):
    @functools.wraps(method)
    def wrapper(request, *args, **kwargs):
        if request.META['username'] == "blah"
            login(request, user) # where does user come from?!
        return method(request, *args, **kwargs)
    return wrapper

如果用户名字段设置为“blah”，则该装饰器将调用执行您的登录函数，然后调用原始方法。

Answer 2

普通装饰器只是一个函数，它接受一个函数或类并返回其他内容（通常是相同的类型，但这不是必需的）。参数化装饰器是返回装饰器的函数。

因此，考虑到这一点，我们创建一个闭包并返回它：

def check_login(func):
  def inner(request, *args, **kwargs):
    if request.META['username'] == 'blah':
      login(request, user) # I have no idea where user comes from
    func(request, *args, **kwargs)
  return inner

Answer 3

这是我编写的一个装饰器，用于检查用户是否属于某个组。它显示了更多一点。

https://github.com/mzupan/django-decorators/blob/master /auth.py

你可以像用户可以在 group1 或 group2 中一样使用它

@group_required(["group1", "group2"])
def show_index(request):
    view_code_here

，如果他们不是，他们会得到 404 页面