2015-08-04 05:22:37 +00:00
|
|
|
# coding=utf-8
|
2016-06-23 04:19:16 +00:00
|
|
|
from __future__ import unicode_literals
|
2015-10-20 12:07:35 +00:00
|
|
|
import urllib
|
2015-10-14 01:57:43 +00:00
|
|
|
import functools
|
2015-08-04 05:22:37 +00:00
|
|
|
|
2015-10-14 01:57:43 +00:00
|
|
|
from django.http import HttpResponseRedirect
|
2016-06-23 04:19:16 +00:00
|
|
|
from django.utils.translation import ugettext as _
|
2015-08-04 05:22:37 +00:00
|
|
|
|
2016-07-31 12:26:11 +00:00
|
|
|
from utils.shortcuts import error_response, error_page, redirect_to_login
|
2016-06-23 04:19:16 +00:00
|
|
|
from .models import AdminType
|
2015-08-04 05:22:37 +00:00
|
|
|
|
|
|
|
|
2015-10-14 01:57:43 +00:00
|
|
|
class BasePermissionDecorator(object):
|
|
|
|
def __init__(self, func):
|
|
|
|
self.func = func
|
2015-08-04 05:22:37 +00:00
|
|
|
|
2015-10-14 01:57:43 +00:00
|
|
|
def __get__(self, obj, obj_type):
|
|
|
|
return functools.partial(self.__call__, obj)
|
|
|
|
|
|
|
|
def __call__(self, *args, **kwargs):
|
2015-09-22 08:18:32 +00:00
|
|
|
if len(args) == 2:
|
2015-10-14 01:57:43 +00:00
|
|
|
self.request = args[1]
|
2015-09-22 08:18:32 +00:00
|
|
|
else:
|
2015-10-14 01:57:43 +00:00
|
|
|
self.request = args[0]
|
|
|
|
|
|
|
|
if self.check_permission():
|
2016-06-23 04:19:16 +00:00
|
|
|
if self.request.user.is_disabled:
|
2016-03-27 10:29:58 +00:00
|
|
|
if self.request.is_ajax():
|
2016-06-23 04:19:16 +00:00
|
|
|
return error_response(_("Your account is disabled"))
|
2016-03-27 10:29:58 +00:00
|
|
|
else:
|
2016-06-23 04:19:16 +00:00
|
|
|
return error_page(self.request, _("Your account is disabled"))
|
2015-10-14 01:57:43 +00:00
|
|
|
return self.func(*args, **kwargs)
|
2015-08-06 04:25:16 +00:00
|
|
|
else:
|
2015-10-14 01:57:43 +00:00
|
|
|
if self.request.is_ajax():
|
2016-06-23 04:19:16 +00:00
|
|
|
return error_response(_("Please login in first"))
|
2015-10-14 01:57:43 +00:00
|
|
|
else:
|
2016-07-31 12:26:11 +00:00
|
|
|
return redirect_to_login(self.request)
|
2015-09-22 08:18:32 +00:00
|
|
|
|
2015-10-14 01:57:43 +00:00
|
|
|
def check_permission(self):
|
|
|
|
raise NotImplementedError()
|
|
|
|
|
|
|
|
|
|
|
|
class login_required(BasePermissionDecorator):
|
|
|
|
def check_permission(self):
|
|
|
|
return self.request.user.is_authenticated()
|
|
|
|
|
|
|
|
|
|
|
|
class super_admin_required(BasePermissionDecorator):
|
|
|
|
def check_permission(self):
|
2016-07-31 12:26:11 +00:00
|
|
|
return self.request.user.is_authenticated() and \
|
|
|
|
self.request.user.admin_type == AdminType.SUPER_ADMIN
|
2015-09-22 08:18:32 +00:00
|
|
|
|
|
|
|
|
2015-10-14 01:57:43 +00:00
|
|
|
class admin_required(BasePermissionDecorator):
|
|
|
|
def check_permission(self):
|
2016-07-31 12:26:11 +00:00
|
|
|
return self.request.user.is_authenticated() and \
|
|
|
|
self.request.user.admin_type in [AdminType.SUPER_ADMIN, AdminType.ADMIN]
|