自定义Django的Decorators

在Django中我们在方法前面会看到这样一类的方法：

@login_requireddef sayhello(request):print "helloworld"

其中的 login_required 我们称为 装饰器。

如果你要自定义装饰器，步骤如下：

1.建一个文件：decorators.py

2.文件中代码如下：

# -*- coding: utf-8 -*-from django.core.urlresolvers import reverse from django.http import HttpResponseRedirectfrom django.template import RequestContextfrom .models import User# 验证用户def check_user(view_func): # 此方法为包裹view_func的方法# 我们可以在这里定义自己需要的功能    def _wrapped_view_func(request, *args, **kwargs):        try:            user = User.objects.get(user=request.user.id)        except CampusUser.DoesNotExist:            user = User.objects.none()        # 如果用户名为bendan，则重定向到登陆页面        if user and user.name == "bendan":            return HttpResponseRedirect('/login')        # 返回包裹的方法        return view_func(request, *args, **kwargs)    return _wrapped_view_func

3.在views.py中：

# -*- coding: utf-8 -*-from .decorators import check_user@check_userdef checkuserinfo(request):<span style="white-space:pre"></span>print "hello"

这样设置后，当用户名为bendan的人调用方法checkuserinfo时，页面就会重定向到登陆页面；而其他用户则不受影响，继续调用checkuserinfo方法。