Фильтр набора запросов модели Django как функция - PullRequest
Новая
       25

Фильтр набора запросов модели Django как функция

6 голосов
/ 09 февраля 2011

Я пытаюсь создать пользовательскую функцию, которую я могу поместить в «цепочку» набора запросов, которая будет применять фильтр к набору запросов. Как и в случае с обычными фильтрами наборов запросов Django, он будет брать набор запросов слева от него и передавать полученный набор запросов вправо.

Перед добавлением моей пользовательской функции в цепочку наборов запросов: 

models.MyModel.objects.all()
models.MyModel.objects.some_manger_function()
models.MyModel.objects.some_manger_function().count()
models.MyModel.objects.some_manger_function().filter(title='something')

После добавления моей пользовательской функции в цепочку наборов запросов: 

models.MyModel.objects.all().my_custom_filter()
models.MyModel.objects.some_manger_function().my_custom_filter()
models.MyModel.objects.some_manger_function().my_custom_filter().count()
models.MyModel.objects.some_manger_function().my_custom_filter()\
   .filter(title='something')

Я не уверен, как построить свою функцию для этого. Нужен ли какой-то декоратор вокруг него? 

???? def my_custom_filter(???):
????    return qs.filter(id__gte=10)

Кто-нибудь знает хороший способ сделать это?

Следующее может сработать, но я надеялся на что-то более похожее на Django. 

def my_custom_filter(qs):
    return qs.filter(id__gte=1)

my_custom_filter(models.MyModel.objects.all()).count()

Любой совет очень ценится.

Спасибо, Джо

ОБНОВЛЕНИЕ : Я пытаюсь выяснить детали решения Игнасио. Я не сделал слишком много с переопределением QuerySet, поэтому я собираю воедино то, что я могу найти ... 

class MyQuerySet(QuerySet):

    def filter(self, *args, **kwargs):
        return super(self.__class__, self).filter(*args, **kwargs).\
                     filter(id__gt=5)


class MyManager(models.Manager):

    def testqs(self):
        return MyQuerySet(self.model)

Однако я не думаю, что это работает так, как я ожидаю. Какие-либо предложения? 

>>> models.MyModel.objects.testqs().filter()

ОБНОВЛЕНИЕ 2 : Эта статья оказалась полезной. http://zmsmith.com/2010/04/using-custom-django-querysets/

1 Ответ

4 голосов
/ 09 февраля 2011

Вам потребуется написать собственный QuerySet дочерний класс с добавленным методом, затем использовать этот класс в менеджере .

...