7

Merge pull request #108 from LabD/feature/segment-template-block

Adds an ifsegment template block
This commit is contained in:
Boris Besemer
2017-05-31 14:02:08 +02:00
committed by GitHub
4 changed files with 125 additions and 5 deletions

View File

@ -97,8 +97,11 @@ class SessionSegmentsAdapter(BaseSegmentsAdapter):
:rtype: wagtail_personalisation.models.Segment or None
"""
return next(item for item in self.request.session['segments']
if item.id == segment_id)
try:
return next(item for item in self.request.session['segments']
if item['id'] == segment_id)
except StopIteration:
return None
def add(self, segment):
"""Add a segment to the request session.