增加 login_required decorator 和对应的测试

This commit is contained in:
virusdefender
2015-08-04 13:22:37 +08:00
parent 5cd58164c9
commit c72624e2d5
3 changed files with 114 additions and 1 deletions

24
account/decorators.py Normal file
View File

@@ -0,0 +1,24 @@
# coding=utf-8
from django.http import HttpResponse
from django.shortcuts import render
from utils.shortcuts import error_response
from .models import User
def login_required(func):
def check(*args, **kwargs):
# 在class based views 里面args 有两个元素一个是self, 第二个才是request
# 在function based views 里面args 只有request 一个参数
request = args[-1]
if request.user.is_authenticated():
return func(*args, **kwargs)
if request.is_ajax():
return error_response(u"请先登录")
else:
return render(request, "utils/error.html", {"error": u"请先登录"})
return check
def admin_required():
pass