如何使用自定义身份验证后端连接django中的管理页面?

时间:2015-04-20 16:16:24

标签: django django-admin django-authentication

我在我的django项目中使用自定义身份验证后端来连接用户,我的问题是我无法再使用管理界面进行连接。 这是我的自定义身份验证文件: 的 auth_backends.py:

from django.conf import settings
from django.contrib.auth.backends import ModelBackend
from django.core.exceptions import ImproperlyConfigured
from django.db.models import get_model


class StudentModelBackend(ModelBackend):
def authenticate(self, username=None, password=None):
    try:
        user = self.user_class_s.objects.get(username=username)
        if user.check_password(password):
            return user
    except self.user_class_s.DoesNotExist:
        return None

def get_user(self, user_id):
    try:
        return self.user_class_s.objects.get(pk=user_id)
    except self.user_class_s.DoesNotExist:
        return None

@property
def user_class_s(self):
    if not hasattr(self, '_user_class'):
        self._user_class = get_model(*settings.CUSTOM_USER_MODEL.split('.', 2))
        if not self._user_class:
            raise ImproperlyConfigured('Could not get student model')
    return self._user_class


class ProfessorModelBackend(ModelBackend):
def authenticate(self, username=None, password=None):
    try:
        user = self.user_class_p.objects.get(username=username)
        if user.check_password(password):
            return user
    except self.user_class_p.DoesNotExist:
        return None

def get_user(self, user_id):
    try:
        return self.user_class_p.objects.get(pk=user_id)
    except self.user_class_p.DoesNotExist:
        return None

@property
def user_class_p(self):
    if not hasattr(self, '_user_class'):
        self._user_class = get_model(*settings.CUSTOM_USER_MODEL_P.split('.', 2))
        if not self._user_class:
            raise ImproperlyConfigured('Could not get student model')
    return self._user_class

settings.py:

AUTHENTICATION_BACKENDS = (
'authentification.auth_backends.StudentModelBackend',
'authentification.auth_backends.ProfessorModelBackend',
)

CUSTOM_USER_MODEL = 'forum.Student'
CUSTOM_USER_MODEL_P = 'forum.Professor'

我试过这个解决方案:     来自django.contrib.auth.decorators导入login_required

admin.autodiscover()
admin.site.login = login_required(admin.site.login)

但它将我重定向到用户auth界面而不是管理界面。

有人可以帮帮我吗?

2 个答案:

答案 0 :(得分:2)

我找到了解决方案, 如果用户不是学生或教授,我只需告诉django尝试使用原始后端进行连接。 只需添加' django.contrib.auth.backends.ModelBackend'在settings.py

settings.py:

AUTHENTICATION_BACKENDS = (
'authentification.auth_backends.StudentModelBackend',
'authentification.auth_backends.ProfessorModelBackend',
'django.contrib.auth.backends.ModelBackend',
)

答案 1 :(得分:0)

默认的login_required装饰器默认为settings.py LOGIN_REDIRECT_URL。但是,您也可以通过传递login_url关键字参数明确告诉它使用哪个登录视图。

https://github.com/django/django/blob/stable/1.8.x/django/contrib/auth/decorators.py#L39

这可以与通用login_required视图一起使用,下面的示例通过使用user_passes_test装饰器显示更精确的条件。

例如:

from django.contrib.auth.decorators import user_passes_test

def superuser_required(*args, **kwargs):
    return user_passes_test(lambda u: u.is_superuser, login_url='admin:login')(*args, **kwargs)

def custom_login_required(*args, **kwargs):
    return user_passes_test(lambda u: getattr(u, 'is_custom_user', False), login_url='custom-url-name-from-urls-py')(*args, **kwargs)

然后你应该可以使用这个新的装饰,如上所述,使用通用的login_required

admin.site.login = custom_login_required(admin.site.login)