2015-08-04 05:22:37 +00:00
|
|
|
|
# coding=utf-8
|
2015-08-12 15:18:22 +00:00
|
|
|
|
from functools import wraps
|
2015-08-23 12:45:02 +00:00
|
|
|
|
from django.http import HttpResponse, HttpResponseRedirect
|
2015-08-04 05:22:37 +00:00
|
|
|
|
from django.shortcuts import render
|
|
|
|
|
|
2015-08-14 02:21:22 +00:00
|
|
|
|
from utils.shortcuts import error_response, error_page
|
2015-09-22 08:18:32 +00:00
|
|
|
|
from .models import User, SUPER_ADMIN
|
2015-08-04 05:22:37 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def login_required(func):
|
2015-08-12 15:18:22 +00:00
|
|
|
|
@wraps(func)
|
2015-08-04 05:22:37 +00:00
|
|
|
|
def check(*args, **kwargs):
|
|
|
|
|
# 在class based views 里面,args 有两个元素,一个是self, 第二个才是request,
|
|
|
|
|
# 在function based views 里面,args 只有request 一个参数
|
2015-09-22 08:18:32 +00:00
|
|
|
|
if len(args) == 2:
|
|
|
|
|
request = args[-1]
|
|
|
|
|
else:
|
|
|
|
|
request = args[0]
|
2015-08-04 05:22:37 +00:00
|
|
|
|
if request.user.is_authenticated():
|
|
|
|
|
return func(*args, **kwargs)
|
|
|
|
|
if request.is_ajax():
|
|
|
|
|
return error_response(u"请先登录")
|
|
|
|
|
else:
|
2015-08-23 12:45:02 +00:00
|
|
|
|
return HttpResponseRedirect("/login/")
|
2015-08-04 05:22:37 +00:00
|
|
|
|
return check
|
|
|
|
|
|
|
|
|
|
|
2015-08-06 04:25:16 +00:00
|
|
|
|
def admin_required(func):
|
2015-08-12 15:18:22 +00:00
|
|
|
|
@wraps(func)
|
2015-08-06 04:25:16 +00:00
|
|
|
|
def check(*args, **kwargs):
|
2015-09-22 08:18:32 +00:00
|
|
|
|
if len(args) == 2:
|
|
|
|
|
request = args[-1]
|
|
|
|
|
else:
|
|
|
|
|
request = args[0]
|
2015-08-06 04:25:16 +00:00
|
|
|
|
if request.user.is_authenticated() and request.user.admin_type:
|
|
|
|
|
return func(*args, **kwargs)
|
|
|
|
|
if request.is_ajax():
|
|
|
|
|
return error_response(u"需要管理员权限")
|
|
|
|
|
else:
|
2015-08-23 12:45:02 +00:00
|
|
|
|
return error_page(request, u"需要管理员权限,如果没有登录,请先登录")
|
2015-08-06 04:25:16 +00:00
|
|
|
|
return check
|
2015-09-22 08:18:32 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def super_admin_required(func):
|
|
|
|
|
@wraps(func)
|
|
|
|
|
def check(*args, **kwargs):
|
|
|
|
|
if len(args) == 2:
|
|
|
|
|
request = args[-1]
|
|
|
|
|
else:
|
|
|
|
|
request = args[0]
|
|
|
|
|
if request.user.is_authenticated() and request.user.admin_type == SUPER_ADMIN:
|
|
|
|
|
return func(*args, **kwargs)
|
|
|
|
|
if request.is_ajax():
|
|
|
|
|
return error_response(u"需要超级管理员权限")
|
|
|
|
|
else:
|
|
|
|
|
return error_page(request, u"需要超级管理员权限")
|
|
|
|
|
|
|
|
|
|
return check
|