8

fixes wagtail2 compatibility

return QuerySets instead of lists
This commit is contained in:
Paul J Stevens
2018-03-17 11:26:56 +01:00
parent 32e73329c3
commit 62d258fd9e
2 changed files with 8 additions and 12 deletions

View File

@@ -80,15 +80,11 @@ class Segment(ClusterableModel):
def get_used_pages(self):
"""Return the pages that have variants using this segment."""
pages = list(PersonalisablePageMetadata.objects.filter(segment=self))
return pages
return PersonalisablePageMetadata.objects.filter(segment=self)
def get_created_variants(self):
"""Return the variants using this segment."""
pages = Page.objects.filter(_personalisable_page_metadata__segment=self)
return pages
return Page.objects.filter(_personalisable_page_metadata__segment=self)
def get_rules(self):
"""Retrieve all rules in the segment."""