Проверить наличие пустых наборов запросов в 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 (или создать внешнюю функцию, которая будет выполнять проверку).