Django: Выполнять поиск без учета регистра по умолчанию - PullRequest
Новая
       10

Django: Выполнять поиск без учета регистра по умолчанию

6 голосов
/ 10 февраля 2009

Мне нужно выполнить запросы без учета регистра для username по умолчанию при использовании Django Auth framework.

Я попытался исправить проблему, написав собственный подкласс Queryset и переопределив метод _filter_or_exclude, а затем с помощью этого подкласс в пользовательском менеджере для модели User - 

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, он работает безупречно.

Ответы [ 3 ]

5 голосов
/ 10 февраля 2009

Вы не хотите связываться с внутренними функциями классов Django. В этом и заключается проблема с каждым обновлением в будущем.

Если вы хотите изменить способ аутентификации, напишите пользовательский бэкэнд аутентификации.

Вот два рецепта.

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

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

Обе они нам по электронной почте вместо имени пользователя. Нетрудно использовать запрос без учета регистра вместо запроса по электронной почте.

4 голосов
/ 10 февраля 2009

Менеджеры не могут быть добавлены в классы с простым присвоением атрибутов (User.objects = MyManager()). Посмотрите на метакласс ModelBase (db / models / base.py), чтобы увидеть, что все делается для вас за кулисами, когда вы создаете подкласс Model.

Вы должны быть в состоянии заставить его работать с User.add_to_class('objects', MyManager()). В качестве альтернативы, вы можете сделать прокси подкласс пользователя и добавить туда менеджера.

0 голосов
/ 11 сентября 2013

Вот рецепт для случая использования auth: Django: регистронезависимое совпадение имени пользователя от пользователя auth? Возможно, вам лучше всего использовать отдельные решения для каждого из ваших вариантов использования.

...