Проверить наличие пустых наборов запросов в Django

У меня есть собственный менеджер для моей модели устройства. Если возвращаемый набор запросов пуст, я хочу вернуть другой объект. Проблема в том, что пустой набор запросов не оценивается как пустой.

Я изучил эти темы, чтобы узнать, как проверить, является ли набор запросов пустым. В Django, какой самый эффективный способ проверить наличие пустого набора запросов? Проверка на пустой набор запросов в Django

Это моя модель:

    class DeviceField(models.Model):
    """
    Can be connected to any class and any field in that class. Specifies how the values and field names are returned for the field.
    """
    device = models.ForeignKey(Device, verbose_name=_("Device"))
    content_type = models.ForeignKey(ContentType, verbose_name=_("Class"))
    field_name = models.CharField(max_length=150, verbose_name=_("field name"))
    display_field = models.CharField(max_length=255, verbose_name=_("How to display the field name?"))
    display_value = models.CharField(max_length=255, verbose_name=_("How to display the value?"))

    objects = DeviceFieldManager()

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

class DeviceFieldManager(models.Manager):
    """
    Behaves like a normal manager. Except in the case that no such device field exists. Then it will return the standard value of the field.
    """
    def get_query_set(self):
        """
        Does the queryset overriding :). If empty, get the original values
        """
        queryset = super(DeviceFieldManager,self).get_query_set()
        try:
            a = queryset[0]
            print "we have something here"
        except IndexError:
            print "It is frigging empty"
        if queryset == super(DeviceFieldManager, self).none():
            print "this queryset is just like an empty one"
        if not queryset:
            print "not queryset"
        if not queryset.count():
            print "not queryset count"
        if queryset:
            print "if queryset"
        if queryset.exists():
            print "queryset exists"
        else:
            print "A value is return, we don't need to do anything!"
        print queryset.count()
        print super(DeviceFieldManager, self).none()
        print queryset == super(DeviceFieldManager, self).none()
        return queryset

И это из скорлупы. Для устройств "мобильный" (который существует) и "корова" (который не существует) менеджер показывает одинаковое поведение. Когда я пытаюсь напечатать первое значение набора запросов, я получаю ожидаемый IndexError для "коровы".

In [8]: a= DeviceField.objects.filter(device__name="mobile")
we have something here
if queryset
queryset exists
1
[]
False

In [9]: a[0]
Out[9]: <DeviceField: DeviceField object>

In [10]: a= DeviceField.objects.filter(device__name="cow")
we have something here
if queryset
queryset exists
1
[]
False

In [11]: a[0]
---------------------------------------------------------------------------
IndexError                                Traceback (most recent call last)
/<ipython-input-11-5ccf417d7af1> in <module>()
----> 1 a[0]

/local/lib/python2.7/site-packages/django/db/models/query.pyc in __getitem__(self, k)
    188             qs = self._clone()
    189             qs.query.set_limits(k, k + 1)
--> 190             return list(qs)[0]
    191         except self.model.DoesNotExist, e:
    192             raise IndexError(e.args)

IndexError: list index out of range

Что я здесь не так делаю? Как я могу проверить в менеджере, если набор запросов пуст или нет?

Большое спасибо:)

1 ответ

Управляющего get_query_set вызывается перед фильтрацией, поэтому вам нужно как-то взломать сам QuerySet (или создать внешнюю функцию, которая будет выполнять проверку).

Другие вопросы по тегам