Расширенная фильтрация Django Tastypie: как выполнять сложные поиски с объектами Q

У меня есть базовая модель Django, например:

class Business(models.Model):
    name = models.CharField(max_length=200, unique=True)
    email = models.EmailField()
    phone = models.CharField(max_length=40, blank=True, null=True)
    description = models.TextField(max_length=500)

Мне нужно выполнить сложный запрос к приведенной выше модели, например:

qset = (
    Q(name__icontains=query) |
    Q(description__icontains=query) |
    Q(email__icontains=query)
    )
results = Business.objects.filter(qset).distinct()

Я пробовал следующее, используя tastypie, но безуспешно:

def build_filters(self, filters=None):
    if filters is None:
        filters = {}
    orm_filters = super(BusinessResource, self).build_filters(filters)

    if('query' in filters):
        query = filters['query']
        print query
        qset = (
                Q(name__icontains=query) |
                Q(description__icontains=query) |
                Q(email__icontains=query)
                )
        results = Business.objects.filter(qset).distinct()
        orm_filters = {'query__icontains': results}

    return orm_filters

а в классе Meta для deliciouspie у меня установлена ​​​​фильтрация как:

filtering = {
        'name: ALL,
        'description': ALL,
        'email': ALL,
        'query': ['icontains',],
    }

Любые идеи о том, как я могу справиться с этим?

Спасибо - Ньютон


person nknganda    schedule 05.04.2012    source источник


Ответы (3)


Ты на правильном пути. Однако предполагается, что build_filters переводит поиск ресурсов в поиск ORM.

Реализация по умолчанию разбивает ключевое слово запроса на основе __ на key_bits, пары значений, а затем пытается найти сопоставление между искомым ресурсом и его эквивалентом ORM.

Ваш код не должен применять фильтр только для его создания. Вот улучшенная и исправленная версия:

def build_filters(self, filters=None):
    if filters is None:
        filters = {}
    orm_filters = super(BusinessResource, self).build_filters(filters)

    if('query' in filters):
        query = filters['query']
        qset = (
                Q(name__icontains=query) |
                Q(description__icontains=query) |
                Q(email__icontains=query)
                )
        orm_filters.update({'custom': qset})

    return orm_filters

def apply_filters(self, request, applicable_filters):
    if 'custom' in applicable_filters:
        custom = applicable_filters.pop('custom')
    else:
        custom = None

    semi_filtered = super(BusinessResource, self).apply_filters(request, applicable_filters)

    return semi_filtered.filter(custom) if custom else semi_filtered

Поскольку вы используете объекты Q, стандартный метод apply_filters недостаточно умен, чтобы применить ваш пользовательский ключ фильтра (поскольку его нет), однако вы можете быстро переопределить его и добавить специальный фильтр под названием «пользовательский». При этом ваш build_filters может найти подходящий фильтр, построить то, что он означает, и передать его как пользовательский в apply_filters, который просто применит его напрямую, а не попытается распаковать его значение из словаря как элемент.

person astevanovic    schedule 05.04.2012
comment
Словарь не имеет метода «расширить». Должно быть: orm_filters.update({'custom': qset}) - person Karmo Rosental; 06.12.2012
comment
Это решение вызывает вызов БД дважды (для полуфильтрованного, а затем для пользовательского фильтра). У меня работает немного другой код: если «пользовательский» в применимых_фильтрах: custom = применимые_фильтры. - person Israel Zalmanov; 24.09.2013

Я решил эту проблему так:

Class MyResource(ModelResource):

  def __init__(self, *args, **kwargs):
    super(MyResource, self).__init__(*args, **kwargs)
    self.q_filters = []

  def build_filters(self, filters=None):
    orm_filters = super(MyResource, self).build_filters(filters)

    q_filter_needed_1 = []
    if "what_im_sending_from_client" in filters:
      if filters["what_im_sending_from_client"] == "my-constraint":
        q_filter_needed_1.append("something to filter")

    if q_filter_needed_1:
      a_new_q_object = Q()
      for item in q_filter_needed:
        a_new_q_object = a_new_q_object & Q(filtering_DB_field__icontains=item)
      self.q_filters.append(a_new_q_object)

  def apply_filters(self, request, applicable_filters):
    filtered = super(MyResource, self).apply_filters(request, applicable_filters)

    if self.q_filters:
      for qf in self.q_filters:
        filtered = filtered.filter(qf)
      self.q_filters = []

    return filtered

Этот метод кажется более четким разделением задач, чем другие, которые я видел.

person Ted    schedule 20.08.2012
comment
Это действительно плохая идея помещать информацию, специфичную для запроса, в экземпляр ресурса. Таким образом, 1_. Это связано с тем, что в развернутой среде с несколькими потоками вы можете столкнуться с тем, что состояние одного запроса будет влиять на состояние другого. Так, например, все фильтры, созданные в одном запросе, на самом деле могут быть применены к совершенно другому запросу, в зависимости от времени. См. документы здесь: django-tastypie.readthedocs.io /en/latest/ Это проблема, которую решает передача объекта bundle повсюду. - person Cameron Lee; 05.12.2017

Взяв идею в ответе астевановича и немного очистив ее, следующее должно работать и быть более кратким.

Основное отличие заключается в том, что apply_filters стал более надежным за счет использования None в качестве ключа вместо custom (что может конфликтовать с именем столбца).

def build_filters(self, filters=None):
    if filters is None:
        filters = {}
    orm_filters = super(BusinessResource, self).build_filters(filters)

    if 'query' in filters:
        query = filters['query']
        qset = (
                Q(name__icontains=query) |
                Q(description__icontains=query) |
                Q(email__icontains=query)
                )
        orm_filters.update({None: qset}) # None is used as the key to specify that these are non-keyword filters

    return orm_filters

def apply_filters(self, request, applicable_filters):
    return self.get_object_list(request).filter(*applicable_filters.pop(None, []), **applicable_filters)
    # Taking the non-keyword filters out of applicable_filters (if any) and applying them as positional arguments to filter()
person Cameron Lee    schedule 05.12.2017