Question

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

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

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

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

Answer 1

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

# 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

Answer 2

这是相当丑陋的，但你可能可以对用户objects属性进行monkeypatch，例如。在中间件中：

# 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：如果有人提出别的建议，你应该听他或她比我更多。

Note4：作为一个可能更好的想法，为什么不尝试查询Profile，这是一个你完全可以控制的模型类？总是可以从配置文件中检索用户对象，所以......

在django中覆盖UserManager

2 个答案: