Django：默认执行不区分大小写的查询

Question

在使用Django认证框架时，默认情况下需要对username执行不区分大小写的查询。

我尝试通过编写自定义Queryset子类并重写_filter_or_exclude方法来解决此问题，然后在用户模型的自定义管理器中使用该子类 -

from django.db.models import Manager
from django.db.models.query import QuerySet
from django.contrib.auth.models import UserManager

class MyQuerySet(QuerySet):
    def _filter_or_exclude(self, negate, *args, **kwargs):
        if 'username' in kwargs:
            kwargs['username__iexact'] = kwargs['username']
            del kwargs['username']
        return super(MyQuerySet, self)._filter_or_exclude(negate, *args, **kwargs)

class MyUserManager(UserManager):
    def get_query_set(self):
        return MyQuerySet(self.model)

User.objects = MyUserManager()

但是这种方法不起作用，当我尝试执行User.objects.get(username='Foo')时，出现了一个奇怪的错误。

任何帮助都将不胜感激。

更新: 我会附上我得到的确切错误。

/usr/lib/python2.5/site-packages/django/db/models/query.py in get(self, *args, **kwargs)
    295         keyword arguments.
    296         """
--> 297         clone = self.filter(*args, **kwargs)
    298         num = len(clone)
    299         if num == 1:

/usr/lib/python2.5/site-packages/django/db/models/query.py in filter(self, *args, **kwargs)
    481         set.
    482         """
--> 483         return self._filter_or_exclude(False, *args, **kwargs)
    484 
    485     def exclude(self, *args, **kwargs):

/home/ghoseb/src/git/ocricket.git/ocricket/user/models.py in _filter_or_exclude(self, negate, *args, **kwargs)
     38             kwargs['username__iexact'] = kwargs['username']
     39             del kwargs['username']
---> 40         return super(MyQuerySet, self)._filter_or_exclude(negate, *args, **kwargs)
     41 
     42 class MyUserManager(UserManager):

/usr/lib/python2.5/site-packages/django/db/models/query.py in _filter_or_exclude(self, negate, *args, **kwargs)
    499             clone.query.add_q(~Q(*args, **kwargs))
    500         else:
--> 501             clone.query.add_q(Q(*args, **kwargs))
    502         return clone
    503 

/usr/lib/python2.5/django/db/models/sql/query.py in add_q(self, q_object, used_aliases)

/usr/lib/python2.5/django/db/models/sql/query.py in add_filter(self, filter_expr, connector, negate, trim, can_reuse, process_extras)

/usr/lib/python2.5/django/db/models/sql/query.py in get_meta(self)

<type 'exceptions.AttributeError'>: 'NoneType' object has no attribute '_meta'

更新: 顺便提一下，当我将我的_filter_or_exclude方法中的逻辑复制到实际的QuerySet类中时，它可以无缝运行。

- Baishampayan Ghose

---

奇怪的错误。您能确认您在这个确切的代码上得到了这个错误吗？QuerySet或Manager没有其他的方法被覆盖，继承层次结构中也没有其他的中间子类或混合类吗？ - Carl Meyer

我怀疑问题可能出在monkeypatching User.objects上？但我还没有进行足够的实验来验证这一点。 - Carl Meyer

我可以确认。但如果有人能够验证它，那就太好了。 - Baishampayan Ghose

我是正确的；将管理器添加到类的过程比仅仅进行属性赋值更为复杂，它涉及调用add_to_class -- 这一切都在ModelBase元类中处理（请参见db/base.py）。因此，您的管理器从未被正确初始化。 - Carl Meyer

请参见https://dev59.com/M2Ys5IYBdhLWcg3wDPit#18731796。 - Bryce

Answer 1

您不应该随意改动Django类的内部特性，否则在未来的每次升级中都会遇到麻烦。

如果您想要更改用户认证方式，请编写自定义的认证后端。

以下是两个示例：

http://www.davidcramer.net/code/224/logging-in-with-email-addresses-in-django.html

http://www.djangosnippets.org/snippets/577/

这两个示例都使用电子邮件地址而不是用户名进行身份验证。使用大小写不敏感的查询代替电子邮件查询并不困难。

- S.Lott

---

我非常了解自定义身份验证后端，但我不仅需要它用于身份验证，还需要用于其他方面，例如URL中使用的用户名等。 - Baishampayan Ghose

Answer 2

经理不能通过简单的属性分配（User.objects = MyManager()）添加到类中。查看ModelBase元类（db/models/base.py），了解在子类化Model时在幕后为您完成的所有操作。

您应该能够使用User.add_to_class('objects', MyManager())使其正常工作。或者，您可以创建一个User的代理子类并在那里添加管理器。

- Carl Meyer

---

我不确定一个具体类的抽象子类是否被允许或有意义，但代理模式在这种情况下会很好地工作。 - rz.

@rz 当此答案撰写时，代理模型尚不存在 :-) 我会进行更新。 - Carl Meyer

酷！我想我应该利用我新获得的编辑超能力为您完成。顺便说一句，卡尔，除了他添加管理器的方式外，这种获取区分大小写存储并进行不区分大小写查找的方法合理吗？ - rz.

@rz 我不认为这是疯狂的...对我来说有点神奇，我更喜欢使用一个经理，只需添加一个额外的lookup_user或类似方法来执行不区分大小写的查找，然后在我的代码中使用该新方法。如果不了解更多要求/限制，很难确定。 - Carl Meyer

Answer 3

这是一个关于认证用例的配方： Django：如何对认证用户的用户名进行不区分大小写的匹配？ 最好针对每个用例使用单独的解决方案。

- Bryce