0

我有以下代码:

def get_elements(self, obj):
  book_elements = Element.objects.filter(book__pk=obj.id)
  elements = Element.objects.filter( (Q(book__pk=obj.id) | Q(theme__pk=obj.theme_id)), ~Q(pk__in = [o.element_id for o in book_elements if o.element_id]))
  serializer = GetElementSerializer(elements, context=self.context, many=True)

元素变量是使用 Q 对象实现的查询。但是,Q(book__pk=obj.id)book_elements引用完全相同的一组值。如何引用Q(book__pk=obj.id)列表理解的内部以避免运行 2 个查询。类似于以下内容:

def get_elements(self, obj):
  elements = Element.objects.filter( (Q(book__pk=obj.id) | Q(theme__pk=obj.theme_id)), ~Q(pk__in = [o.element_id for o in Q(book__pk=obj.id) if o.element_id]))
  serializer = GetElementSerializer(elements, context=self.context, many=True)

我的元素模型按要求:

class Element(models.Model):
  name = models.CharField(max_length=100)
  pub_date = models.DateTimeField('date published', auto_now_add=True)
  mod_date = models.DateTimeField('modified date', auto_now=True)
  book = models.ForeignKey(Book, blank=True, null=True, related_name='elements')
  book_part = models.ForeignKey(BookPart)
  theme = models.ForeignKey(Theme, blank=True, null=True, related_name='elements')
  element = models.ForeignKey('self', blank=True, null=True, related_name='parent')
  def __str__(self):
    return self.name

任何帮助表示赞赏!

4

2 回答 2

0

您应该能够在这样的一个查询中做到这一点。它利用了反向关系。

elements = Element.objects.filter( 
    Q(book=obj) | Q(theme__pk=obj.theme_id)
).exclude(
    book=obj, element__isnull=False
)
于 2015-04-01T01:35:14.820 回答
0

这是一个带有子查询解决方案的查询:

elements = Element.objects.filter( 
    (Q(book__pk=obj.id) | Q(theme__pk=obj.theme_id)),
    ~Q(pk__in = Element.objects.filter(
            book__pk=obj.id,
            element__isnull=False
        ).values_list('element', flat=True)
    )
)

这只会命中数据库一次。

顺便说一句,在阅读了您实际想要实现的目标之后:

给我所有元素,带有 book_id 或 theme_id,等于本书的书籍 ID 和主题 ID,不包括与本书中的另一个元素有 FK 关系的元素

在 ORM 方式中,我认为这也可以这样实现:

elements = Element.objects \
    .filter(Q(book_id=book.id)|Q(theme_id=book.theme_id)) \
    .exclude(element__in=book.elements.all()) #not tested, if throws an error transform it into: book.elements.values_list('id', flat=True)

同样,这只会访问数据库一次,因为querysets它们很懒惰并且Django足够聪明,可以在它们分解为另一个时将它们转换为子查询queryset

于 2015-03-31T22:32:22.303 回答