[英]Check if object_id occurs more than once in queryset.annotate Case When parameter
在我的情况下,文档字段查找并没有真正帮助
我的查询现在是什么样子
date_delta = 2
queryset = TrendData.objects.filter(owner__trend_type__mnemonic='posts',
date_trend__date__range=[date_from, date_to]).values('owner_id', 'owner__name')
queryset.annotate(owner_name=F('owner_id__name')).values('owner_name', 'owner_id').annotate(
views = Sum(Case(When(owner_id__gt=1, then=F('views') / date_delta)), default=('views')...,
output_field=IntegerField() )
)
该queryset
输出如下所示:
{'owner_id': 1306, 'owner__name': 'Some name123'},
{'owner_id': 1307, 'owner__name': 'Somename as well'},
{'owner_id': 1308, 'owner__name': 'aand another name'},
{'owner_id': 1306, 'owner__name': 'Some name123'}
如您所见,存在匹配的 owner_id,并且查询集 len() 每天 100k,因此如果日期范围是 5 天,则查询集 len() == 500k。 我的models.py 看起来像这样
class Owner(models.Model):
class Meta:
verbose_name_plural = 'Objects'
TREND_OWNERS = Choices('group', 'user')
link = models.CharField(max_length=255)
name = models.CharField(max_length=255)
owner_type = models.CharField(choices=TREND_OWNERS, max_length=50)
trend_type = models.ForeignKey(TrendType, on_delete=models.CASCADE)
def __str__(self):
return f'{self.link}[{self.trend_type}]'
class TrendData(models.Model):
class Meta:
verbose_name_plural = 'Trends'
owner = models.ForeignKey(Owner, on_delete=models.CASCADE)
views = models.IntegerField()
views_u = models.IntegerField()
likes = models.IntegerField()
shares = models.IntegerField()
interaction_rate = models.DecimalField(max_digits=20, decimal_places=10)
mean_age = models.IntegerField()
date_trend = models.DateTimeField()
我意识到它可以正常工作,但它会出错,因为如果 owner_id 很好,它将除以 date_delta,在我的情况下,我想要如果 owner_id 在查询集中出现不止一次。 我试过owner_id__count__gt
但那不存在:(
我很想知道是否有办法计算我的注释Case(When())
查询集中的 owner_id 出现次数。 这将真正解决我的问题。 如果它大于 1 比我们除以 date_delta,否则我们保持原样
更新:
需要明确的是,这个注释做得很好,但是它也划分了一些我不想划分的查询(在我的情况下,非重复的 owner_id 查询集仍然将它的视图、共享等除以 2)所以这就是为什么我上面提到的用例(When())
queryset.values('owner__name', 'owner_id').annotate(
views=Sum('views') / 2,
views_u=Sum('views_u') / 2,
likes=Sum('likes') / 2,
shares=Sum('shares') / 2,
interaction_rate=Sum('interaction_rate') / 2,
mean_age=Sum('mean_age') / 2)
更新 #2这是我的逻辑,但在 python 中
json_output = []
for item in (queryset
.values('owner__name', 'owner_id')
.annotate(owner_count=Count('owner_id'))
.annotate(views=Sum('views'), views_u=Sum('views_u'),
likes=Sum('likes'),
shares=Sum('shares'),
interaction_rate=Sum('interaction_rate'),
mean_age=Sum('mean_age')):
if item['owner_count'] > 1:
item['views'] = item['views'] / date_delta
item['views_u'] = item['views_u'] / date_delta
item['likes'] = item['likes'] / date_delta
item['shares'] = item['shares'] / date_delta
item['interaction_rate'] = '{:.10f}'.format(
Decimal(item['interaction_rate']) / date_delta)
item['mean_age'] = item['mean_age'] / date_delta
json_output.append(item)
else:
json_output.append(item)
首先,我认为这是错误的owner_name=F('owner_id__name'
它应该是owner_name=F('owner__name'
。
如果我理解,您想用拥有所有者的 TrendData 实例的数量来注释 TrendData 查询集。
您可以使用子查询来实现:
owner_td_count = Owner.objects.annotate(
td_count=Count('trenddata_set')
).filter(
id=OuterRef('owner_id')
).values('td_count')[:1]
然后首先通过计算 owner_id 的出现来注释:
queryset.annotate(
owner_name=F('owner__name'),
owner_id_count=Subquery(owner_td_count) # How many DataTrend's have the owner with id=owner_id
).values('owner_name', 'owner_id').annotate(
# ...
)
)
然后你可以在你的情况下/建造时:
Case(
When(
owner_id_count=1, then=F('views'),
default=F('views') / date_delta)),
output_field=IntegerField()
)
)
更新:事实证明我毕竟没有完全测试过(我以为我有,抱歉)。 您需要将Case
包裹在Sum
周围, Sum
( Sum
围绕Case
)将不起作用,无论 Django 版本如何:
(queryset
.values('owner', owner_name=F('owner__name'))
.annotate(owner_count=Count('owner'))
.annotate(views = Case(
When(owner_count__gt=1,
then=Sum(F('views') / date_delta)),
default=Sum('views'),
output_field=IntegerField()
))
)
一个细微的变化是使用子查询。 Raydel 为每个Owner
计算Trenddata
计数的子查询在原则上是有效的,但由于它对Trenddata
每一行(不仅仅是唯一的Owner
s)进行聚合,因此速度会非常慢。
不同的子查询提供了获得相同结果的更快方法。 它只对Trenddata
Owner
进行一次计数,然后检查每个Trenddata
对象是否其所有者在列表中。 我认为这应该仍然比我的第一个查询慢,但奇怪的是,它在我的简短测试中表现得相当(大约 300 万行)。
(queryset
.values('owner', owner_name=F('owner__name'))
.annotate(multi=Case(
When(owner__in=Subquery(TrendData.objects
.values('owner')
.annotate(cnt=Count('owner'))
.filter(cnt__gt=0)
.values('owner')),
then=1),
default=0,
output_field=IntegerField())
)
.annotate(views = Case(
When(multi=1,
then=Sum(F('views') / date_delta)),
default=Sum('views'),
output_field=IntegerField())
)
)
您可以将两个注释合二为一,但是如果您要为多个注释重用multi
而不是像我的示例中那样只使用一个注释,则将两者分开可以避免为每个注释重复子查询。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.