Cavemanon/cavemanon-wagtail-personalisation
Archived
7
Fork 0
Code Issues Pull Requests Activity

Simpler queries

Browse Source
This commit is contained in:
Jasper Berghoef
2017-06-02 15:06:53 +02:00
committed by Michael van Tellingen
parent f21c423b1c
commit aafc8c4ed5
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/wagtail_personalisation/models.py
View File

@ -81,17 +81,15 @@ class Segment(ClusterableModel):
def get_used_pages(self):
"""Return the pages that have variants using this segment."""
pages = [page for page
in PersonalisablePageMetadata.objects.filter(segment=self)]
pages = list(PersonalisablePageMetadata.objects.filter(segment=self))
return pages
def get_created_variants(self):
"""Return the variants using this segment."""
pages = [page.variations for page
in PersonalisablePageMetadata.objects.filter(segment=self)]
pages = Page.objects.filter(_personalisable_page_metadata__segment=self)
return list(itertools.chain(*pages))
return pages
def get_rules(self):
"""Retrieve all rules in the segment."""