10

如何auth_user在 django 中为类使用自定义管理器？

在我的 django 项目中，我正在使用auth_user并且我有一个基本的配置文件类。在我网站的每个页面中，我都会使用一些用户和个人资料数据，因此每个用户查询都应该加入个人资料。

我想在自定义管理器中使用select_related该get_query_set()方法，但我找不到任何合适的方法来定义一个或覆盖现有的UserManager. 有任何想法吗？

注意：我不想覆盖用户模型。或者，更准确地说，我已经在不同的代理模型中覆盖了它。我希望在每个代理模型中都使用这个自定义管理器。

4

2 回答 2

8

好的，终于找到正确答案了。最简洁的方法是使用自定义身份验证后端。

# in settings:
AUTHENTICATION_BACKENDS = ('accounts.backends.AuthenticationBackend',)


# in accounts/backends.py:
from django.contrib.auth.backends import ModelBackend
from django.contrib.auth.models import User


class AuthenticationBackend(ModelBackend):
    def get_user(self, user_id):
        try:
            # This is where the magic happens
            return User.objects. \
                select_related('profile'). \
                get(pk=user_id)
        except User.DoesNotExist:
            return None
于 2013-06-13T13:27:00.923 回答
4

这相当难看，但您可能可以对 Userobjects属性进行猴子补丁，例如。在中间件中：

# manager.py
from django.contrib.auth.models import UserManager

class MyUserManager(UserManager):
    def get_query_set(self):
        qs = super(MyUserManager, self).get_query_set()
        return qs.select_related('profile')

# middleware.py
from django.contrib.auth.middleware import AuthenticationMiddleware
from managers import MyUserManager

class MyAuthMiddleware(AuthenticationMiddleware):
    def process_request(self, request):
        super(AuthenticationMiddleware, self).process_request(request)
        User.objects = MyUserManager()
        return None

然后替换以下行settings.py：

MIDDLEWARE_CLASSES = (
    # ...
    'django.contrib.auth.middleware.AuthenticationMiddleware',
    # ...
)

经过：

# settings.py
MIDDLEWARE_CLASSES = (
    # ...
    'yourapp.middleware.MyAuthMiddleware',
    # ...
)

注意1：此代码纯粹是理论上的，从未测试过，我也没有时间。

注意 2：从长期可维护性的角度来看，我不建议使用此解决方案。

注3：如果有人提出其他建议，你可能应该多听他或她而不是我。

注意 4：作为一个更好的主意，为什么不尝试查询 Profile，这是一个您可以完全控制的模型类？无论如何，您始终可以从配置文件中检索用户对象，所以……</p>

于 2011-06-29T13:14:12.337 回答