我正在尝试将以下visible
函数中的两个计数查询合并为一个查询。
如果没有关系或者存在关系且某些特定过滤为真,则函数应返回True。
class OtherModel(models.Model):
starts = models.DateField()
ends = models.DateField()
class MyModel(models.Model):
m2m = models.ManyToManyField('OtherModel', blank=True, )
def visible(self):
# Should always return True if no relations to OtherModel are present.
if self.m2m.exists():
# If relations to OtherModel are present check for starts and ends.
# The reason for the first check is that if there are no relations
# and the below query returns 0 the function will return False
today = datetime.date.today()
return self.m2m.filter(starts__lte=today, ends__gte=today).exists()
return True
编辑:更多代码和评论,替换为存在的计数。
m2m关系用于日期限制,但如果没有可用的日期限制,则该函数应返回True(如果没有任何限制则对象可见,但如果存在限制但不匹配当前日期则不可见)。
示例代码只是一个简化示例,我将需要执行此操作并实际返回查询集。
答案 0 :(得分:3)
用Q和聚合解决它,感谢Aamir for exists():
def visible(self):
today = datetime.date.today()
return MyModel.objects.annotate(num_m2m=Count('m2m')).filter(
Q(pk=self.pk) &
(
Q(m2m__starts__lte=today) &
Q(m2m__ends__gte=today) |
Q(num_m2m=0)
)
).exists()
答案 1 :(得分:1)
不确定您的实际要求是什么,但我认为您希望避免self.m2m.count() > 0
额外检查并希望从函数返回布尔值(如果存在与否),那么您需要.exists如果QuerySet包含任何结果,则返回True
;如果不包含,则返回False
:
def visible(self):
# return True or False
return self.m2m.filter(some_filter=some_value).exists()