【发布时间】:2010-10-06 14:57:37
【问题描述】:
我需要默认对username执行不区分大小写的查询
使用 Django Auth 框架时。
我尝试通过编写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 类中时,它可以完美运行。
【问题讨论】:
-
奇怪的错误。你能验证你在这个确切的代码上得到这个错误吗? QuerySet 或 Manager 上没有额外的方法被覆盖,继承层次结构中没有额外的干预子类,没有额外的 mixins?
-
我怀疑问题可能在于猴子修补 User.objects?但我还没有做足够的实验来验证这一点。
-
我可以确认。但如果有人可以验证它,那就太好了。
-
我是对的;将管理器添加到类的过程比属性分配更复杂,它涉及调用 add_to_class ——这一切都在 ModelBase 元类中处理(参见 db/base.py)。所以你的经理永远不会正确初始化。
标签: python django django-models django-orm