自定义Django的Decorators

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

@login_required
def sayhello(request):
	print "helloworld"

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

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

1.建一个文件：decorators.py

2.文件中代码如下：

# -*- coding: utf-8 -*-
from django.core.urlresolvers import reverse 
from django.http import HttpResponseRedirect
from django.template import RequestContext
from .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_user
def checkuserinfo(request):
<span style="white-space:pre">	</span>print "hello"

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