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

Fix segment model naming for forced_segment and the userbar

Browse Source
This commit is contained in:
blurrah
2018-05-26 16:37:40 +02:00
parent f1b62a7546
commit 65a46f2bd9
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/wagtail_personalisation/wagtail_hooks.py
View File

@ -57,7 +57,7 @@ def segment_user(page, request, serve_args, serve_kwargs):
forced_segment = request.GET.get('segment', None) forced_segment = request.GET.get('segment', None)
if request.user.is_superuser and forced_segment is not None: if request.user.is_superuser and forced_segment is not None:
segment = Segment.objects.filter(pk=forced_segment).first() segment = models.Segment.objects.filter(pk=forced_segment).first()
if segment: if segment:
adapter.set_segments([segment]) adapter.set_segments([segment])
@ -75,7 +75,7 @@ class UserbarSegmentedLinkItem:
@hooks.register('construct_wagtail_userbar') @hooks.register('construct_wagtail_userbar')
def add_segment_link_items(request, items): def add_segment_link_items(request, items):
for item in Segment.objects.enabled(): for item in models.Segment.objects.enabled():
items.append(UserbarSegmentedLinkItem(item)) items.append(UserbarSegmentedLinkItem(item))
return items return items