如何在自定义类管理器的方法中获取上一个查询返回的数据?

时间:2011-09-19 18:02:50

标签: python django oop django-managers

我使用可链接方法开发自定义管理器类。出了问题。我需要随机化过滤后的查询。要获得随机记录,我需要计算过滤和不同的记录。但我不知道如何得到它。相反,我有所有记录的数量。

class RandomQueryset(models.query.QuerySet):

    def randomize(self):        
        count = self.aggregate(count=Count('id'))['count']
        random_index = random.randint(0, count - 1)
        return self.all()[random_index]    


class RandomManager(models.Manager):

    def get_query_set(self):
        return RandomQueryset(self.model, using=self._db)

    def randomize(self):
        return self.get_query_set().randomize()

使用:

>>> posts = PostPages.random_objects.filter(image_gallery__isnull=False).distinct()

>>> posts.randomize()

迟早会收到错误,因为该计数超过了当前查询中的记录数。

IndexError: list index out of range

1 个答案:

答案 0 :(得分:1)

你已经让我发布了我的一个问题作为答案,所以这里就是。

这看起来你需要使用Django的内置count()函数docs.djangoproject.com/en/dev/ref/models/querysets

我没有在你的代码中看到它的使用。