将QuerySet传递给Django 1.11中的Celery Task

Question

我想将一些长查询移至Celery中的异步任务，然后使用AJAX检索该额外信息。 现在，我得到这样的QuerySet：

brands = Brand.objects.filter(shops__shop_name__in=[shop])

任务：

@task()
def brand_count(querystr):
    querystr.annotate(amount_of_products=Count('products'))

我想在这里做：

task_run = brand_count.delay(brands)

问题是： 如何将QuerySet传递给Celery任务？ 现在，它抛出一个错误，表明此操作无法完成。

我发现这里可能会使用pickle ，但是我找不到正确的使用方法。 特别是- # Assuming 's' is the pickled string. 在query = pickle.loads(s)均值。

Answer 1

如您所知，您无法将查询集直接传递给任务，因为任务参数必须可序列化。 酸洗也不是一个理想的选择，因为您实际上不需要序列化整个查询集。

相反，您应该将对象ID列表传递给任务，然后从任务本身获取查询集。

brand_ids = Brand.objects.filter(shops__shop_name__in=[shop]).values_list('id', flat=True)
task_run = brand_count.delay(list(brand_ids))

values_list将为您提供品牌ID的列表。

然后，在您的任务中，重新创建查询集并对其执行所需的操作：

@task()
def brand_count(brand_ids):
    queryset = Brand.objects.filter(id__in=brand_ids)
    queryset.annotate(amount_of_products=Count('products'))